| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
| H A D | Pooling2dEndToEndTestImpl.hpp | 21 armnn::INetworkPtr CreatePooling2dNetwork(const armnn::TensorShape& inputShape, in CreatePooling2dNetwork()
58 const TensorShape& inputShape = { 1, 3, 3, 1 }; in MaxPool2dEndToEnd() local
84 const TensorShape& inputShape = { 1, 3, 3, 1 }; in MaxPool2dEndToEndFloat16() local
107 const TensorShape& inputShape = { 1, 3, 3, 1 }; in AvgPool2dEndToEnd() local
148 const TensorShape& inputShape = { 1, 3, 3, 1 }; in AvgPool2dEndToEndFloat16() local
|
| H A D | SpaceToDepthEndToEndTestImpl.cpp | 25 armnn::INetworkPtr CreateSpaceToDepthNetwork(const armnn::TensorShape& inputShape, in CreateSpaceToDepthNetwork()
104 TensorShape inputShape{1, 2, 2, 1}; in SpaceToDepthNhwcEndToEndTest1() local
135 TensorShape inputShape{1, 2, 2, 1}; in SpaceToDepthNchwEndToEndTest1() local
166 TensorShape inputShape{1, 2, 2, 2}; in SpaceToDepthNhwcEndToEndTest2() local
197 TensorShape inputShape{1, 2, 2, 2}; in SpaceToDepthNchwEndToEndTest2() local
|
| H A D | InstanceNormalizationEndToEndTestImpl.cpp | 25 armnn::INetworkPtr CreateInstanceNormalizationNetwork(const armnn::TensorShape& inputShape, in CreateInstanceNormalizationNetwork()
106 TensorShape inputShape{2, 2, 2, 2}; in InstanceNormalizationNhwcEndToEndTest1() local
176 TensorShape inputShape{2, 2, 2, 2}; in InstanceNormalizationNchwEndToEndTest1() local
247 TensorShape inputShape{2, 2, 2, 2}; in InstanceNormalizationNhwcEndToEndTest2() local
318 TensorShape inputShape{2, 2, 2, 2}; in InstanceNormalizationNchwEndToEndTest2() local
|
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | PackTest.cpp | 21 std::vector<int32_t> inputShape { 3, 2, 3 }; in PackFp32Axis0Test() local
84 std::vector<int32_t> inputShape { 3, 2, 3 }; in PackFp32Axis1Test() local
148 std::vector<int32_t> inputShape { 3, 2, 3 }; in PackFp32Axis2Test() local
210 std::vector<int32_t> inputShape { 3, 2, 3 }; in PackFp32Axis3Test() local
280 std::vector<int32_t> inputShape { 3, 3 }; in PackFp32Inputs3Test() local
|
| H A D | PadTest.cpp | 23 std::vector<int32_t> inputShape { 2, 2, 2 }; in Pad2dTest() local
69 std::vector<int32_t> inputShape { 2, 2, 2 }; in Pad3dTest() local
115 std::vector<int32_t> inputShape { 2, 2, 3, 2 }; in Pad4dTest() local
317 std::vector<int32_t> inputShape { 2, 2, 2 }; in PadInt8Test() local
368 std::vector<int32_t> inputShape { 2, 2, 2 }; in PadUint8Test() local
|
| H A D | SpaceDepthTest.cpp | 21 std::vector<int32_t> inputShape { 1, 2, 2, 4 }; in DepthToSpaceFp32Test() local
47 std::vector<int32_t> inputShape { 2, 1, 1, 4 }; in DepthToSpaceUint8Test() local
69 std::vector<int32_t> inputShape { 1, 2, 2, 2 }; in SpaceToDepthFp32Test() local
88 std::vector<int32_t> inputShape { 1, 2, 2, 1 }; in SpaceToDepthUint8Test() local
|
| H A D | ControlTest.cpp | 21 std::vector<int32_t> inputShape { 2, 2 }; in ConcatUint8TwoInputsTest() local
44 std::vector<int32_t> inputShape { 2, 2 }; in ConcatInt16TwoInputsTest() local
66 std::vector<int32_t> inputShape { 2, 2 }; in ConcatFloat32TwoInputsTest() local
88 std::vector<int32_t> inputShape { 2, 2 }; in ConcatThreeInputsTest() local
112 std::vector<int32_t> inputShape { 1, 2, 2 }; in ConcatAxisTest() local
|
| H A D | FillTest.cpp | 22 std::vector<int32_t> inputShape { 2 }; in Fill2dTest() local
40 std::vector<int32_t> inputShape { 3 }; in Fill3dTest() local
67 std::vector<int32_t> inputShape { 4 }; in Fill4dTest() local
102 std::vector<int32_t> inputShape { 2 }; in FillInt32Test() local
|
| H A D | Convolution3dTest.cpp | 41 std::vector<int32_t> inputShape { 1, 2, 2, 2, 1 }; in Conv3DWithBiasesSimpleWithPaddingFp32Test() local
82 std::vector<int32_t> inputShape { 1, 3, 10, 10, 1 }; in Conv3DWithBiasesStridesFp32Test() local
141 std::vector<int32_t> inputShape { 1, 5, 5, 5, 2 }; in Conv3DWithBiasesDilationFp32Test() local
192 std::vector<int32_t> inputShape { 1, 3, 10, 10, 1 }; in Conv3DFp32SmallTest() local
|
| /aosp_15_r20/external/armnn/src/armnn/test/optimizations/ |
| H A D | ReduceMultipleAxesTests.cpp | 19 TensorShape& inputShape, in CreateSimpleReduceNetwork()
123 TensorShape inputShape = {1, 3, 2, 4}; in ReduceSumWithTwoAxesKeepDimsTest() local
155 TensorShape inputShape = {1, 3, 2, 4}; in ReduceSumWithTwoAxesTest() local
187 TensorShape inputShape = {2, 2, 2, 2}; in ReduceSumWithThreeAxesKeepDimsTest() local
222 TensorShape inputShape = {2, 2, 2, 2}; in ReduceSumWithThreeAxesTest() local
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | MirrorPadTestImpl.cpp | 164 const armnn::TensorShape inputShape{ 3, 3 }; in PadSymmetric2dTest() local
215 const armnn::TensorShape inputShape{ 3, 3 }; in PadReflect2dTest() local
266 const armnn::TensorShape inputShape{ 2, 2, 2 }; in PadSymmetric3dTest() local
333 const armnn::TensorShape inputShape{ 2, 2, 2 }; in PadReflect3dTest() local
400 const armnn::TensorShape inputShape{ 2, 2, 2, 2 }; in PadSymmetric4dTest() local
713 const armnn::TensorShape inputShape{ 2, 2, 2, 2 }; in PadReflect4dTest() local
850 const armnn::TensorShape inputShape{ 3, 3 }; in PadSymmetricFloat16() local
894 const armnn::TensorShape inputShape{ 3, 3 }; in PadReflectFloat16() local
|
| H A D | ReduceSumTestImpl.cpp | 90 const armnn::TensorShape inputShape{ 1, 1, 1, 5 }; in ReduceSumSimpleTest() local
123 const armnn::TensorShape inputShape{ 1, 3, 2, 4 }; in ReduceSumSingleAxisTest1() local
164 const armnn::TensorShape inputShape{ 1, 6, 3, 4 }; in ReduceSumSingleAxisTest2() local
223 const armnn::TensorShape inputShape{ 1, 6, 3, 4 }; in ReduceSumSingleAxisTest3() local
288 const armnn::TensorShape inputShape{ 1, 3, 2, 4 }; in ReduceSumMultipleAxisTest() local
|
| H A D | ReduceProdTestImpl.cpp | 89 const armnn::TensorShape inputShape{ 1, 1, 1, 5 }; in ReduceProdSimpleTest() local
122 const armnn::TensorShape inputShape{ 1, 3, 2, 4 }; in ReduceProdSingleAxisTest1() local
159 const armnn::TensorShape inputShape{ 1, 6, 3, 4 }; in ReduceProdSingleAxisTest2() local
218 const armnn::TensorShape inputShape{ 1, 6, 3, 4 }; in ReduceProdSingleAxisTest3() local
283 const armnn::TensorShape inputShape{ 1, 3, 2, 4 }; in ReduceProdMultipleAxisTest() local
|
| H A D | ReductionTestImpl.cpp | 91 const armnn::TensorShape inputShape{ 1, 1, 2, 3 }; in ReduceMaxSimpleTest() local
131 const armnn::TensorShape inputShape{ 1, 1, 2, 3 }; in ReduceMaxNegativeAxisTest() local
172 const armnn::TensorShape inputShape{ 1, 1, 2, 3 }; in ReduceMaxSimpleTest2() local
214 const armnn::TensorShape inputShape { 1, 1, 2, 3 }; in ReduceMinSimpleTest() local
254 const armnn::TensorShape inputShape{ 1, 1, 2, 3 }; in ReduceMinNegativeAxisTest() local
|
| H A D | ExpTestImpl.cpp | 15 const unsigned int inputShape[] = { 2, 2 }; in Exp2dTest() local
46 const unsigned int inputShape[] = { 3, 1, 2 }; in Exp3dTest() local
79 const unsigned int inputShape[] = { 1, 2 }; in ExpZeroTest() local
108 const unsigned int inputShape[] = { 1, 2 }; in ExpNegativeTest() local
|
| H A D | RsqrtTestImpl.cpp | 16 const unsigned int inputShape[] = { 2, 2 }; in Rsqrt2dTest() local
47 const unsigned int inputShape[] = { 3, 1, 2 }; in Rsqrt3dTest() local
78 const unsigned int inputShape[] = { 1, 2 }; in RsqrtZeroTest() local
107 const unsigned int inputShape[] = { 1, 2 }; in RsqrtNegativeTest() local
|
| H A D | SinTestImpl.cpp | 15 const unsigned int inputShape[] = { 2, 2 }; in Sin2dTest() local
46 const unsigned int inputShape[] = { 3, 1, 2 }; in Sin3dTest() local
79 const unsigned int inputShape[] = { 1, 2 }; in SinZeroTest() local
108 const unsigned int inputShape[] = { 1, 2 }; in SinNegativeTest() local
|
| H A D | SqrtTestImpl.cpp | 16 const unsigned int inputShape[] = { 2, 2 }; in Sqrt2dTest() local
47 const unsigned int inputShape[] = { 3, 1, 2 }; in Sqrt3dTest() local
78 const unsigned int inputShape[] = { 1, 2 }; in SqrtZeroTest() local
107 const unsigned int inputShape[] = { 1, 2 }; in SqrtNegativeTest() local
|
| H A D | LogTestImpl.cpp | 15 const unsigned int inputShape[] = { 2, 2 }; in Log2dTest() local
46 const unsigned int inputShape[] = { 3, 1, 2 }; in Log3dTest() local
79 const unsigned int inputShape[] = { 1, 2 }; in LogZeroTest() local
108 const unsigned int inputShape[] = { 1, 2 }; in LogNegativeTest() local
|
| H A D | NegTestImpl.cpp | 16 const unsigned int inputShape[] = { 2, 2 }; in Neg2dTest() local
47 const unsigned int inputShape[] = { 3, 1, 2 }; in Neg3dTest() local
78 const unsigned int inputShape[] = { 1, 2 }; in NegZeroTest() local
107 const unsigned int inputShape[] = { 1, 2 }; in NegNegativeTest() local
|
| /aosp_15_r20/external/armnn/src/armnn/test/ |
| H A D | InferOutputTests.hpp | 238 const armnn::TensorShape& inputShape, in CreatePreluLayerHelper()
454 armnn::TensorShape inputShape(4, inputSize.data()); in Convolution2dInferOutputShapeTest() local
491 armnn::TensorShape inputShape(5, inputSize.data()); in Convolution3dInferOutputShapeTest() local
522 armnn::TensorShape inputShape(4, inputSize.data()); in TransposeConvolution2dInferOutputShapeTest() local
555 armnn::TensorShape inputShape(4, inputSize.data()); in DepthwiseConvolution2dInferOutputShapeTest() local
593 armnn::TensorShape inputShape(5, inputSize.data()); in Pooling3dInferOutputShapeTest() local
620 const std::vector<unsigned int> inputShape{ 2, 5 }; in QLstmInferOutputShapeTest() local
671 const std::vector<unsigned int> inputShape{ 2, 5 }; in QuantizedLstmInferOutputShapeTest() local
|
| H A D | ShapeInferenceTests.cpp | 36 for (auto inputShape : inputShapes) in BuildGraph() local
285 const TensorShape inputShape{1, 1, 10, 10}; variable
441 const TensorShape inputShape{2, 5}; variable
530 const TensorShape inputShape{2, 5}; variable
564 const TensorShape inputShape{2, 5}; variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Conv2D.cpp | 105 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
191 bool convNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in convNhwc()
220 bool convNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in convNhwc()
274 bool convNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in convNhwc()
303 bool convNhwc(const _Float16* inputData, const Shape& inputShape, const _Float16* filterData, in convNhwc()
330 bool conv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in conv()
349 bool convQuant8PerChannelNhwc(const uint8_t* inputData, const Shape& inputShape, in convQuant8PerChannelNhwc()
442 bool convQuant8PerChannelNhwc(const int8_t* inputData, const Shape& inputShape, in convQuant8PerChannelNhwc()
506 bool convQuant8PerChannel(const T* inputData, const Shape& inputShape, const int8_t* filterData, in convQuant8PerChannel()
|
| D | DepthwiseConv2D.cpp | 92 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
129 bool depthwiseConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvNhwc()
163 bool depthwiseConvNhwc(const _Float16* inputData, const Shape& inputShape, in depthwiseConvNhwc()
189 bool depthwiseConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in depthwiseConvNhwc()
240 bool depthwiseConvNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in depthwiseConvNhwc()
273 const T* inputData, const Shape& inputShape, const int8_t* filterData, in depthwiseConvQuant8PerChannelNhwc()
369 bool depthwiseConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in depthwiseConv()
390 bool depthwiseConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in depthwiseConvQuant8PerChannel()
|
| D | Softmax.cpp | 50 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta, in softmaxSlowFloat32()
82 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxFloat32()
98 bool softmaxFloat16(const _Float16* inputData, const Shape& inputShape, const float beta, in softmaxFloat16()
113 bool softmaxQuant8Impl(const T* inputData, const Shape& inputShape, const float /*beta*/, in softmaxQuant8Impl()
200 bool softmaxQuant8(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8()
|