Home
last modified time | relevance | path

Searched refs:outputTensorShape (Results 1 – 25 of 30) sorted by relevance

12

/external/armnn/delegate/test/
DFullyConnectedTest.cpp16 std::vector<int32_t> outputTensorShape { 1, 1 }; in FullyConnectedFp32Test() local
30 outputTensorShape, in FullyConnectedFp32Test()
42 std::vector<int32_t> outputTensorShape { 1, 1 }; in FullyConnectedActivationTest() local
56 outputTensorShape, in FullyConnectedActivationTest()
68 std::vector<int32_t> outputTensorShape { 2, 1 }; in FullyConnectedInt8Test() local
84 outputTensorShape, in FullyConnectedInt8Test()
DSliceTestHelper.hpp30 const std::vector<int32_t>& outputTensorShape) in CreateSliceTfLiteModel() argument
67 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateSliceTfLiteModel()
68 outputTensorShape.size()), in CreateSliceTfLiteModel()
125 std::vector<int32_t>& outputTensorShape) in SliceTestImpl() argument
135 outputTensorShape); in SliceTestImpl()
154 armnnDelegate::CompareOutputShape(tfLiteOutputShape, armnnOutputShape, outputTensorShape); in SliceTestImpl()
DStridedSliceTestHelper.hpp32 const std::vector<int32_t>& outputTensorShape, in CreateStridedSliceTfLiteModel() argument
84 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateStridedSliceTfLiteModel()
85 outputTensorShape.size()), in CreateStridedSliceTfLiteModel()
149 std::vector<int32_t>& outputTensorShape, in StridedSliceTestImpl() argument
167 outputTensorShape, in StridedSliceTestImpl()
192 armnnDelegate::CompareOutputShape(tfLiteOutputShape, armnnOutputShape, outputTensorShape); in StridedSliceTestImpl()
DFullyConnectedTestHelper.hpp30 std::vector <int32_t>& outputTensorShape, in CreateFullyConnectedTfLiteModel()
116 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateFullyConnectedTfLiteModel()
117 outputTensorShape.size()), in CreateFullyConnectedTfLiteModel()
175 std::vector <int32_t>& outputTensorShape, in FullyConnectedTest()
190 outputTensorShape, in FullyConnectedTest()
235 armnnDelegate::CompareOutputShape(tfLiteOutputShape, armnnOutputShape, outputTensorShape); in FullyConnectedTest()
DRedefineTestHelper.hpp28 const std::vector<int32_t>& outputTensorShape, in CreateRedefineTfLiteModel() argument
64 … flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateRedefineTfLiteModel()
65 … outputTensorShape.size()), in CreateRedefineTfLiteModel()
92 … flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateRedefineTfLiteModel()
93 … outputTensorShape.size()), in CreateRedefineTfLiteModel()
DConvolutionTestHelper.hpp37 const std::vector <int32_t>& outputTensorShape, in CreateConv2dTfLiteModel()
124 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(),
125 outputTensorShape.size()),
289 const std::vector<int32_t>& outputTensorShape, in CreateConv3dTfLiteModel() argument
374 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(),
375 outputTensorShape.size()),
522 const std::vector <int32_t>& outputTensorShape, in CreateTransposeConvTfLiteModel()
584 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateTransposeConvTfLiteModel()
585 outputTensorShape.size()), in CreateTransposeConvTfLiteModel()
643 const std::vector <int32_t>& outputTensorShape, in TransposeConvTest()
[all …]
DControlTestHelper.hpp27 const std::vector <int32_t>& outputTensorShape, in CreateConcatTfLiteModel()
71 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateConcatTfLiteModel()
72 outputTensorShape.size()), in CreateConcatTfLiteModel()
119 const std::vector <int32_t>& outputTensorShape, in CreateMeanTfLiteModel()
160 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateMeanTfLiteModel()
161 outputTensorShape.size()), in CreateMeanTfLiteModel()
DTransposeTestHelper.hpp26 const std::vector <int32_t>& outputTensorShape, in CreateTransposeTfLiteModel()
50 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateTransposeTfLiteModel()
51 outputTensorShape.size()), in CreateTransposeTfLiteModel()
DShapeTestHelper.hpp26 const std::vector<int32_t>& outputTensorShape, in CreateShapeTfLiteModel() argument
54 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateShapeTfLiteModel()
55 outputTensorShape.size()), in CreateShapeTfLiteModel()
DQuantizationTestHelper.hpp28 const std::vector <int32_t>& outputTensorShape, in CreateQuantizationTfLiteModel()
58 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateQuantizationTfLiteModel()
59 outputTensorShape.size()), in CreateQuantizationTfLiteModel()
DResizeTestHelper.hpp29 const std::vector <int32_t>& outputTensorShape) in CreateResizeTfLiteModel()
58 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateResizeTfLiteModel()
59 outputTensorShape.size()), in CreateResizeTfLiteModel()
DPackTestHelper.hpp27 const std::vector <int32_t>& outputTensorShape, in CreatePackTfLiteModel()
70 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreatePackTfLiteModel()
71 outputTensorShape.size()), in CreatePackTfLiteModel()
DSpaceDepthTestHelper.hpp26 const std::vector <int32_t>& outputTensorShape, in CreateSpaceDepthTfLiteModel()
53 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateSpaceDepthTfLiteModel()
54 outputTensorShape.size()), in CreateSpaceDepthTfLiteModel()
DUnpackTestHelper.hpp27 const std::vector <int32_t>& outputTensorShape, in CreateUnpackTfLiteModel()
67 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateUnpackTfLiteModel()
68 outputTensorShape.size()), in CreateUnpackTfLiteModel()
DArgMinMaxTestHelper.hpp29 const std::vector<int32_t>& outputTensorShape, in CreateArgMinMaxTfLiteModel() argument
61 … flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateArgMinMaxTfLiteModel()
62 … outputTensorShape.size()), in CreateArgMinMaxTfLiteModel()
DReduceTestHelper.hpp28 const std::vector <int32_t>& outputTensorShape, in CreateReduceTfLiteModel()
99 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateReduceTfLiteModel()
100 outputTensorShape.size()), in CreateReduceTfLiteModel()
DBatchMatMulTestHelper.hpp28 const std::vector <int32_t>& outputTensorShape, in CreateBatchMatMulTfLiteModel()
68 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateBatchMatMulTfLiteModel()
69 outputTensorShape.size()), in CreateBatchMatMulTfLiteModel()
DLogicalTestHelper.hpp28 const std::vector <int32_t>& outputTensorShape, in CreateLogicalBinaryTfLiteModel()
65 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateLogicalBinaryTfLiteModel()
66 outputTensorShape.size()), in CreateLogicalBinaryTfLiteModel()
DComparisonTestHelper.hpp28 const std::vector <int32_t>& outputTensorShape, in CreateComparisonTfLiteModel()
64 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateComparisonTfLiteModel()
65 outputTensorShape.size()), in CreateComparisonTfLiteModel()
DBatchSpaceTestHelper.hpp27 std::vector <int32_t>& outputTensorShape, in CreateBatchSpaceTfLiteModel()
87 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateBatchSpaceTfLiteModel()
88 outputTensorShape.size()), in CreateBatchSpaceTfLiteModel()
DNormalizationTestHelper.hpp27 const std::vector<int32_t>& outputTensorShape, in CreateNormalizationTfLiteModel() argument
54 … flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateNormalizationTfLiteModel()
55 … outputTensorShape.size()), in CreateNormalizationTfLiteModel()
DPadTestHelper.hpp31 const std::vector<int32_t>& outputTensorShape, in CreatePadTfLiteModel() argument
63 … flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreatePadTfLiteModel()
64 … outputTensorShape.size()), in CreatePadTfLiteModel()
DElementwiseBinaryTestHelper.hpp30 const std::vector <int32_t>& outputTensorShape, in CreateElementwiseBinaryTfLiteModel()
79 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreateElementwiseBinaryTfLiteModel()
80 outputTensorShape.size()), in CreateElementwiseBinaryTfLiteModel()
DPooling3dTestHelper.hpp33 const std::vector<int32_t>& outputTensorShape, in CreatePooling3dTfLiteModel() argument
72 flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), in CreatePooling3dTfLiteModel()
73 outputTensorShape.size()), in CreatePooling3dTfLiteModel()
/external/armnn/src/armnnUtils/
DParserHelper.cpp93 … armnn::TensorShape outputTensorShape(inputTensorInfo.GetNumDimensions(), &outputShapeVector[0]); in CalculateReducedOutputTensoInfo() local
94 outputTensorInfo = armnn::TensorInfo(outputTensorShape, inputTensorInfo.GetDataType()); in CalculateReducedOutputTensoInfo()
129 armnn::TensorShape outputTensorShape(inputTensorInfo.GetNumDimensions(), &outputShapeVector[0]); in CalculateStridedSliceOutputTensorInfo() local
130 …outputTensorInfo = armnn::TensorInfo(armnn::TensorShape(outputTensorShape), inputTensorInfo.GetDat… in CalculateStridedSliceOutputTensorInfo()

12