Home
last modified time | relevance | path

Searched defs:tensorInfo (Results 1 – 25 of 92) sorted by relevance

1234

/external/armnn/include/armnnTestUtils/
DDataLayoutUtils.hpp14 void PermuteTensorNchwToNhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNchwToNhwc()
26 void PermuteTensorNhwcToNchw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNhwcToNchw()
39 void PermuteTensorNdhwcToNcdhw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNdhwcToNcdhw()
51 void PermuteTensorNcdhwToNdhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNcdhwToNdhwc()
DWorkloadTestUtils.hpp25 const armnn::TensorInfo& tensorInfo, in AddInputToWorkload()
35 const armnn::TensorInfo& tensorInfo, in AddOutputToWorkload()
46 const armnn::TensorInfo& tensorInfo, in SetWorkloadInput()
57 const armnn::TensorInfo& tensorInfo, in SetWorkloadOutput()
/external/armnn/src/backends/reference/test/
DRefPerAxisIteratorTests.cpp86 TensorInfo tensorInfo ({3,1,2,2},DataType::QSymmS8); variable
109 TensorInfo tensorInfo ({3,1,2,2},DataType::QSymmS8); variable
132 TensorInfo tensorInfo ({3,1,2,2},DataType::QSymmS8); variable
155 TensorInfo tensorInfo ({3,1,2,2},DataType::QSymmS8); variable
194 TensorInfo tensorInfo ({2,2,2,2},DataType::QSymmS8); variable
228 TensorInfo tensorInfo ({3,1,2,2},DataType::QSymmS8); variable
DRefPerChannelDecoderTests.cpp54 TensorInfo tensorInfo ({2,2,2,3},DataType::QSymmS8,{1.0f, 2.0f},0); variable
80 TensorInfo tensorInfo ({4,1,2,2},DataType::QSymmS8,{1.0f, 2.0f, 3.0f, 4.0f},0); variable
113 TensorInfo tensorInfo ({6,1,2,2},DataType::QSymmS8,{1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f},0); variable
146 TensorInfo tensorInfo ({6,1,2,2},DataType::Signed32,{1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f},0); variable
/external/armnn/src/armnn/optimizations/
DFoldPadIntoLayer2d.hpp21 inline float GetZeroElement(const TensorInfo& tensorInfo) in GetZeroElement()
26 inline float GetLowestElement(const TensorInfo& tensorInfo) in GetLowestElement()
56 inline bool IsNeutralElement(const Convolution2dDescriptor&, const TensorInfo& tensorInfo, const fl… in IsNeutralElement()
62 const TensorInfo& tensorInfo, in IsNeutralElement()
69 const Pooling2dDescriptor& descriptor, const TensorInfo& tensorInfo, const float tensorValue) in IsNeutralElement()
89 const PadDescriptor& padDescriptor, Descriptor& layerDescriptor, const TensorInfo& tensorInfo) in TryFoldPadIntoLayer2d()
117 const TensorInfo& tensorInfo, in TryFoldPadIntoLayer2d()
/external/armnn/src/backends/backendsCommon/
DTensorHandle.cpp15 TensorShape GetUnpaddedTensorStrides(const TensorInfo& tensorInfo) in GetUnpaddedTensorStrides()
31 ConstTensorHandle::ConstTensorHandle(const TensorInfo& tensorInfo) in ConstTensorHandle()
43 TensorHandle::TensorHandle(const TensorInfo& tensorInfo) in TensorHandle()
55 ScopedTensorHandle::ScopedTensorHandle(const TensorInfo& tensorInfo) in ScopedTensorHandle()
/external/armnn/src/dynamic/sample/
DSampleDynamicTensorHandleFactory.cpp38 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle()
46 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle()
60 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle()
DSampleTensorHandle.cpp11 SampleTensorHandle::SampleTensorHandle(const armnn::TensorInfo &tensorInfo, in SampleTensorHandle()
23 SampleTensorHandle::SampleTensorHandle(const armnn::TensorInfo& tensorInfo, in SampleTensorHandle()
DSampleDynamicWorkloadFactory.cpp45 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle()
52 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle()
/external/armnn/src/backends/reference/
DRefTensorHandleFactory.cpp35 …:unique_ptr<ITensorHandle> RefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
42 …:unique_ptr<ITensorHandle> RefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
55 …:unique_ptr<ITensorHandle> RefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
DRefTensorHandle.cpp10 RefTensorHandle::RefTensorHandle(const TensorInfo &tensorInfo, std::shared_ptr<RefMemoryManager> &m… in RefTensorHandle()
20 RefTensorHandle::RefTensorHandle(const TensorInfo& tensorInfo) in RefTensorHandle()
/external/armnn/src/armnnTestUtils/
DMockTensorHandleFactory.cpp32 …unique_ptr<ITensorHandle> MockTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
39 …unique_ptr<ITensorHandle> MockTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
52 …unique_ptr<ITensorHandle> MockTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
DMockTensorHandle.cpp11 MockTensorHandle::MockTensorHandle(const TensorInfo& tensorInfo, std::shared_ptr<MockMemoryManager>… in MockTensorHandle()
21 MockTensorHandle::MockTensorHandle(const TensorInfo& tensorInfo, MemorySourceFlags importFlags) in MockTensorHandle()
/external/armnn/src/backends/tosaReference/
DTosaRefTensorHandleFactory.cpp36 …que_ptr<ITensorHandle> TosaRefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
43 …que_ptr<ITensorHandle> TosaRefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
56 …que_ptr<ITensorHandle> TosaRefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
DTosaRefWorkloadFactory.cpp35 … auto checkType = [](const TensorInfo& tensorInfo) {return tensorInfo.GetDataType() == ArmnnType;}; in IsDataType()
79 …:unique_ptr<ITensorHandle> TosaRefWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
92 …:unique_ptr<ITensorHandle> TosaRefWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
DTosaRefTensorHandle.cpp10 TosaRefTensorHandle::TosaRefTensorHandle(const TensorInfo& tensorInfo, in TosaRefTensorHandle()
21 TosaRefTensorHandle::TosaRefTensorHandle(const TensorInfo& tensorInfo, in TosaRefTensorHandle()
/external/armnn/src/backends/cl/
DClTensorHandleFactory.cpp62 …::unique_ptr<ITensorHandle> ClTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
68 …::unique_ptr<ITensorHandle> ClTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
80 …::unique_ptr<ITensorHandle> ClTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
DClImportTensorHandleFactory.cpp63 …ue_ptr<ITensorHandle> ClImportTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
72 …ue_ptr<ITensorHandle> ClImportTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
82 …ue_ptr<ITensorHandle> ClImportTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
/external/android-nn-driver/
DConversionUtils.cpp21 …tHandle::LayerInputHandle(bool valid, armnn::IOutputSlot* outputSlot, armnn::TensorInfo tensorInfo) in LayerInputHandle()
82 ConstTensorPin::ConstTensorPin(armnn::TensorInfo& tensorInfo, in ConstTensorPin()
137 armnn::IConnectableLayer* ProcessActivation(const armnn::TensorInfo& tensorInfo, in ProcessActivation()
/external/armnn/src/backends/neon/
DNeonTensorHandleFactory.cpp52 …unique_ptr<ITensorHandle> NeonTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
58 …unique_ptr<ITensorHandle> NeonTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
73 …unique_ptr<ITensorHandle> NeonTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
/external/armnn/src/backends/aclCommon/
DArmComputeTensorUtils.cpp200 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo) in BuildArmComputeTensorInfo()
213 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo, in BuildArmComputeTensorInfo()
222 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo, unsigned int… in BuildArmComputeTensorInfo()
234 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo, in BuildArmComputeTensorInfo()
411 arm_compute::PixelValue GetPixelValue(const arm_compute::ITensorInfo* tensorInfo, float value) in GetPixelValue()
/external/armnn/src/armnnUtils/
DTensorUtils.cpp132 TensorInfo ReduceDims(const TensorInfo& tensorInfo, unsigned int dimensions) in ReduceDims()
262 void CheckSizes(const std::vector<PrimitiveType>& data, const armnn::TensorInfo& tensorInfo, unsign… in CheckSizes()
273 …<float[]> ToFloatArray(const std::vector<PrimitiveType>& data, const armnn::TensorInfo& tensorInfo) in ToFloatArray()
314 …ue_ptr<float[]> ToFloatArray(const std::vector<uint8_t>& data, const armnn::TensorInfo& tensorInfo) in ToFloatArray()
/external/armnn/include/armnn/backends/
DITensorHandleFactory.hpp67 virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
74 virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle()
DTensorHandle.hpp150 PassthroughTensorHandle(const TensorInfo& tensorInfo, void* mem) in PassthroughTensorHandle()
169 ConstPassthroughTensorHandle(const TensorInfo& tensorInfo, const void* mem) in ConstPassthroughTensorHandle()
/external/armnn/src/armnn/test/
DRuntimeTests.cpp85 TensorInfo tensorInfo{{4}, armnn::DataType::Signed32}; variable
190 TensorInfo tensorInfo{{4}, armnn::DataType::Float32, 0.0f, 0, true}; variable
1319 TensorInfo tensorInfo{{4}, armnn::DataType::Signed32}; variable
1364 TensorInfo tensorInfo{{4}, armnn::DataType::Signed32}; variable
1409 TensorInfo tensorInfo{{4}, armnn::DataType::Signed32}; variable
1454 TensorInfo tensorInfo{{4}, armnn::DataType::Signed32}; variable
1504 TensorInfo tensorInfo{ { 4 }, armnn::DataType::Signed32 }; variable

1234