Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/armnn/src/armnn/optimizations/
H A 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()
/aosp_15_r20/external/armnn/src/backends/reference/test/
H A 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
H A 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
H A DRefEndToEndTests.cpp197 TensorInfo tensorInfo(TensorShape({3, 4}), DataType::Float32); variable
296 TensorInfo tensorInfo(TensorShape({ 10 }), DataType::Float32); variable
359 TensorInfo tensorInfo(TensorShape({1, 1, 1, 4}), DataType::Float32); variable
/aosp_15_r20/external/armnn/include/armnnTestUtils/
H A 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()
H A 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()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A 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()
/aosp_15_r20/external/armnn/src/backends/aclCommon/
H A 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()
/aosp_15_r20/external/armnn/src/armnn/test/
H A 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
/aosp_15_r20/external/armnn/src/armnnTestUtils/
H A DCreateWorkload.hpp82 armnn::TensorInfo tensorInfo({1, 1}, DataType); in CreateActivationWorkloadTest() local
119 armnn::TensorInfo tensorInfo({2, 3}, DataType); in CreateElementwiseWorkloadTest() local
153 armnn::TensorInfo tensorInfo({2, 3}, DataType); in CreateElementwiseBinaryWorkloadTest() local
192 armnn::TensorInfo tensorInfo({2, 3}, DataType); in CreateSubtractionWithBlobWorkloadTest() local
251 armnn::TensorInfo tensorInfo({2, 3}, DataType); in CreateMultiplicationWithBlobWorkloadTest() local
307 armnn::TensorInfo tensorInfo({2, 3}, DataType); in CreateAdditionWithBlobWorkloadTest() local
354 armnn::TensorInfo tensorInfo({ 2, 3 }, DataType); in CreateElementwiseUnaryWorkloadTest() local
405 armnn::TensorInfo tensorInfo(tensorShape, DataType); in CreateBatchNormalizationWorkloadTest() local
478 armnn::TensorInfo tensorInfo(tensorShape, DataType); in CreateBatchNormalizationWithBlobWorkloadTest() local
1539 armnn::TensorInfo tensorInfo({4, 1}, DataType); in CreateSoftmaxWorkloadTest() local
[all …]
H A 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()
H A DMockTensorHandle.cpp11 MockTensorHandle::MockTensorHandle(const TensorInfo& tensorInfo, std::shared_ptr<MockMemoryManager>… in MockTensorHandle()
21 MockTensorHandle::MockTensorHandle(const TensorInfo& tensorInfo, MemorySourceFlags importFlags) in MockTensorHandle()
/aosp_15_r20/external/armnn/src/dynamic/sample/
H A DSampleDynamicTensorHandleFactory.cpp38 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle()
46 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle()
60 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle()
H A DSampleTensorHandle.cpp11 SampleTensorHandle::SampleTensorHandle(const armnn::TensorInfo &tensorInfo, in SampleTensorHandle()
23 SampleTensorHandle::SampleTensorHandle(const armnn::TensorInfo& tensorInfo, in SampleTensorHandle()
/aosp_15_r20/external/armnn/src/backends/reference/
H A 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()
H A DRefTensorHandle.cpp10 RefTensorHandle::RefTensorHandle(const TensorInfo &tensorInfo, std::shared_ptr<RefMemoryManager> &m… in RefTensorHandle()
20 RefTensorHandle::RefTensorHandle(const TensorInfo& tensorInfo) in RefTensorHandle()
/aosp_15_r20/external/armnn/src/backends/tosaReference/
H A 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()
H A 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()
H A DTosaRefTensorHandle.cpp10 TosaRefTensorHandle::TosaRefTensorHandle(const TensorInfo& tensorInfo, in TosaRefTensorHandle()
21 TosaRefTensorHandle::TosaRefTensorHandle(const TensorInfo& tensorInfo, in TosaRefTensorHandle()
/aosp_15_r20/external/armnn/src/armnnUtils/
H A 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()
/aosp_15_r20/external/armnn/src/backends/cl/
H A 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()
H A 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()
/aosp_15_r20/external/android-nn-driver/
H A 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()
/aosp_15_r20/external/armnn/src/backends/neon/
H A 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()
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DConversionUtils.cpp21 …tHandle::LayerInputHandle(bool valid, armnn::IOutputSlot* outputSlot, armnn::TensorInfo tensorInfo) in LayerInputHandle()
79 ConstTensorPin::ConstTensorPin(armnn::TensorInfo& tensorInfo, in ConstTensorPin()
187 armnn::TensorInfo tensorInfo = GetTensorInfoForOperand(operand); in ConvertOperandToConstTensorPin() local
719 armnn::TensorInfo tensorInfo(operand->dimensions.size(), in DequantizeIfRequired() local
885 armnn::IConnectableLayer* ProcessActivation(const armnn::TensorInfo& tensorInfo, in ProcessActivation()

1234