Home
last modified time | relevance | path

Searched defs:outputShape (Results 1 – 25 of 221) sorted by relevance

123456789

/external/armnn/src/armnnUtils/test/
DTensorUtilsTest.cpp22 armnn::TensorShape outputShape = ExpandDims(inputShape, 0); variable
35 armnn::TensorShape outputShape = ExpandDims(inputShape, 1); variable
48 armnn::TensorShape outputShape = ExpandDims(inputShape, 2); variable
61 armnn::TensorShape outputShape = ExpandDims(inputShape, 3); variable
74 armnn::TensorShape outputShape = ExpandDims(inputShape, -1); variable
87 armnn::TensorShape outputShape = ExpandDims(inputShape, -2); variable
100 armnn::TensorShape outputShape = ExpandDims(inputShape, -3); variable
113 armnn::TensorShape outputShape = ExpandDims(inputShape, -4); variable
142 armnn::TensorShape outputShape = ExpandDimsToRank(inputShape, 4); variable
155 armnn::TensorShape outputShape = ExpandDimsToRank(inputShape, 4); variable
[all …]
/external/armnn/delegate/test/
DPooling2dTest.cpp26 std::vector<int32_t> outputShape { 1, 1, 2, 1 }; in MaxPool2dFP32PaddingValidTest() local
52 std::vector<int32_t> outputShape { 1, 1, 2, 1 }; in MaxPool2dInt8PaddingValidTest() local
81 std::vector<int32_t> outputShape { 1, 2, 2, 1 }; in MaxPool2dFP32PaddingSameTest() local
107 std::vector<int32_t> outputShape { 1, 2, 2, 1 }; in MaxPool2dInt8PaddingSameTest() local
136 std::vector<int32_t> outputShape { 1, 2, 3, 1 }; in MaxPool2dFP32ReluTest() local
163 std::vector<int32_t> outputShape { 1, 2, 3, 1 }; in MaxPool2dInt8ReluTest() local
192 std::vector<int32_t> outputShape { 1, 2, 2, 1 }; in MaxPool2dFP32Relu6Test() local
219 std::vector<int32_t> outputShape { 1, 2, 2, 1 }; in MaxPool2dInt8Relu6Test() local
248 std::vector<int32_t> outputShape { 1, 2, 2, 1 }; in MaxPool2dUint8PaddingSameTest() local
277 std::vector<int32_t> outputShape { 1, 2, 3, 1 }; in MaxPool2dUint8ReluTest() local
[all …]
DBatchMatMulTest.cpp23 std::vector<int32_t> outputShape { 2, 2 }; in BatchMatMul2DFp32SimpleTest() local
51 std::vector<int32_t> outputShape { 2, 2 }; in BatchMatMul2DInt8SimpleTest() local
80 std::vector<int32_t> outputShape { 1,2,2 }; in BatchMatMul3DFp32SimpleTest() local
109 std::vector<int32_t> outputShape { 1,2,2 }; in BatchMatMul3DInt8SimpleTest() local
138 std::vector<int32_t> outputShape { 1,1,2,2 }; in BatchMatMul4DFp32SimpleTest() local
167 std::vector<int32_t> outputShape { 1,1,2,2 }; in BatchMatMul4DInt8SimpleTest() local
196 std::vector<int32_t> outputShape { 2,2,2 }; in BatchMatMul3DFp32BatchTest() local
234 std::vector<int32_t> outputShape { 2,2,2 }; in BatchMatMul3DInt8BatchTest() local
272 std::vector<int32_t> outputShape { 2,2,2 }; in BatchMatMul3DFp32BroadcastTest() local
307 std::vector<int32_t> outputShape { 2,2,2 }; in BatchMatMul3DInt8BroadcastTest() local
[all …]
DQuantizationTest.cpp22 std::vector<int32_t> outputShape { 2, 4 }; in DequantizeUint8Test() local
49 std::vector<int32_t> outputShape { 2, 4 }; in DequantizeInt8Test() local
75 std::vector<int32_t> outputShape { 2, 5 }; in DequantizeInt16Test() local
102 std::vector<int32_t> outputShape { 2, 4 }; in QuantizeFloat32Uint8Test() local
129 std::vector<int32_t> outputShape { 2, 4 }; in QuantizeFloat32Int8Test() local
155 std::vector<int32_t> outputShape { 2, 4 }; in QuantizeFloat32Int16Test() local
181 std::vector<int32_t> outputShape { 2, 4 }; in QuantizeInt16Int16Test() local
207 std::vector<int32_t> outputShape { 2, 4 }; in QuantizeInt16Int8Test() local
233 std::vector<int32_t> outputShape { 2, 4 }; in QuantizeInt8Uint8Test() local
259 std::vector<int32_t> outputShape { 2, 4 }; in QuantizeUint8Int8Test() local
DMirrorPadTest.cpp22 std::vector<int32_t> outputShape { 7, 7 }; in MirrorPadSymmetric2dTest() local
64 std::vector<int32_t> outputShape { 7, 7 }; in MirrorPadReflect2dTest() local
106 std::vector<int32_t> outputShape { 4, 4, 4 }; in MirrorPadSymmetric3dTest() local
164 std::vector<int32_t> outputShape { 4, 4, 4 }; in MirrorPadReflect3dTest() local
222 std::vector<int32_t> outputShape { 5, 7 }; in MirrorPadSymmetricUint8Test() local
262 std::vector<int32_t> outputShape { 7, 5 }; in MirrorPadReflectInt8Test() local
DPooling3dTest.cpp29 std::vector<int32_t> outputShape = { 1, 1, 2, 3, 1 }; in MaxPool3dFP32PaddingValidTest() local
62 std::vector<int32_t> outputShape = { 1, 2, 3, 4, 1 }; in MaxPool3dFP32PaddingSameTest() local
95 std::vector<int32_t> outputShape = { 1, 1, 3, 3, 1 }; in MaxPool3dFP32H1Test() local
128 std::vector<int32_t> outputShape = { 1, 2, 3, 4, 1 }; in MaxPool3dFP32Test() local
161 std::vector<int32_t> outputShape = { 1, 1, 2, 3, 1 }; in AveragePool3dFP32PaddingValidTest() local
194 std::vector<int32_t> outputShape = { 4, 2, 3, 1, 1 }; in AveragePool3dFP32PaddingSameTest() local
227 std::vector<int32_t> outputShape = { 1, 1, 2, 2, 1 }; in AveragePool3dFP32H1Test() local
260 std::vector<int32_t> outputShape = { 1, 2, 2, 4, 1 }; in AveragePool3dFP32Test() local
DReshapeTest.cpp26 std::vector<int32_t> outputShape { 1, 3, 2, 2 }; in ReshapeSimpleTest() local
54 std::vector<int32_t> outputShape { 1, 3, 2, 2 }; in ReshapeSimpleFloat16Test() local
80 std::vector<int32_t> outputShape { 1, 4, 3 }; in ReshapeReduceDimTest() local
106 std::vector<int32_t> outputShape { 6, 2 }; in ReshapeFlattenTest() local
132 std::vector<int32_t> outputShape { 12 }; in ReshapeFlattenAllTest() local
158 std::vector<int32_t> outputShape { 6, 2 }; in ReshapeInt8Test() local
186 std::vector<int32_t> outputShape { 6, 2 }; in ReshapeUint8Test() local
214 std::vector<int32_t> outputShape { 6, 2 }; in ReshapeInt16Test() local
DSpaceDepthTest.cpp22 std::vector<int32_t> outputShape { 1, 4, 4, 1 }; in DepthToSpaceFp32Test() local
48 std::vector<int32_t> outputShape { 2, 2, 2, 1 }; in DepthToSpaceUint8Test() local
70 std::vector<int32_t> outputShape { 1, 1, 1, 8 }; in SpaceToDepthFp32Test() local
89 std::vector<int32_t> outputShape { 1, 1, 1, 4 }; in SpaceToDepthUint8Test() local
DConvolution2dTest.cpp28 std::vector<int32_t> outputShape { 1, 3, 3, 1 }; in Conv2DWithBiasesFp32Test() local
82 std::vector<int32_t> outputShape { 1, 2, 2, 1 }; in Conv2DWithBiasesInt8Test() local
125 std::vector<int32_t> outputShape { 1, 2, 2, 1 }; in Conv2DWithBiasesReluUint8Test() local
183 std::vector<int32_t> outputShape { 1, 2, 2, 1 }; in Conv2DWithBiasesRelu6Uint8Test() local
232 std::vector<int32_t> outputShape { 1,4,4,4 }; in Conv2DPerChannelInt8Test() local
377 std::vector<int32_t> outputShape { 1, 3, 3, 1 }; in TransposeConvInt8Test() local
410 std::vector<int32_t> outputShape { 1, 3, 3, 1 }; in TransposeConvFp32Test() local
/external/armnn/src/backends/tosaCommon/test/
DOneToOneMappingTests.cpp19 std::vector<std::vector<int32_t>> outputShape = {{ 1, 2, 4, 2 }}; variable
53 std::vector<std::vector<int32_t>> outputShape = {{ 1, 2, 4, 2 }}; variable
79 std::vector<std::vector<int32_t>> outputShape = { { 2, 6, 2, 2 } }; variable
125 std::vector<std::vector<int32_t>> outputShape = { { 2, 6, 2, 2 } }; variable
140 std::vector<std::vector<int32_t>> outputShape = {{ 1, 2, 4, 2 }}; variable
157 std::vector<std::vector<int32_t>> outputShape = {{ 1, 2, 4, 2 }}; variable
193 std::vector<std::vector<int32_t>> outputShape = {{ 1, 3, 3, 1 }}; variable
229 std::vector<std::vector<int32_t>> outputShape = {{ 1, 3, 3, 1 }}; variable
262 std::vector<std::vector<int32_t>> outputShape = {{ 2, 2 }}; variable
295 std::vector<std::vector<int32_t>> outputShape = {{ 2, 2 }}; variable
[all …]
/external/armnn/src/backends/backendsCommon/test/
DArgMinMaxEndToEndTestImpl.hpp40 const armnn::TensorShape& outputShape, in ArgMinMaxEndToEndImpl()
71 const armnn::TensorShape outputShape{ 1, 1, 1 }; in ArgMaxEndToEndSimple() local
89 const armnn::TensorShape outputShape{ 1, 1, 1 }; in ArgMinEndToEndSimple() local
107 const armnn::TensorShape outputShape{ 2, 1, 4 }; in ArgMaxAxis0EndToEnd() local
132 const armnn::TensorShape outputShape{ 2, 1, 4 }; in ArgMinAxis0EndToEnd() local
157 const armnn::TensorShape outputShape{ 1, 2, 4 }; in ArgMaxAxis1EndToEnd() local
182 const armnn::TensorShape outputShape{ 1, 2, 4 }; in ArgMinAxis1EndToEnd() local
207 const armnn::TensorShape outputShape{ 1, 3, 4 }; in ArgMaxAxis2EndToEnd() local
233 const armnn::TensorShape outputShape{ 1, 3, 4 }; in ArgMinAxis2EndToEnd() local
259 const armnn::TensorShape outputShape{ 1, 3, 2 }; in ArgMaxAxis3EndToEnd() local
[all …]
DConcatEndToEndTestImpl.hpp24 const TensorShape &outputShape, in CreateConcatNetwork()
62 const TensorShape& outputShape = { 4, 3, 2, 2 }; in ConcatDim0EndToEnd() local
126 const TensorShape& outputShape = { 2, 6, 2, 2 }; in ConcatDim1EndToEnd() local
188 const TensorShape& outputShape = { 2, 3, 4, 2 }; in ConcatDim2EndToEnd() local
249 const TensorShape& outputShape = { 2, 3, 2, 4 }; in ConcatDim3EndToEnd() local
DPooling2dEndToEndTestImpl.hpp22 const armnn::TensorShape& outputShape, in CreatePooling2dNetwork()
59 const TensorShape& outputShape = { 1, 3, 3, 1 }; in MaxPool2dEndToEnd() local
85 const TensorShape& outputShape = { 1, 3, 3, 1 }; in MaxPool2dEndToEndFloat16() local
108 const TensorShape& outputShape = { 1, 3, 3, 1 }; in AvgPool2dEndToEnd() local
149 const TensorShape& outputShape = { 1, 3, 3, 1 }; in AvgPool2dEndToEndFloat16() local
DSpaceToDepthEndToEndTestImpl.cpp26 const armnn::TensorShape& outputShape, in CreateSpaceToDepthNetwork()
107 TensorShape outputShape{1, 1, 1, 4}; in SpaceToDepthNhwcEndToEndTest1() local
138 TensorShape outputShape{1, 1, 1, 4}; in SpaceToDepthNchwEndToEndTest1() local
167 TensorShape outputShape{1, 1, 1, 8}; in SpaceToDepthNhwcEndToEndTest2() local
198 TensorShape outputShape{1, 1, 1, 8}; in SpaceToDepthNchwEndToEndTest2() local
DWorkloadDataValidation.cpp25 TensorShape outputShape { 1, 3, 2, 2 }; variable
63 unsigned int outputShape[] = {2, 3, 4, 5}; variable
84 unsigned int outputShape[] = {2, 3, 4, 5, 6}; variable
115 unsigned int outputShape[] = { outputNum, outputChannels, outputHeight, outputWidth }; variable
149 unsigned int outputShape[] = { outputNum, outputChannels, outputHeight, outputWidth }; variable
191 unsigned int outputShape[] = {outputNum, outputChannels, outputHeight, outputWidth}; variable
237 unsigned int outputShape[] = {outputNum, outputChannels, outputHeight, outputWidth}; variable
294 unsigned int outputShape[] = {outputNum, outputChannels, outputHeight, outputWidth}; variable
447 unsigned int outputShape[dimensionCount]; variable
477 unsigned int outputShape[] = { 1, 1, 1, 2 }; variable
[all …]
/external/pytorch/torch/csrc/jit/tensorexpr/
Dlowerings.cpp59 at::Device device) { in nnc_lowerings_lazy_registration()
109 const std::vector<ExprHandle>& outputShape, \ in nnc_lowerings_lazy_registration()
156 const std::vector<ExprHandle>& outputShape, \ in nnc_lowerings_lazy_registration()
246 const std::vector<ExprHandle>& outputShape, in nnc_lowerings_lazy_registration()
265 const std::vector<ExprHandle>& outputShape, in nnc_lowerings_lazy_registration()
284 const std::vector<ExprHandle>& outputShape, in nnc_lowerings_lazy_registration()
303 const std::vector<ExprHandle>& outputShape, in nnc_lowerings_lazy_registration()
322 const std::vector<ExprHandle>& outputShape, in nnc_lowerings_lazy_registration()
341 const std::vector<ExprHandle>& outputShape, in nnc_lowerings_lazy_registration()
360 const std::vector<ExprHandle>& outputShape, in nnc_lowerings_lazy_registration()
[all …]
/external/armnn/src/backends/backendsCommon/test/layerTests/
DBatchToSpaceNdTestImpl.hpp35 const unsigned int *outputShape, in BatchToSpaceNdHelper()
95 const unsigned int outputShape[] = {1, 4, 4, 1}; in BatchToSpaceNdNhwcTest1() local
143 const unsigned int outputShape[] = {1, 2, 2, 1}; in BatchToSpaceNdNhwcTest2() local
167 const unsigned int outputShape[] = {1, 2, 2, 3}; in BatchToSpaceNdNhwcTest3() local
188 const unsigned int outputShape[] = {2, 2, 4, 1}; in BatchToSpaceNdNhwcTest4() local
223 const unsigned int outputShape[] = {1, 4, 4, 1}; in BatchToSpaceNdNhwcTest5() local
243 const unsigned int outputShape[] = {1, 2, 2, 1}; in BatchToSpaceNdNhwcTest6() local
267 const unsigned int outputShape[] = {1, 2, 2, 3}; in BatchToSpaceNdNhwcTest7() local
288 const unsigned int outputShape[] = {1, 3, 2, 2}; in BatchToSpaceNdNchwTest1() local
321 const unsigned int outputShape[] = {1, 1, 2, 2}; in BatchToSpaceNdNchwTest2() local
[all …]
DMeanTestImpl.hpp27 const unsigned int* outputShape, in MeanTestHelper()
85 const unsigned int outputShape[] = { 1 }; in MeanSimpleTest() local
101 const unsigned int outputShape[] = { 3, 1, 2 }; in MeanSimpleAxisTest() local
117 const unsigned int outputShape[] = { 1, 1, 1, 2 }; in MeanKeepDimsTest() local
133 const unsigned int outputShape[] = { 1, 3, 1, 1 }; in MeanMultipleDimsTest() local
150 const unsigned int outputShape[] = { 2 }; in MeanVts1Test() local
168 const unsigned int outputShape[] = { 1, 3, 1 }; in MeanVts2Test() local
186 const unsigned int outputShape[] = { 1, 2, 1 }; in MeanVts3Test() local
DArgMinMaxTestImpl.cpp72 const armnn::TensorShape outputShape{ 1, 1, 1 }; in ArgMaxSimpleTest() local
100 const armnn::TensorShape outputShape{ 1, 1, 1 }; in ArgMinSimpleTest() local
128 const armnn::TensorShape outputShape{ 1, 2, 4 }; in ArgMinChannelTest() local
164 const armnn::TensorShape outputShape{ 1, 2, 4 }; in ArgMaxChannelTest() local
200 const armnn::TensorShape outputShape{ 1, 3, 4 }; in ArgMaxHeightTest() local
236 const armnn::TensorShape outputShape{ 1, 3, 2 }; in ArgMinWidthTest() local
DReduceProdTestImpl.cpp90 const armnn::TensorShape outputShape{ 1, 1, 1, 1 }; in ReduceProdSimpleTest() local
123 const armnn::TensorShape outputShape{ 1, 1, 2, 4 }; in ReduceProdSingleAxisTest1() local
160 const armnn::TensorShape outputShape{ 1, 1, 3, 4}; in ReduceProdSingleAxisTest2() local
219 const armnn::TensorShape outputShape{ 1, 6, 3, 1 }; in ReduceProdSingleAxisTest3() local
284 const armnn::TensorShape outputShape{ 1, 1, 1, 4 }; in ReduceProdMultipleAxisTest() local
/external/armnn/src/armnn/test/optimizations/
DReduceMultipleAxesTests.cpp20 TensorShape& outputShape) in CreateSimpleReduceNetwork()
45 const TensorShape& outputShape, in ReduceWithMultipleAxesTest()
124 TensorShape outputShape = {1, 1, 1, 4}; in ReduceSumWithTwoAxesKeepDimsTest() local
156 TensorShape outputShape = {1, 4}; in ReduceSumWithTwoAxesTest() local
188 TensorShape outputShape = {1, 2, 1, 1}; in ReduceSumWithThreeAxesKeepDimsTest() local
223 TensorShape outputShape = {2}; in ReduceSumWithThreeAxesTest() local
/external/pytorch/torch/csrc/jit/tensorexpr/operators/
Dpointwise.cpp10 const std::vector<ExprHandle>& outputShape, in computeSign()
27 const std::vector<ExprHandle>& outputShape, in computeOneOperand()
50 const std::vector<ExprHandle>& outputShape, in computeTwoOperand()
75 const std::vector<ExprHandle>& outputShape, in computeTwoOperandWithAlpha()
101 const std::vector<ExprHandle>& outputShape, in computeConditionWithTwoOperand()
130 const std::vector<ExprHandle>& outputShape, in computeThreeOperand()
160 const std::vector<ExprHandle>& outputShape, in computeFourOperand()
190 const std::vector<ExprHandle>& outputShape, in computeNoop()
206 const std::vector<ExprHandle>& outputShape, in computeScalar()
Dquantization.cpp140 const std::vector<ExprHandle>& outputShape, in computeQuantizePerTensor()
180 const std::vector<ExprHandle>& outputShape, in computeQuantizedAdd()
223 const std::vector<ExprHandle>& outputShape, in computeQuantizePerTensorExternalCall()
255 const std::vector<ExprHandle>& outputShape, in computeDequantizeExternalCall()
280 const std::vector<ExprHandle>& outputShape, in computeQuantizedConv2dPrepack()
323 const std::vector<ExprHandle>& outputShape, in computeQuantizedConv1d()
353 const std::vector<ExprHandle>& outputShape, in computeQuantizedConv2d()
383 const std::vector<ExprHandle>& outputShape, in computeQuantizedConv2dRelu()
413 const std::vector<ExprHandle>& outputShape, in computeQuantizedLinear()
443 const std::vector<ExprHandle>& outputShape, in computeQuantizedLinearRelu()
[all …]
/external/armnn/src/armnn/layers/
DArgMinMaxLayer.cpp53 TensorShape outputShape(1, tensorDimensions.data()); in InferOutputShapes() local
69 TensorShape outputShape = TensorShape(inputNumDimensions - 1, tensorDimensions.data()); in InferOutputShapes() local
78 const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape(); in ValidateTensorShapesFromInputs() local
/external/armnn/src/backends/reference/test/
DRefCreateWorkloadTests.cpp297 TensorShape outputShape; in RefCreateBatchNormalizationWorkloadTest() local
325 TensorShape outputShape; variable
413 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in RefCreateConvolution2dWorkloadTest() local
442 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… variable
460 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in RefCreateDepthwiseConvolutionWorkloadTest() local
545 TensorShape outputShape; in RefCreateNormalizationWorkloadTest() local
602 TensorShape outputShape; in RefCreatePooling2dWorkloadTest() local
841 TensorShape outputShape; in RefCreateResizeBilinearTest() local
928 TensorShape outputShape; in RefCreateL2NormalizationTest() local
1007 static void RefCreateConcatWorkloadTest(const armnn::TensorShape& outputShape, in RefCreateConcatWorkloadTest()
[all …]

123456789