Home
last modified time | relevance | path

Searched defs:tensorShape (Results 1 – 25 of 44) sorted by relevance

12

/aosp_15_r20/external/armnn/delegate/test/
H A DDelegateOptionsTest.cpp23 std::vector<int32_t> tensorShape { 1, 2, 2, 1 }; variable
52 std::vector<int32_t> tensorShape { 1, 2, 2, 1 }; variable
80 std::vector<int32_t> tensorShape { 1, 2, 2, 1 }; variable
119 std::vector<int32_t> tensorShape { 1, 2, 2, 1 }; variable
143 std::vector<int32_t> tensorShape { 1, 2, 2, 1 }; variable
178 std::vector<int32_t> tensorShape { 1, 2, 2, 1 }; variable
215 std::vector<int32_t> tensorShape { 1, 2, 2, 1 }; variable
255 std::vector<int32_t> tensorShape { 1, 2, 2, 1 }; variable
287 std::vector<int32_t> tensorShape { 1, 2, 2, 1 }; in CreateFp16StringParsingTestRun() local
H A DFillTest.cpp23 std::vector<int32_t> tensorShape { 2, 2 }; in Fill2dTest() local
41 std::vector<int32_t> tensorShape { 3, 3, 3 }; in Fill3dTest() local
68 std::vector<int32_t> tensorShape { 2, 2, 4, 4 }; in Fill4dTest() local
103 std::vector<int32_t> tensorShape { 2, 2 }; in FillInt32Test() local
H A DDelegateOptionsTestHelper.hpp38 const std::vector<int32_t>& tensorShape, in CreateAddDivTfLiteModel()
155 std::vector<char> CreateCosTfLiteModel(tflite::TensorType tensorType, in CreateCosTfLiteModel()
228 std::vector<int32_t>& tensorShape, in DelegateOptionTest()
272 std::vector<int32_t>& tensorShape, in DelegateOptionNoFallbackTest()
H A DFillTestHelper.hpp25 std::vector<char> CreateFillTfLiteModel(tflite::BuiltinOperator fillOperatorCode, in CreateFillTfLiteModel()
113 void FillTest(tflite::BuiltinOperator fillOperatorCode, in FillTest()
H A DActivationTestHelper.hpp24 std::vector<char> CreateActivationTfLiteModel(tflite::BuiltinOperator activationOperatorCode, in CreateActivationTfLiteModel()
/aosp_15_r20/external/armnn/src/armnn/test/
H A DShapeInferenceTests.cpp305 const TensorShape tensorShape; variable
343 const TensorShape tensorShape{5, 7, 6, 2}; variable
391 const TensorShape tensorShape{5, 7, 6, 2}; variable
416 const TensorShape tensorShape{5, 7, 6, 2}; variable
425 const TensorShape tensorShape{5, 7, 6, 2}; variable
434 const TensorShape tensorShape{5, 7, 6, 2}; variable
494 const TensorShape tensorShape{ 5, 7, 6, 2 }; variable
500 const TensorShape tensorShape{5, 7, 6, 2}; variable
592 const TensorShape tensorShape { 5, 4, 7, 6 }; variable
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DBaseIterator.hpp128 std::vector<float> DecodeTensor (const TensorShape& tensorShape, in DecodeTensor()
165 std::vector<float> DecodeTensor (const TensorShape& tensorShape, in DecodeTensor()
202 std::vector<float> DecodeTensor (const TensorShape& tensorShape, in DecodeTensor()
239 std::vector<float> DecodeTensor (const TensorShape& tensorShape, in DecodeTensor()
278 std::vector<float> DecodeTensor (const TensorShape& tensorShape, in DecodeTensor()
314 std::vector<float> DecodeTensor (const TensorShape& tensorShape, in DecodeTensor()
341 std::vector<float> DecodeTensor (const TensorShape& tensorShape, in DecodeTensor()
377 std::vector<float> DecodeTensor (const TensorShape& tensorShape, in DecodeTensor()
409 std::vector<float> DecodeTensor (const TensorShape& tensorShape, in DecodeTensor()
441 std::vector<float> DecodeTensor (const TensorShape& tensorShape, in DecodeTensor()
[all …]
H A DRefChannelShuffleWorkload.cpp43 { in Execute()
51 const TensorShape tensorShape = GetTensorInfo(inputs[0]).GetShape(); in Execute() local
H A DRefNormalizationWorkload.cpp28 const TensorShape& tensorShape, in NormalizeWithinUingLbr()
94 const TensorShape& tensorShape, in NormalizeAcrossUingLbr()
/aosp_15_r20/external/armnn/src/backends/cl/test/
H A DClImportTensorHandleFactoryTests.cpp33 TensorShape tensorShape{ 6, 7, 8, 9 }; variable
61 TensorShape tensorShape{ 224, 224, 1, 1 }; variable
80 TensorShape tensorShape{ 224, 224, 1, 1 }; variable
99 TensorShape tensorShape{ 224, 224, 1, 1 }; variable
117 TensorShape tensorShape{ 224, 224, 1, 1 }; variable
/aosp_15_r20/external/android-nn-driver/
H A DUtils.cpp116 TensorShape tensorShape(Dimensionality::NotSpecified); in GetTensorInfoForOperand() local
122 …TensorShape tensorShape(operand.dimensions.size(), operand.dimensions.data(), dimensionsSpecificit… in GetTensorInfoForOperand() local
173 TensorShape tensorShape(Dimensionality::NotSpecified); in GetTensorInfoForOperand() local
179 …TensorShape tensorShape(operand.dimensions.size(), operand.dimensions.data(), dimensionsSpecificit… in GetTensorInfoForOperand() local
261 TensorShape tensorShape(Dimensionality::NotSpecified); in GetTensorInfoForOperand() local
267 …TensorShape tensorShape(operand.dimensions.size(), operand.dimensions.data(), dimensionsSpecificit… in GetTensorInfoForOperand() local
H A DUtils.hpp173 armnn::TensorShape tensorShape = info.GetShape(); in ComputeShape() local
/aosp_15_r20/external/armnn/src/armnnUtils/
H A DTensorUtils.cpp106 TensorShape ReduceDims(const TensorShape& tensorShape, unsigned int dimensions) in ReduceDims()
140 TensorShape ExpandDims(const TensorShape& tensorShape, int axis) in ExpandDims()
168 TensorShape ExpandDimsToRank(const TensorShape& tensorShape, unsigned int rank) in ExpandDimsToRank()
193 std::vector<unsigned int> SqueezeDims(const TensorShape& tensorShape) in SqueezeDims()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DDequantizeEndToEndTestImpl.hpp35 const armnn::TensorShape& tensorShape, in DequantizeEndToEndLayerTestImpl()
63 const armnn::TensorShape tensorShape({ 1, 2, 2, 4 }); in DequantizeEndToEndSimple() local
85 const armnn::TensorShape tensorShape({ 1, 2, 2, 4 }); in DequantizeEndToEndOffset() local
/aosp_15_r20/external/armnn/src/backends/aclCommon/
H A DArmComputeTensorUtils.cpp125 arm_compute::TensorShape BuildArmComputeTensorShape(const armnn::TensorShape& tensorShape) in BuildArmComputeTensorShape()
149 std::vector<unsigned int> ReduceDimsForACL(const armnn::TensorShape tensorShape, unsigned int dimen… in ReduceDimsForACL()
176 arm_compute::TensorShape BuildArmComputeTensorShape(const armnn::TensorShape& tensorShape, unsigned… in BuildArmComputeTensorShape()
/aosp_15_r20/external/armnn/src/armnn/
H A DDescriptors.cpp478 const TensorShape& tensorShape) in GetAxesToMul()
499 const TensorShape& tensorShape) in GetAxesNotMul()
516 const TensorShape& tensorShape) in GetPermuteVec()
/aosp_15_r20/external/armnn/src/backends/reference/test/
H A DRefPerAxisIteratorTests.cpp45 …MockPerAxisIterator(const int8_t* data, const armnn::TensorShape& tensorShape, const unsigned int … in MockPerAxisIterator()
54 virtual std::vector<float> DecodeTensor(const TensorShape &tensorShape, in DecodeTensor()
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DCanonicalUtils.cpp150 TensorShape tensorShape(Dimensionality::NotSpecified); in GetTensorInfoForOperand() local
168 …TensorShape tensorShape(operand.dimensions.size(), operand.dimensions.data(), dimensionsSpecificit… in GetTensorInfoForOperand() local
H A DCanonicalUtils.hpp99 armnn::TensorShape tensorShape = info.GetShape(); in ComputeShape() local
/aosp_15_r20/external/armnn/src/backends/aclCommon/test/
H A DArmComputeTensorUtilsTests.cpp17 const armnn::TensorShape tensorShape = { 1, 2, 3, 4 }; variable
/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A DConvertConstPermuteLayersToConstLayersTest.cpp52 TensorShape tensorShape = constant->GetOutputSlot(0).GetTensorInfo().GetShape(); variable
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DReverse.cpp65 const tflite::RuntimeShape tensorShape = in reverse() local
DPack.cpp95 const Shape tensorShape = context->getInputShape(kInputFirstTensor); in packParams() local
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DPadLayer.cpp57 TensorShape tensorShape = TensorShape( rank, outputDimensionSizes.data()); in InferOutputShapes() local
H A DResizeLayer.cpp51 TensorShape tensorShape = m_Param.m_DataLayout == armnn::DataLayout::NHWC ? in InferOutputShapes() local

12