286 CheckInstanceNormalizationLayerVisitorNameAndDescriptor)
307 CheckAdditionLayerVisitorNameNullptrAndDescriptor)
309 CheckArgMinMaxLayerVisitorNameNullptrAndDescriptor)
311 CheckDepthToSpaceLayerVisitorNameNullptrAndDescriptor)
313 CheckBatchToSpaceNdLayerVisitorNameNullptrAndDescriptor)
315 CheckComparisonLayerVisitorNameNullptrAndDescriptor)
317 CheckConcatLayerVisitorNameNullptrAndDescriptor)
319 CheckElementwiseUnaryLayerVisitorNameNullptrAndDescriptor)
321 CheckFillLayerVisitorNameNullptrAndDescriptor)
323 CheckGatherLayerVisitorNameNullptrAndDescriptor)
325 CheckInstanceNormalizationLayerVisitorNameNullptrAndDescriptor)
327 CheckL2NormalizationLayerVisitorNameNullptrAndDescriptor)
329 CheckLogicalBinaruLayerVisitorNameNullptrAndDescriptor)
331 CheckLogSoftmaxLayerVisitorNameNullptrAndDescriptor)
333 CheckMeanLayerVisitorNameNullptrAndDescriptor)
335 CheckNormalizationLayerVisitorNameNullptrAndDescriptor)
337 CheckPadLayerVisitorNameNullptrAndDescriptor)
339 CheckPermuteLayerVisitorNameNullptrAndDescriptor)
341 CheckPooling2dLayerVisitorNameNullptrAndDescriptor)
343 CheckReshapeLayerVisitorNameNullptrAndDescriptor)
345 CheckResizeLayerVisitorNameNullptrAndDescriptor)
347 CheckSliceLayerVisitorNameNullptrAndDescriptor)
349 CheckSoftmaxLayerVisitorNameNullptrAndDescriptor)
351 CheckSpaceToBatchNdLayerVisitorNameNullptrAndDescriptor)
353 CheckSpaceToDepthLayerVisitorNameNullptrAndDescriptor)
355 CheckSplitterLayerVisitorNameNullptrAndDescriptor)
357 CheckStackLayerVisitorNameNullptrAndDescriptor)
359 CheckStridedSliceLayerVisitorNameNullptrAndDescriptor)
361 CheckTransposeLayerVisitorNameNullptrAndDescriptor)
void Slice(const TensorInfo &inputInfo, const SliceDescriptor &descriptor, const void *inputData, void *outputData, unsigned int dataTypeSize)
void Splitter(const SplitterQueueDescriptor &data, std::vector< ITensorHandle *> inputs, std::vector< ITensorHandle *> outputs)
void Stack(const StackQueueDescriptor &data, std::vector< std::unique_ptr< Decoder< float >>> &inputs, Encoder< float > &output, const TensorInfo &inputInfo, const TensorInfo &outputInfo)
#define TEST_CASE_CHECK_LAYER_VISITOR_NAME_AND_DESCRIPTOR(name, testName)
void Fill(Encoder< float > &output, const TensorShape &desiredOutputShape, const float value)
Creates a tensor and fills it with a scalar value.
void Transpose(const armnn::TensorShape &dstShape, const armnn::PermutationVector &mappings, const void *src, void *dst, size_t dataTypeSize)
void DepthToSpace(const TensorInfo &inputInfo, const DepthToSpaceDescriptor &descriptor, const void *inputData, void *outputData, unsigned int dataTypeSize)
void ArgMinMax(Decoder< float > &in, OUT *out, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, ArgMinMaxFunction function, int axis)
#define TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR_AND_DESCRIPTOR(name, testName)
void Permute(const armnn::TensorShape &dstShape, const armnn::PermutationVector &mappings, const void *src, void *dst, size_t dataTypeSize)
void Gather(const TensorInfo ¶msInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo, Decoder< float > ¶ms, const int32_t *indices, Encoder< float > &output, const int32_t axis)
float Activation(float in, ActivationFunction function, float a, float b)
void Pad(const TensorInfo &inputInfo, const TensorInfo &outputInfo, const ITensorHandle *inputHandle, ITensorHandle *outputHandle, const PadQueueDescriptor &data)
void LogSoftmax(Decoder< float > &input, Encoder< float > &output, const TensorInfo &inputInfo, const LogSoftmaxDescriptor &descriptor)
void SpaceToBatchNd(const TensorInfo &inputInfo, const TensorInfo &outputInfo, const SpaceToBatchNdDescriptor ¶ms, Decoder< float > &inputData, Encoder< float > &outputData)
void StridedSlice(const TensorInfo &inputInfo, const StridedSliceDescriptor ¶ms, const void *inputData, void *outputData, unsigned int dataTypeSize)
void SpaceToDepth(const TensorInfo &inputInfo, const TensorInfo &outputInfo, const SpaceToDepthDescriptor ¶ms, Decoder< float > &inputData, Encoder< float > &outputData)
void BatchToSpaceNd(const DataLayoutIndexed &dataLayout, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, const std::vector< unsigned int > &blockShape, const std::vector< std::pair< unsigned int, unsigned int >> &cropsData, Decoder< float > &inputDecoder, Encoder< float > &outputEncoder)
void Pooling2d(Decoder< float > &rInputDecoder, Encoder< float > &rOutputEncoder, const TensorInfo &inputInfo, const TensorInfo &outputInfo, const Pooling2dDescriptor ¶ms)
Computes the Pooling2d operation.
void Softmax(Decoder< float > &in, Encoder< float > &out, const TensorInfo &inputTensorInfo, float beta, int axis)
Computes the softmax function on some inputs, into outputs, with a shape given by tensorInfo...
void Resize(Decoder< float > &in, const TensorInfo &inputInfo, Encoder< float > &out, const TensorInfo &outputInfo, DataLayoutIndexed dataLayout, armnn::ResizeMethod resizeMethod, bool alignCorners, bool halfPixelCenters)