Home
last modified time | relevance | path

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

12345678910>>...12

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A 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 …]
/aosp_15_r20/external/armnn/delegate/test/
H A 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 …]
H A 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 …]
H A 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
H A 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
H A 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
H A 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
H A 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
/aosp_15_r20/external/armnn/src/backends/tosaCommon/test/
H A 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 …]
/aosp_15_r20/external/armnn/src/armnnUtils/test/
H A 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 …]
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A 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 …]
H A DStridedSliceTestImpl.cpp86 unsigned int outputShape[] = {1, 2, 3, 1}; in StridedSlice4dTest() local
125 unsigned int outputShape[] = {1, 2, 3, 1}; in StridedSlice4dReverseTest() local
164 unsigned int outputShape[] = {2, 1, 2, 1}; in StridedSliceSimpleStrideTest() local
205 unsigned int outputShape[] = {3, 2, 3, 1}; in StridedSliceSimpleRangeMaskTest() local
250 unsigned int outputShape[] = {3, 1}; in StridedSliceShrinkAxisMaskTest() local
291 unsigned int outputShape[] = {2, 3, 1}; in StridedSliceShrinkAxisMaskBitPosition0Test() local
332 unsigned int outputShape[] = {3, 3, 1}; in StridedSliceShrinkAxisMaskBitPosition1Test() local
373 unsigned int outputShape[] = {3, 2, 1}; in StridedSliceShrinkAxisMaskBitPosition2Test() local
414 unsigned int outputShape[] = {3, 2, 3}; in StridedSliceShrinkAxisMaskBitPosition3Test() local
459 unsigned int outputShape[] = {3, 1}; in StridedSliceShrinkAxisMaskBitPosition0And1Test() local
[all …]
H A 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
H A 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
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DActivation.cpp73 const Shape& outputShape) { in relu1Float()
83 const Shape& outputShape) { in relu6Float()
144 const Shape& outputShape) { in reluQuant8()
150 const Shape& outputShape) { in relu1Quant8()
156 const Shape& outputShape) { in relu6Quant8()
162 const Shape& outputShape) { in tanhQuant8()
192 const Shape& outputShape) { in logisticQuant8()
239 const Shape& outputShape) { in reluQuant8Signed()
245 const Shape& outputShape) { in relu1Quant8Signed()
251 const Shape& outputShape) { in relu6Quant8Signed()
[all …]
DPooling.cpp144 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
154 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
167 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
177 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
190 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
200 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc()
212 float* outputData, const Shape& outputShape) { in maxPoolNhwc()
222 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
232 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
244 _Float16* outputData, const Shape& outputShape) { in maxPoolNhwc()
[all …]
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A 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 …]
H A 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 …]
H A 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
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/
H A 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 …]
H A 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()
/aosp_15_r20/external/armnn/src/backends/reference/test/
H A 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 …]
/aosp_15_r20/external/armnn/src/armnn/test/
H A DOptimizerTests.cpp424 const unsigned int* weightsShape, const unsigned int* outputShape, in CreateConvolution2dGraph()
463 const unsigned int outputShape[] = { 1, 2, 4, 14 }; variable
474 const unsigned int outputShape[] = { 1, 4, 14, 2 }; variable
481 … const unsigned int* weightsShape, const unsigned int* outputShape, in CreateDepthwiseConvolution2dGraph()
518 const unsigned int outputShape[] = { 1, 2, 1, 1 }; variable
529 const unsigned int outputShape[] = { 1, 1, 1, 2 }; variable
535 …reatePooling2dGraph(Graph& graph, const unsigned int* inputShape, const unsigned int* outputShape, in CreatePooling2dGraph()
567 const unsigned int outputShape[] = { 5, 3, 11, 13 }; variable
577 const unsigned int outputShape[] = { 5, 11, 13, 3 }; variable
585 const unsigned int* outputShape, in CreateResizeBilinearGraph()
[all …]
/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A 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
H A DFoldPadTests.cpp24 const unsigned int outputShape[] = {1, 2, 1, 1}; variable
106 const unsigned int outputShape[] = {1, 2, 1, 3}; variable
191 const unsigned int outputShape[] = {1, 2, 2, 3}; variable
276 const unsigned int outputShape[] = {1, 2, 2, 3}; variable
347 const unsigned int outputShape[] = {1, 2, 2, 3}; variable
417 const unsigned int outputShape[] = {1, 2, 2, 3}; variable
482 const unsigned int outputShape[] = {1, 2, 2, 3}; variable
547 const unsigned int outputShape[] = {1, 4, 4, 2}; variable
647 const unsigned int outputShape[] = {1, 5, 5, 4}; // NHWCout variable
771 const unsigned int outputShape[] = {1, 5, 5, 12}; // NHWCout variable

12345678910>>...12