| /external/armnn/src/armnnUtils/ |
| D | TensorUtils.cpp | 132 TensorInfo ReduceDims(const TensorInfo& tensorInfo, unsigned int dimensions) in ReduceDims() argument
134 TensorInfo strippedTensor(tensorInfo); in ReduceDims()
135 TensorShape strippedShape = ReduceDims(tensorInfo.GetShape(), dimensions); in ReduceDims()
262 void CheckSizes(const std::vector<PrimitiveType>& data, const armnn::TensorInfo& tensorInfo, unsign… in CheckSizes() argument
264 if (data.size() / size != tensorInfo.GetNumElements()) in CheckSizes()
268 data.size(), tensorInfo.GetNumElements(), CHECK_LOCATION().AsString())); in CheckSizes()
273 …<float[]> ToFloatArray(const std::vector<PrimitiveType>& data, const armnn::TensorInfo& tensorInfo) in ToFloatArray() argument
275 CheckSizes(data, tensorInfo); in ToFloatArray()
277 std::unique_ptr<float[]> returnBuffer(new float[tensorInfo.GetNumElements()]); in ToFloatArray()
279 if (tensorInfo.HasPerAxisQuantization()) in ToFloatArray()
[all …]
|
| /external/armnn/include/armnnTestUtils/ |
| D | DataLayoutUtils.hpp | 14 void PermuteTensorNchwToNhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNchwToNhwc() argument
18 tensorInfo = armnnUtils::Permuted(tensorInfo, nchwToNhwc); in PermuteTensorNchwToNhwc()
21 … armnnUtils::Permute(tensorInfo.GetShape(), nchwToNhwc, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNchwToNhwc()
26 void PermuteTensorNhwcToNchw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNhwcToNchw() argument
30 tensorInfo = armnnUtils::Permuted(tensorInfo, nhwcToNchw); in PermuteTensorNhwcToNchw()
33 … armnnUtils::Permute(tensorInfo.GetShape(), nhwcToNchw, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNhwcToNchw()
39 void PermuteTensorNdhwcToNcdhw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNdhwcToNcdhw() argument
43 tensorInfo = armnnUtils::Permuted(tensorInfo, ndhwcToNcdhw); in PermuteTensorNdhwcToNcdhw()
46 …armnnUtils::Permute(tensorInfo.GetShape(), ndhwcToNcdhw, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNdhwcToNcdhw()
51 void PermuteTensorNcdhwToNdhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNcdhwToNdhwc() argument
[all …]
|
| D | WorkloadTestUtils.hpp | 25 const armnn::TensorInfo& tensorInfo, in AddInputToWorkload() argument
29 info.m_InputTensorInfos.push_back(tensorInfo); in AddInputToWorkload()
35 const armnn::TensorInfo& tensorInfo, in AddOutputToWorkload() argument
39 info.m_OutputTensorInfos.push_back(tensorInfo); in AddOutputToWorkload()
46 const armnn::TensorInfo& tensorInfo, in SetWorkloadInput() argument
50 info.m_InputTensorInfos[index] = tensorInfo; in SetWorkloadInput()
57 const armnn::TensorInfo& tensorInfo, in SetWorkloadOutput() argument
61 info.m_OutputTensorInfos[index] = tensorInfo; in SetWorkloadOutput()
|
| D | MemCopyTestImpl.hpp | 25 const armnn::TensorInfo tensorInfo(tensorShape, dataType); in MemCopyTest() local
36 LayerTestResult<T, 4> ret(tensorInfo); in MemCopyTest()
39 std::vector<T> actualOutput(tensorInfo.GetNumElements()); in MemCopyTest()
42 auto inputTensorHandle = srcWorkloadFactory.CreateTensorHandle(tensorInfo); in MemCopyTest()
43 auto outputTensorHandle = dstWorkloadFactory.CreateTensorHandle(tensorInfo); in MemCopyTest()
63 AddInputToWorkload(memCopyQueueDesc, workloadInfo, tensorInfo, workloadInput.get()); in MemCopyTest()
64 AddOutputToWorkload(memCopyQueueDesc, workloadInfo, tensorInfo, workloadOutput.get()); in MemCopyTest()
|
| /external/armnn/src/backends/cl/test/ |
| D | ClImportTensorHandleFactoryTests.cpp | 21 TensorInfo tensorInfo; variable
24 REQUIRE_THROWS_AS(factory.CreateTensorHandle(tensorInfo, true), InvalidArgumentException);
25 …REQUIRE_THROWS_AS(factory.CreateTensorHandle(tensorInfo, DataLayout::NCHW, true), InvalidArgumentE…
34 TensorInfo tensorInfo(tensorShape, armnn::DataType::Float32);
37 auto tensorHandle = factory.CreateTensorHandle(tensorInfo);
43 tensorHandle = factory.CreateTensorHandle(tensorInfo, false);
49 tensorHandle = factory.CreateTensorHandle(tensorInfo, DataLayout::NHWC);
62 TensorInfo tensorInfo(tensorShape, armnn::DataType::Float32);
63 auto tensorHandle = factory.CreateTensorHandle(tensorInfo);
81 TensorInfo tensorInfo(tensorShape, armnn::DataType::Float32);
[all …]
|
| /external/armnn/src/backends/reference/test/ |
| D | RefPerChannelDecoderTests.cpp | 54 TensorInfo tensorInfo ({2,2,2,3},DataType::QSymmS8,{1.0f, 2.0f},0); variable
55 auto decoder = MakeDecoder<float>(tensorInfo, input.data());
57 std::vector<float> output = decoder->DecodeTensor(tensorInfo.GetShape());
80 TensorInfo tensorInfo ({4,1,2,2},DataType::QSymmS8,{1.0f, 2.0f, 3.0f, 4.0f},0); variable
81 auto decoder = MakeDecoder<float>(tensorInfo, input.data());
83 std::vector<float> output = decoder->DecodeTensor(tensorInfo.GetShape(), true);
113 TensorInfo tensorInfo ({6,1,2,2},DataType::QSymmS8,{1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f},0); variable
114 auto decoder = MakeDecoder<float>(tensorInfo, input.data());
116 std::vector<float> output = decoder->DecodeTensor(tensorInfo.GetShape(), true);
146 TensorInfo tensorInfo ({6,1,2,2},DataType::Signed32,{1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f},0); variable
[all …]
|
| D | RefPerAxisIteratorTests.cpp | 86 TensorInfo tensorInfo ({3,1,2,2},DataType::QSymmS8); variable
90 auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 0);
109 TensorInfo tensorInfo ({3,1,2,2},DataType::QSymmS8); variable
113 auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 1);
132 TensorInfo tensorInfo ({3,1,2,2},DataType::QSymmS8); variable
136 auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 2);
155 TensorInfo tensorInfo ({3,1,2,2},DataType::QSymmS8); variable
159 auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 3);
194 TensorInfo tensorInfo ({2,2,2,2},DataType::QSymmS8); variable
195 auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 1);
[all …]
|
| /external/armnn/src/armnnTestUtils/ |
| D | MockTensorHandleFactory.cpp | 27 …_ptr<ITensorHandle> MockTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo) const in CreateTensorHandle()
29 return std::make_unique<MockTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
32 …unique_ptr<ITensorHandle> MockTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
36 return std::make_unique<MockTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
39 …unique_ptr<ITensorHandle> MockTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
44 return std::make_unique<MockTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
48 return std::make_unique<MockTensorHandle>(tensorInfo, m_ImportFlags); in CreateTensorHandle()
52 …unique_ptr<ITensorHandle> MockTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
59 return std::make_unique<MockTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
63 return std::make_unique<MockTensorHandle>(tensorInfo, m_ImportFlags); in CreateTensorHandle()
|
| /external/armnn/src/backends/reference/ |
| D | RefTensorHandleFactory.cpp | 30 …e_ptr<ITensorHandle> RefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo) const in CreateTensorHandle()
32 return std::make_unique<RefTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
35 …:unique_ptr<ITensorHandle> RefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
39 return std::make_unique<RefTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
42 …:unique_ptr<ITensorHandle> RefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
47 return std::make_unique<RefTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
51 return std::make_unique<RefTensorHandle>(tensorInfo); in CreateTensorHandle()
55 …:unique_ptr<ITensorHandle> RefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
62 return std::make_unique<RefTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
66 return std::make_unique<RefTensorHandle>(tensorInfo); in CreateTensorHandle()
|
| /external/armnn/src/backends/tosaReference/ |
| D | TosaRefTensorHandleFactory.cpp | 31 …r<ITensorHandle> TosaRefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo) const in CreateTensorHandle()
33 return std::make_unique<TosaRefTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
36 …que_ptr<ITensorHandle> TosaRefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
40 return std::make_unique<TosaRefTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
43 …que_ptr<ITensorHandle> TosaRefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
48 return std::make_unique<TosaRefTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
52 return std::make_unique<TosaRefTensorHandle>(tensorInfo, m_ImportFlags); in CreateTensorHandle()
56 …que_ptr<ITensorHandle> TosaRefTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
63 return std::make_unique<TosaRefTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
67 return std::make_unique<TosaRefTensorHandle>(tensorInfo, m_ImportFlags); in CreateTensorHandle()
|
| D | TosaRefWorkloadFactory.cpp | 35 … auto checkType = [](const TensorInfo& tensorInfo) {return tensorInfo.GetDataType() == ArmnnType;}; in IsDataType() argument
79 …:unique_ptr<ITensorHandle> TosaRefWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
84 return std::make_unique<TosaRefTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
88 …return std::make_unique<TosaRefTensorHandle>(tensorInfo, static_cast<unsigned int>(MemorySource::M… in CreateTensorHandle()
92 …:unique_ptr<ITensorHandle> TosaRefWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
102 return std::make_unique<TosaRefTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
106 …return std::make_unique<TosaRefTensorHandle>(tensorInfo, static_cast<unsigned int>(MemorySource::M… in CreateTensorHandle()
|
| /external/armnn/src/armnn/optimizations/ |
| D | FoldPadIntoLayer2d.hpp | 21 inline float GetZeroElement(const TensorInfo& tensorInfo) in GetZeroElement() argument
23 return static_cast<float>(tensorInfo.IsQuantized() ? tensorInfo.GetQuantizationOffset() : 0); in GetZeroElement()
26 inline float GetLowestElement(const TensorInfo& tensorInfo) in GetLowestElement() argument
29 const float scale = tensorInfo.GetQuantizationScale(); in GetLowestElement()
30 const int32_t offset = tensorInfo.GetQuantizationOffset(); in GetLowestElement()
32 switch (tensorInfo.GetDataType()) in GetLowestElement()
56 inline bool IsNeutralElement(const Convolution2dDescriptor&, const TensorInfo& tensorInfo, const fl… in IsNeutralElement() argument
58 return tensorValue == GetZeroElement(tensorInfo); in IsNeutralElement()
62 const TensorInfo& tensorInfo, in IsNeutralElement() argument
65 return tensorValue == GetZeroElement(tensorInfo); in IsNeutralElement()
[all …]
|
| /external/armnn/src/dynamic/sample/ |
| D | SampleDynamicTensorHandleFactory.cpp | 32 const armnn::TensorInfo& tensorInfo) const in CreateTensorHandle()
34 return std::make_unique<SampleTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
38 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle() argument
42 return std::make_unique<SampleTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
46 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle() argument
51 return std::make_unique<SampleTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
55 return std::make_unique<SampleTensorHandle>(tensorInfo, m_ImportFlags); in CreateTensorHandle()
60 const armnn::TensorInfo& tensorInfo, in CreateTensorHandle() argument
67 return std::make_unique<SampleTensorHandle>(tensorInfo, m_MemoryManager); in CreateTensorHandle()
71 return std::make_unique<SampleTensorHandle>(tensorInfo, m_ImportFlags); in CreateTensorHandle()
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | FakeQuantizationTestImpl.cpp | 25 const armnn::TensorInfo tensorInfo({ height, width }, armnn::DataType::Float32); in FakeQuantizationTest() local
34 std::vector<float> actualOutput(tensorInfo.GetNumElements()); in FakeQuantizationTest()
35 std::vector<float> expectedOutput(tensorInfo.GetNumElements()); in FakeQuantizationTest()
37 …nique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(tensorInfo); in FakeQuantizationTest()
38 …nique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(tensorInfo); in FakeQuantizationTest()
43 AddInputToWorkload(data, info, tensorInfo, inputHandle.get()); in FakeQuantizationTest()
44 AddOutputToWorkload(data, info, tensorInfo, outputHandle.get()); in FakeQuantizationTest()
52 armnn::PassthroughTensorHandle refHandle(tensorInfo, expectedOutput.data()); in FakeQuantizationTest()
55 SetWorkloadOutput(refData, refInfo, 0, tensorInfo, &refHandle); in FakeQuantizationTest()
81 tensorInfo.GetShape()); in FakeQuantizationTest()
|
| D | BatchNormalizationTestImpl.cpp | 47 armnn::TensorInfo tensorInfo({ inputOutputTensorShape[dataLayoutIndexed.GetChannelsIndex()] }, in BatchNormTestImpl() local
57 tensorInfo.SetQuantizationScale(qScale); in BatchNormTestImpl()
58 tensorInfo.SetQuantizationOffset(qOffset); in BatchNormTestImpl()
75 armnn::ScopedTensorHandle meanTensor(tensorInfo); in BatchNormTestImpl()
76 armnn::ScopedTensorHandle varianceTensor(tensorInfo); in BatchNormTestImpl()
77 armnn::ScopedTensorHandle betaTensor(tensorInfo); in BatchNormTestImpl()
78 armnn::ScopedTensorHandle gammaTensor(tensorInfo); in BatchNormTestImpl()
132 armnn::TensorInfo tensorInfo({channels}, ArmnnType); in BatchNormTestNhwcImpl() local
141 tensorInfo.SetQuantizationScale(qScale); in BatchNormTestNhwcImpl()
142 tensorInfo.SetQuantizationOffset(qOffset); in BatchNormTestNhwcImpl()
[all …]
|
| /external/android-nn-driver/ |
| D | ConversionUtils.cpp | 21 …tHandle::LayerInputHandle(bool valid, armnn::IOutputSlot* outputSlot, armnn::TensorInfo tensorInfo) in LayerInputHandle() argument
24 , m_TensorInfo(tensorInfo) in LayerInputHandle()
82 ConstTensorPin::ConstTensorPin(armnn::TensorInfo& tensorInfo, in ConstTensorPin() argument
89 if (tensorInfo.GetNumBytes() != numBytes) in ConstTensorPin()
97 m_SwizzledTensorData.resize(tensorInfo.GetNumBytes()); in ConstTensorPin()
98 … SwizzleAndroidNn4dTensorToArmNn(tensorInfo, valueStart, m_SwizzledTensorData.data(), mappings); in ConstTensorPin()
100 m_ConstTensor = armnn::ConstTensor(tensorInfo, m_SwizzledTensorData.data()); in ConstTensorPin()
104 m_ConstTensor = armnn::ConstTensor(tensorInfo, valueStart); in ConstTensorPin()
137 armnn::IConnectableLayer* ProcessActivation(const armnn::TensorInfo& tensorInfo, in ProcessActivation() argument
147 prevLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo); in ProcessActivation()
[all …]
|
| /external/armnn/src/backends/backendsCommon/ |
| D | TensorHandle.cpp | 15 TensorShape GetUnpaddedTensorStrides(const TensorInfo& tensorInfo) in GetUnpaddedTensorStrides() argument
17 TensorShape shape(tensorInfo.GetShape()); in GetUnpaddedTensorStrides()
18 auto size = GetDataTypeSize(tensorInfo.GetDataType()); in GetUnpaddedTensorStrides()
31 ConstTensorHandle::ConstTensorHandle(const TensorInfo& tensorInfo) in ConstTensorHandle() argument
32 : m_TensorInfo(tensorInfo) in ConstTensorHandle()
43 TensorHandle::TensorHandle(const TensorInfo& tensorInfo) in TensorHandle() argument
44 : ConstTensorHandle(tensorInfo) in TensorHandle()
55 ScopedTensorHandle::ScopedTensorHandle(const TensorInfo& tensorInfo) in ScopedTensorHandle() argument
56 : TensorHandle(tensorInfo) in ScopedTensorHandle()
|
| /external/armnn/src/backends/tosaCommon/operatorMappings/ |
| D | TosaOperatorUtils.hpp | 284 auto tensorInfo = tensorHandle->GetTensorInfo(); in ConvertConstantTensorDataToBuffer() local
286 switch (tensorInfo.GetDataType()) in ConvertConstantTensorDataToBuffer()
290 std::vector<float> data(tensorInfo.GetNumElements()); in ConvertConstantTensorDataToBuffer()
291 memcpy(data.data(), tensorHandle->Map(true), tensorInfo.GetNumBytes()); in ConvertConstantTensorDataToBuffer()
298 std::vector<float> data(tensorInfo.GetNumElements()); in ConvertConstantTensorDataToBuffer()
299 memcpy(data.data(), tensorHandle->Map(true), tensorInfo.GetNumBytes()); in ConvertConstantTensorDataToBuffer()
307 std::vector<int8_t> data(tensorInfo.GetNumElements()); in ConvertConstantTensorDataToBuffer()
308 memcpy(data.data(), tensorHandle->Map(true), tensorInfo.GetNumBytes()); in ConvertConstantTensorDataToBuffer()
315 memcpy(uint8Data.data(), tensorHandle->Map(true), tensorInfo.GetNumBytes()); in ConvertConstantTensorDataToBuffer()
320 std::vector<int16_t> data(tensorInfo.GetNumElements()); in ConvertConstantTensorDataToBuffer()
[all …]
|
| /external/armnn/src/backends/aclCommon/ |
| D | ArmComputeTensorUtils.cpp | 200 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo) in BuildArmComputeTensorInfo() argument
202 bool multiScales = tensorInfo.HasMultipleQuantizationScales(); in BuildArmComputeTensorInfo()
203 … const arm_compute::TensorShape aclTensorShape = BuildArmComputeTensorShape(tensorInfo.GetShape()); in BuildArmComputeTensorInfo()
204 …const arm_compute::DataType aclDataType = GetArmComputeDataType(tensorInfo.GetDataType(), mu… in BuildArmComputeTensorInfo()
207 arm_compute::QuantizationInfo(tensorInfo.GetQuantizationScales()) : in BuildArmComputeTensorInfo()
208 …arm_compute::QuantizationInfo(tensorInfo.GetQuantizationScale(), tensorInfo.GetQuantizationOffset(… in BuildArmComputeTensorInfo()
213 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo, in BuildArmComputeTensorInfo() argument
216 arm_compute::TensorInfo aclTensorInfo = BuildArmComputeTensorInfo(tensorInfo); in BuildArmComputeTensorInfo()
222 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo, unsigned int… in BuildArmComputeTensorInfo() argument
224 bool multiScales = tensorInfo.HasMultipleQuantizationScales(); in BuildArmComputeTensorInfo()
[all …]
|
| /external/armnn/src/backends/cl/ |
| D | ClTensorHandleFactory.cpp | 57 …ue_ptr<ITensorHandle> ClTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo) const in CreateTensorHandle()
59 return ClTensorHandleFactory::CreateTensorHandle(tensorInfo, true); in CreateTensorHandle()
62 …::unique_ptr<ITensorHandle> ClTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
65 return ClTensorHandleFactory::CreateTensorHandle(tensorInfo, dataLayout, true); in CreateTensorHandle()
68 …::unique_ptr<ITensorHandle> ClTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
71 std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo); in CreateTensorHandle()
80 …::unique_ptr<ITensorHandle> ClTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
84 …std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLa… in CreateTensorHandle()
|
| D | ClImportTensorHandleFactory.cpp | 56 …<ITensorHandle> ClImportTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo) const in CreateTensorHandle()
58 …:unique_ptr<ClImportTensorHandle> tensorHandle = std::make_unique<ClImportTensorHandle>(tensorInfo, in CreateTensorHandle()
63 …ue_ptr<ITensorHandle> ClImportTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
66 …:unique_ptr<ClImportTensorHandle> tensorHandle = std::make_unique<ClImportTensorHandle>(tensorInfo, in CreateTensorHandle()
72 …ue_ptr<ITensorHandle> ClImportTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
79 return CreateTensorHandle(tensorInfo); in CreateTensorHandle()
82 …ue_ptr<ITensorHandle> ClImportTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
90 return CreateTensorHandle(tensorInfo, dataLayout); in CreateTensorHandle()
|
| /external/armnn/include/armnn/backends/ |
| D | TensorHandle.hpp | 21 TensorShape GetUnpaddedTensorStrides(const TensorInfo& tensorInfo);
59 ConstTensorHandle(const TensorInfo& tensorInfo);
96 TensorHandle(const TensorInfo& tensorInfo);
118 explicit ScopedTensorHandle(const TensorInfo& tensorInfo);
150 PassthroughTensorHandle(const TensorInfo& tensorInfo, void* mem) in PassthroughTensorHandle() argument
151 : TensorHandle(tensorInfo) in PassthroughTensorHandle()
169 ConstPassthroughTensorHandle(const TensorInfo& tensorInfo, const void* mem) in ConstPassthroughTensorHandle() argument
170 : ConstTensorHandle(tensorInfo) in ConstPassthroughTensorHandle()
|
| D | ITensorHandleFactory.hpp | 59 … virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo) const = 0;
61 virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo,
67 virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
71 return CreateTensorHandle(tensorInfo); in CreateTensorHandle()
74 virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
79 return CreateTensorHandle(tensorInfo, dataLayout); in CreateTensorHandle()
|
| /external/armnn/src/backends/neon/ |
| D | NeonTensorHandleFactory.cpp | 47 …_ptr<ITensorHandle> NeonTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo) const in CreateTensorHandle()
49 return NeonTensorHandleFactory::CreateTensorHandle(tensorInfo, true); in CreateTensorHandle()
52 …unique_ptr<ITensorHandle> NeonTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
55 return NeonTensorHandleFactory::CreateTensorHandle(tensorInfo, dataLayout, true); in CreateTensorHandle()
58 …unique_ptr<ITensorHandle> NeonTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
61 auto tensorHandle = std::make_unique<NeonTensorHandle>(tensorInfo); in CreateTensorHandle()
73 …unique_ptr<ITensorHandle> NeonTensorHandleFactory::CreateTensorHandle(const TensorInfo& tensorInfo, in CreateTensorHandle() argument
77 auto tensorHandle = std::make_unique<NeonTensorHandle>(tensorInfo, dataLayout); in CreateTensorHandle()
|
| /external/armnn/samples/ |
| D | PreImportMemorySample.cpp | 40 TensorInfo tensorInfo{{4}, armnn::DataType::Float32}; in main() local
43 inputLayer1->GetOutputSlot(0).SetTensorInfo(tensorInfo); in main()
45 inputLayer2->GetOutputSlot(0).SetTensorInfo(tensorInfo); in main()
48 addLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo); in main()
68 ConstTensor inputTensor1(tensorInfo, inputData1.data()); in main()
69 ConstTensor inputTensor2(tensorInfo, inputData2.data()); in main()
72 Tensor outputTensor1{tensorInfo, outputData1.data()}; in main()
|