Home
last modified time | relevance | path

Searched refs:subTensorShape (Results 1 – 24 of 24) sorted by relevance

/external/armnn/src/backends/cl/test/
DClImportTensorHandleFactoryTests.cpp65 TensorShape subTensorShape{ 16, 16, 1, 1 }; variable
68 auto subTensor = factory.CreateSubTensorHandle(*tensorHandle, subTensorShape, origin);
70 ARMNN_ASSERT(subTensor->GetShape() == subTensorShape);
84 TensorShape subTensorShape{ 16, 16, 1, 1 }; variable
88 auto subTensor = factory.CreateSubTensorHandle(*tensorHandle, subTensorShape, origin);
103 TensorShape subTensorShape{ 16, 16, 2, 2 }; variable
106 auto subTensor = factory.CreateSubTensorHandle(*tensorHandle, subTensorShape, origin);
121 TensorShape subTensorShape{ 224, 224, 1, 1 }; variable
124 auto subTensor = factory.CreateSubTensorHandle(*tensorHandle, subTensorShape, origin);
/external/armnn/src/backends/cl/
DClTensorHandleFactory.cpp22 … const TensorShape& subTensorShape, in CreateSubTensorHandle() argument
26 …arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape); in CreateSubTensorHandle()
28 coords.set_num_dimensions(subTensorShape.GetNumDimensions()); in CreateSubTensorHandle()
29 for (unsigned int i = 0; i < subTensorShape.GetNumDimensions(); ++i) in CreateSubTensorHandle()
32 unsigned int revertedIndex = subTensorShape.GetNumDimensions() - i - 1; in CreateSubTensorHandle()
DClImportTensorHandleFactory.cpp21 …ITensorHandle& parent, const TensorShape& subTensorShape, const unsigned int* subTensorOrigin) con… in CreateSubTensorHandle() argument
24 …arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape); in CreateSubTensorHandle()
26 coords.set_num_dimensions(subTensorShape.GetNumDimensions()); in CreateSubTensorHandle()
27 for (unsigned int i = 0; i < subTensorShape.GetNumDimensions(); ++i) in CreateSubTensorHandle()
30 unsigned int revertedIndex = subTensorShape.GetNumDimensions() - i - 1; in CreateSubTensorHandle()
DClTensorHandleFactory.hpp30 const TensorShape& subTensorShape,
DClImportTensorHandleFactory.hpp40 const TensorShape& subTensorShape,
DClWorkloadFactory.cpp223 … TensorShape const& subTensorShape, in CreateSubTensorHandle() argument
227 …arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape); in CreateSubTensorHandle()
229 coords.set_num_dimensions(subTensorShape.GetNumDimensions()); in CreateSubTensorHandle()
230 for (unsigned int i = 0; i < subTensorShape.GetNumDimensions(); i++) in CreateSubTensorHandle()
233 unsigned int revertedIndex = subTensorShape.GetNumDimensions() - i - 1; in CreateSubTensorHandle()
DClWorkloadFactory.hpp46 TensorShape const& subTensorShape,
/external/armnn/src/backends/neon/
DNeonTensorHandleFactory.cpp21 … const TensorShape& subTensorShape, in CreateSubTensorHandle() argument
25 …arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape); in CreateSubTensorHandle()
28 coords.set_num_dimensions(subTensorShape.GetNumDimensions()); in CreateSubTensorHandle()
29 for (unsigned int i = 0; i < subTensorShape.GetNumDimensions(); ++i) in CreateSubTensorHandle()
32 unsigned int revertedIndex = subTensorShape.GetNumDimensions() - i - 1; in CreateSubTensorHandle()
DNeonTensorHandleFactory.hpp44 const TensorShape& subTensorShape,
DNeonWorkloadFactory.cpp87 TensorShape const& subTensorShape, in CreateSubTensorHandle() argument
90 …arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape); in CreateSubTensorHandle()
93 coords.set_num_dimensions(subTensorShape.GetNumDimensions()); in CreateSubTensorHandle()
94 for (unsigned int i = 0; i < subTensorShape.GetNumDimensions(); i++) in CreateSubTensorHandle()
97 unsigned int revertedIndex = subTensorShape.GetNumDimensions() - i - 1; in CreateSubTensorHandle()
DNeonWorkloadFactory.hpp43 TensorShape const& subTensorShape,
/external/armnn/src/backends/tosaReference/
DTosaRefWorkloadFactory.hpp41 TensorShape const& subTensorShape, in CreateSubTensorHandle() argument
44 IgnoreUnused(parent, subTensorShape, subTensorOrigin); in CreateSubTensorHandle()
DTosaRefTensorHandleFactory.cpp23 … const TensorShape& subTensorShape, in CreateSubTensorHandle() argument
27 IgnoreUnused(parent, subTensorShape, subTensorOrigin); in CreateSubTensorHandle()
DTosaRefTensorHandleFactory.hpp28 TensorShape const& subTensorShape,
/external/armnn/src/backends/reference/
DRefTensorHandleFactory.cpp23 … TensorShape const& subTensorShape, in CreateSubTensorHandle() argument
26 IgnoreUnused(parent, subTensorShape, subTensorOrigin); in CreateSubTensorHandle()
DRefTensorHandleFactory.hpp28 TensorShape const& subTensorShape,
DRefWorkloadFactory.hpp53 TensorShape const& subTensorShape, in CreateSubTensorHandle() argument
56 IgnoreUnused(parent, subTensorShape, subTensorOrigin); in CreateSubTensorHandle()
/external/armnn/src/dynamic/sample/
DSampleDynamicWorkloadFactory.hpp35 armnn::TensorShape const& subTensorShape, in CreateSubTensorHandle() argument
38 IgnoreUnused(parent, subTensorShape, subTensorOrigin); in CreateSubTensorHandle()
DSampleDynamicTensorHandleFactory.cpp24 armnn::TensorShape const& subTensorShape, in CreateSubTensorHandle() argument
27 IgnoreUnused(parent, subTensorShape, subTensorOrigin); in CreateSubTensorHandle()
DSampleDynamicTensorHandleFactory.hpp28 … armnn::TensorShape const& subTensorShape,
/external/armnn/src/armnn/test/
DTensorHandleStrategyTest.cpp45 TensorShape const& subTensorShape, in CreateSubTensorHandle() argument
48 IgnoreUnused(parent, subTensorShape, subTensorOrigin); in CreateSubTensorHandle()
86 TensorShape const& subTensorShape, in CreateSubTensorHandle() argument
89 IgnoreUnused(parent, subTensorShape, subTensorOrigin); in CreateSubTensorHandle()
/external/armnn/src/armnnTestUtils/
DMockTensorHandleFactory.hpp30 TensorShape const& subTensorShape,
/external/armnn/include/armnn/backends/
DITensorHandleFactory.hpp56 TensorShape const& subTensorShape,
DWorkloadFactory.hpp55 TensorShape const& subTensorShape,