Home
last modified time | relevance | path

Searched defs:inputShape (Results 1 – 25 of 112) sorted by relevance

12345

/external/armnn/src/armnnUtils/test/
DTensorUtilsTest.cpp19 armnn::TensorShape inputShape({ 2, 3, 4 }); in BOOST_AUTO_TEST_CASE() local
32 armnn::TensorShape inputShape({ 2, 3, 4 }); in BOOST_AUTO_TEST_CASE() local
45 armnn::TensorShape inputShape({ 2, 3, 4 }); in BOOST_AUTO_TEST_CASE() local
58 armnn::TensorShape inputShape({ 2, 3, 4 }); in BOOST_AUTO_TEST_CASE() local
71 armnn::TensorShape inputShape({ 2, 3, 4 }); in BOOST_AUTO_TEST_CASE() local
84 armnn::TensorShape inputShape({ 2, 3, 4 }); in BOOST_AUTO_TEST_CASE() local
97 armnn::TensorShape inputShape({ 2, 3, 4 }); in BOOST_AUTO_TEST_CASE() local
110 armnn::TensorShape inputShape({ 2, 3, 4 }); in BOOST_AUTO_TEST_CASE() local
123 armnn::TensorShape inputShape({ 2, 3, 4 }); in BOOST_AUTO_TEST_CASE() local
131 armnn::TensorShape inputShape({ 2, 3, 4 }); in BOOST_AUTO_TEST_CASE() local
/external/armnn/delegate/src/test/
DPooling2dTest.cpp25 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in MaxPool2dFP32PaddingValidTest() local
51 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in MaxPool2dInt8PaddingValidTest() local
80 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in MaxPool2dFP32PaddingSameTest() local
106 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in MaxPool2dInt8PaddingSameTest() local
135 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in MaxPool2dFP32ReluTest() local
162 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in MaxPool2dInt8ReluTest() local
191 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in MaxPool2dFP32Relu6Test() local
218 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in MaxPool2dInt8Relu6Test() local
247 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in MaxPool2dUint8PaddingSameTest() local
276 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in MaxPool2dUint8ReluTest() local
[all …]
DQuantizationTest.cpp21 std::vector<int32_t> inputShape { 2, 4 }; in DequantizeUint8Test() local
48 std::vector<int32_t> inputShape { 2, 4 }; in DequantizeInt8Test() local
74 std::vector<int32_t> inputShape { 2, 5 }; in DequantizeInt16Test() local
101 std::vector<int32_t> inputShape { 2, 4 }; in QuantizeFloat32Uint8Test() local
128 std::vector<int32_t> inputShape { 2, 4 }; in QuantizeFloat32Int8Test() local
154 std::vector<int32_t> inputShape { 2, 4 }; in QuantizeFloat32Int16Test() local
180 std::vector<int32_t> inputShape { 2, 4 }; in QuantizeInt16Int16Test() local
206 std::vector<int32_t> inputShape { 2, 4 }; in QuantizeInt16Int8Test() local
232 std::vector<int32_t> inputShape { 2, 4 }; in QuantizeInt8Uint8Test() local
258 std::vector<int32_t> inputShape { 2, 4 }; in QuantizeUint8Int8Test() local
DReshapeTest.cpp21 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in ReshapeSimpleTest() local
47 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in ReshapeReduceDimTest() local
73 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in ReshapeFlattenTest() local
99 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in ReshapeFlattenAllTest() local
125 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in ReshapeInt8Test() local
153 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in ReshapeUint8Test() local
181 std::vector<int32_t> inputShape { 1, 3, 4, 1 }; in ReshapeInt16Test() local
DConvolution2dTest.cpp25 std::vector<int32_t> inputShape { 1, 5, 5, 1 }; in Conv2DWithBiasesFp32Test() local
79 std::vector<int32_t> inputShape { 1, 2, 2, 1 }; in Conv2DWithBiasesInt8Test() local
122 std::vector<int32_t> inputShape { 1, 2, 2, 1 }; in Conv2DWithBiasesReluUint8Test() local
178 std::vector<int32_t> inputShape { 1, 2, 2, 1 }; in Conv2DWithBiasesRelu6Uint8Test() local
279 std::vector<int32_t> inputShape { 1, 2, 2, 1 }; in TransposeConvInt8Test() local
312 std::vector<int32_t> inputShape { 1, 2, 2, 1 }; in TransposeConvFp32Test() local
DControlTest.cpp21 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
/external/armnn/src/backends/backendsCommon/test/
DArgMinMaxEndToEndTestImpl.hpp39 void ArgMinMaxEndToEndImpl(const armnn::TensorShape& inputShape, in ArgMinMaxEndToEndImpl()
70 const armnn::TensorShape inputShape{ 1, 1, 1, 5 }; in ArgMaxEndToEndSimple() local
88 const armnn::TensorShape inputShape{ 1, 1, 1, 5 }; in ArgMinEndToEndSimple() local
106 const armnn::TensorShape inputShape{ 3, 2, 1, 4 }; in ArgMaxAxis0EndToEnd() local
131 const armnn::TensorShape inputShape{ 3, 2, 1, 4 }; in ArgMinAxis0EndToEnd() local
156 const armnn::TensorShape inputShape{ 1, 3, 2, 4 }; in ArgMaxAxis1EndToEnd() local
181 const armnn::TensorShape inputShape{ 1, 3, 2, 4 }; in ArgMinAxis1EndToEnd() local
206 const armnn::TensorShape inputShape{ 1, 3, 2, 4 }; in ArgMaxAxis2EndToEnd() local
232 const armnn::TensorShape inputShape{ 1, 3, 2, 4 }; in ArgMinAxis2EndToEnd() local
258 const armnn::TensorShape inputShape{ 1, 3, 2, 4 }; in ArgMaxAxis3EndToEnd() local
[all …]
DSplitterEndToEndTestImpl.hpp23 INetworkPtr CreateSplitterNetwork(const TensorShape& inputShape, in CreateSplitterNetwork()
83 const TensorShape& inputShape = { 4 }; in Splitter1dEndToEnd() local
111 const TensorShape& inputShape = { 4, 3 }; in Splitter2dDim0EndToEnd() local
146 const TensorShape& inputShape = { 4, 3 }; in Splitter2dDim1EndToEnd() local
184 const TensorShape& inputShape = { 2, 4, 3 }; in Splitter3dDim0EndToEnd() local
232 const TensorShape& inputShape = { 2, 4, 3 }; in Splitter3dDim1EndToEnd() local
280 const TensorShape& inputShape = { 2, 4, 3 }; in Splitter3dDim2EndToEnd() local
320 const TensorShape& inputShape = { 4, 3, 2, 2 }; in Splitter4dDim0EndToEnd() local
400 const TensorShape& inputShape = { 2, 6, 2, 2 }; in Splitter4dDim1EndToEnd() local
480 const TensorShape& inputShape = { 2, 3, 4, 2 }; in Splitter4dDim2EndToEnd() local
[all …]
DSpaceToDepthEndToEndTestImpl.cpp25 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
DInstanceNormalizationEndToEndTestImpl.cpp25 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
DBatchToSpaceNdEndToEndTestImpl.hpp21 INetworkPtr CreateBatchToSpaceNdNetwork(const armnn::TensorShape& inputShape, in CreateBatchToSpaceNdNetwork()
57 const TensorShape& inputShape = { 4, 1, 1, 1 }; in BatchToSpaceNdEndToEnd() local
86 const TensorShape& inputShape = (dataLayout == DataLayout::NCHW) in BatchToSpaceNdComplexEndToEnd() local
/external/armnn/src/backends/backendsCommon/test/layerTests/
DBatchToSpaceNdTestImpl.hpp34 const unsigned int *inputShape, in BatchToSpaceNdHelper()
94 const unsigned int inputShape[] = {4, 2, 2, 1}; in BatchToSpaceNdNhwcTest1() local
142 const unsigned int inputShape[] = {4, 1, 1, 1}; in BatchToSpaceNdNhwcTest2() local
166 const unsigned int inputShape[] = {4, 1, 1, 3}; in BatchToSpaceNdNhwcTest3() local
187 const unsigned int inputShape[] = {8, 1, 3, 1}; in BatchToSpaceNdNhwcTest4() local
222 const unsigned int inputShape[] = {4, 2, 2, 1}; in BatchToSpaceNdNhwcTest5() local
242 const unsigned int inputShape[] = {4, 1, 1, 1}; in BatchToSpaceNdNhwcTest6() local
266 const unsigned int inputShape[] = {4, 1, 1, 3}; in BatchToSpaceNdNhwcTest7() local
287 const unsigned int inputShape[] = {4, 3, 1, 1}; in BatchToSpaceNdNchwTest1() local
320 const unsigned int inputShape[] = {4, 1, 1, 1}; in BatchToSpaceNdNchwTest2() local
[all …]
DMeanTestImpl.hpp23 const unsigned int* inputShape, in MeanTestHelper()
82 const unsigned int inputShape[] = { 3, 2 }; in MeanSimpleTest() local
98 const unsigned int inputShape[] = { 2, 3, 1, 2 }; in MeanSimpleAxisTest() local
114 const unsigned int inputShape[] = { 1, 1, 3, 2 }; in MeanKeepDimsTest() local
130 const unsigned int inputShape[] = { 2, 3, 1, 2 }; in MeanMultipleDimsTest() local
147 const unsigned int inputShape[] = { 4, 3, 2 }; in MeanVts1Test() local
165 const unsigned int inputShape[] = { 4, 3, 2 }; in MeanVts2Test() local
183 const unsigned int inputShape[] = { 1, 2, 2, 1 }; in MeanVts3Test() local
DExpTestImpl.cpp15 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
DNegTestImpl.cpp16 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
DRsqrtTestImpl.cpp16 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
DArgMinMaxTestImpl.cpp69 const armnn::TensorShape inputShape{ 1, 1, 1, 5 }; in ArgMaxSimpleTest() local
97 const armnn::TensorShape inputShape{ 1, 1, 1, 5 }; in ArgMinSimpleTest() local
125 const armnn::TensorShape inputShape{ 1, 3, 2, 4}; in ArgMinChannelTest() local
161 const armnn::TensorShape inputShape{ 1, 3, 2, 4}; in ArgMaxChannelTest() local
197 const armnn::TensorShape inputShape{ 1, 3, 2, 4}; in ArgMaxHeightTest() local
233 const armnn::TensorShape inputShape{ 1, 3, 2, 4}; in ArgMinWidthTest() local
DAbsTestImpl.cpp16 const unsigned int inputShape[] = { 2, 2 }; in Abs2dTest() local
49 const unsigned int inputShape[] = { 3, 1, 2 }; in Abs3dTest() local
81 const unsigned int inputShape[] = { 1, 2 }; in AbsZeroTest() local
DStridedSliceTestImpl.cpp83 unsigned int inputShape[] = {3, 2, 3, 1}; in StridedSlice4dTest() local
122 unsigned int inputShape[] = {3, 2, 3, 1}; in StridedSlice4dReverseTest() local
161 unsigned int inputShape[] = {3, 2, 3, 1}; in StridedSliceSimpleStrideTest() local
202 unsigned int inputShape[] = {3, 2, 3, 1}; in StridedSliceSimpleRangeMaskTest() local
247 unsigned int inputShape[] = {3, 2, 3, 1}; in StridedSliceShrinkAxisMaskTest() local
288 unsigned int inputShape[] = {3, 2, 3, 1}; in StridedSliceShrinkAxisMaskBitPosition0Test() local
329 unsigned int inputShape[] = {3, 2, 3, 1}; in StridedSliceShrinkAxisMaskBitPosition1Test() local
370 unsigned int inputShape[] = {3, 2, 3, 1}; in StridedSliceShrinkAxisMaskBitPosition2Test() local
411 unsigned int inputShape[] = {3, 2, 3, 1}; in StridedSliceShrinkAxisMaskBitPosition3Test() local
456 unsigned int inputShape[] = {3, 2, 3, 1}; in StridedSliceShrinkAxisMaskBitPosition0And1Test() local
[all …]
DSoftmaxTestImpl.cpp32 const armnn::TensorShape inputShape{ 1, 8, 1 }; member
43 const armnn::TensorShape inputShape{ 1, 8, 1, 1 }; member
64 const armnn::TensorShape& inputShape, in SimpleSoftmaxBaseTestImpl()
127 const armnn::TensorShape inputShape{ 2, 4 }; in SimpleSoftmaxTestImpl() local
157 armnn::TensorShape inputShape; in SimpleSoftmaxTestImpl() local
211 const armnn::TensorShape& inputShape, in Simple3dSoftmaxTestImpl()
226 const armnn::TensorShape& inputShape, in Simple4dSoftmaxTestImpl()
251 unsigned int inputShape[] = { batchSize, channels }; in CompareSoftmaxTestImpl() local
348 armnn::TensorShape inputShape; in Simple3dAxisSoftmaxTest() local
454 armnn::TensorShape inputShape; in Simple4dAxisSoftmaxTest() local
DDepthToSpaceTestImpl.cpp84 unsigned int inputShape[] = { 1, 1, 1, 8 }; in DepthToSpaceTest1() local
113 unsigned int inputShape[] = { 1, 2, 2, 4 }; in DepthToSpaceTest2() local
163 unsigned int inputShape[] = { 2, 1, 1, 4 }; in DepthToSpaceTest3() local
190 unsigned int inputShape[] = { 2, 2, 2, 4 }; in DepthToSpaceTest4() local
DPermuteTestImpl.hpp67 unsigned int inputShape[] = { 1, 2, 2, 2 }; in SimplePermuteTest() local
117 unsigned int inputShape[] = { 1, 2, 2, 3 }; in PermuteValueSet1Test() local
168 unsigned int inputShape[] = { 1, 3, 2, 2 }; in PermuteValueSet2Test() local
219 unsigned int inputShape[] = { 1, 2, 3, 3 }; in PermuteValueSet3Test() local
DTransposeTestImpl.hpp68 unsigned int inputShape[] = { 1, 2, 2, 2 }; in SimpleTransposeTest() local
118 unsigned int inputShape[] = { 1, 2, 2, 3 }; in TransposeValueSet1Test() local
169 unsigned int inputShape[] = { 1, 3, 2, 2 }; in TransposeValueSet2Test() local
220 unsigned int inputShape[] = { 1, 2, 3, 3 }; in TransposeValueSet3Test() local
/external/armnn/src/armnn/test/
DQuantizerTest.cpp44 TestQuantization(const TensorShape& inputShape, const TensorShape& outputShape) in TestQuantization()
50 …TestQuantization(const QuantizerOptions& options, const TensorShape& inputShape, const TensorShape… in TestQuantization()
203 TestAdditionQuantization(const TensorShape& inputShape, const TensorShape& outputShape) in TestAdditionQuantization()
207 const TensorShape& inputShape, in TestAdditionQuantization()
272 TestActivationQuantization(const TensorShape& inputShape, const TensorShape& outputShape) in TestActivationQuantization()
276 const TensorShape& inputShape, in TestActivationQuantization()
533 …TestBoundedReluActivationQuantization(const TensorShape& inputShape, const TensorShape& outputShap… in BOOST_AUTO_TEST_CASE()
537 const TensorShape& inputShape, in BOOST_AUTO_TEST_CASE()
589 … TestTanHActivationQuantization(const TensorShape& inputShape, const TensorShape& outputShape) in BOOST_AUTO_TEST_CASE()
593 const TensorShape& inputShape, in BOOST_AUTO_TEST_CASE()
[all …]
/external/android-nn-driver/1.0/
DFullyConnected.hpp15 inline armnn::TensorShape FlattenFullyConnectedInput(const armnn::TensorShape& inputShape, in FlattenFullyConnectedInput()
38 inline bool VerifyFullyConnectedShapes(const armnn::TensorShape& inputShape, in VerifyFullyConnectedShapes()

12345