| /external/armnn/src/backends/cl/test/ |
| D | ClImportTensorHandleFactoryTests.cpp | 37 auto tensorHandle = factory.CreateTensorHandle(tensorInfo); variable
38 ARMNN_ASSERT(tensorHandle);
39 …ARMNN_ASSERT(tensorHandle->GetImportFlags() == static_cast<MemorySourceFlags>(MemorySource::Malloc…
40 ARMNN_ASSERT(tensorHandle->GetShape() == tensorShape);
43 tensorHandle = factory.CreateTensorHandle(tensorInfo, false);
44 CHECK(tensorHandle);
45 …ARMNN_ASSERT(tensorHandle->GetImportFlags() == static_cast<MemorySourceFlags>(MemorySource::Malloc…
46 ARMNN_ASSERT(tensorHandle->GetShape() == tensorShape);
49 tensorHandle = factory.CreateTensorHandle(tensorInfo, DataLayout::NHWC);
50 CHECK(tensorHandle);
[all …]
|
| /external/armnn/src/armnnTestUtils/ |
| D | TensorCopyUtils.cpp | 10 void CopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory) in CopyDataToITensorHandle() argument
12 tensorHandle->CopyInFrom(memory); in CopyDataToITensorHandle()
15 void CopyDataFromITensorHandle(void* memory, const armnn::ITensorHandle* tensorHandle) in CopyDataFromITensorHandle() argument
17 tensorHandle->CopyOutTo(memory); in CopyDataFromITensorHandle()
20 void AllocateAndCopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory) in AllocateAndCopyDataToITensorHandle() argument
22 tensorHandle->Allocate(); in AllocateAndCopyDataToITensorHandle()
23 CopyDataToITensorHandle(tensorHandle, memory); in AllocateAndCopyDataToITensorHandle()
|
| /external/armnn/src/backends/neon/ |
| D | NeonTensorHandleFactory.cpp | 61 auto tensorHandle = std::make_unique<NeonTensorHandle>(tensorInfo); in CreateTensorHandle() local
64 tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup()); in CreateTensorHandle()
67 tensorHandle->SetImportEnabledFlag(!IsMemoryManaged); in CreateTensorHandle()
68 tensorHandle->SetImportFlags(GetImportFlags()); in CreateTensorHandle()
70 return tensorHandle; in CreateTensorHandle()
77 auto tensorHandle = std::make_unique<NeonTensorHandle>(tensorInfo, dataLayout); in CreateTensorHandle() local
80 tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup()); in CreateTensorHandle()
83 tensorHandle->SetImportEnabledFlag(!IsMemoryManaged); in CreateTensorHandle()
84 tensorHandle->SetImportFlags(GetImportFlags()); in CreateTensorHandle()
86 return tensorHandle; in CreateTensorHandle()
|
| /external/armnn/include/armnnTestUtils/ |
| D | WorkloadTestUtils.hpp | 26 armnn::ITensorHandle* tensorHandle) in AddInputToWorkload() argument
28 descriptor.m_Inputs.push_back(tensorHandle); in AddInputToWorkload()
36 armnn::ITensorHandle* tensorHandle) in AddOutputToWorkload() argument
38 descriptor.m_Outputs.push_back(tensorHandle); in AddOutputToWorkload()
47 armnn::ITensorHandle* tensorHandle) in SetWorkloadInput() argument
49 descriptor.m_Inputs[index] = tensorHandle; in SetWorkloadInput()
58 armnn::ITensorHandle* tensorHandle) in SetWorkloadOutput() argument
60 descriptor.m_Outputs[index] = tensorHandle; in SetWorkloadOutput()
|
| D | TensorCopyUtils.hpp | 12 void CopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory);
14 void CopyDataFromITensorHandle(void* mem, const armnn::ITensorHandle* tensorHandle);
16 void AllocateAndCopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory);
|
| /external/armnn/src/armnn/ |
| D | Graph.cpp | 203 auto IsPreallocated = [&](ITensorHandle* const tensorHandle) in AllocateDynamicBuffers()
205 return tensorHandle && preallocatedTensors.find(tensorHandle) != preallocatedTensors.end(); in AllocateDynamicBuffers()
216 … ITensorHandle *tensorHandle = TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()); in AllocateDynamicBuffers() local
218 if (tensorHandle && !IsPreallocated(tensorHandle)) in AllocateDynamicBuffers()
220 tensorHandle->Allocate(); in AllocateDynamicBuffers()
221 preallocatedTensors.insert(tensorHandle); in AllocateDynamicBuffers()
234 … ITensorHandle *tensorHandle = TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()); in AllocateDynamicBuffers() local
236 if (tensorHandle && !IsPreallocated(tensorHandle)) in AllocateDynamicBuffers()
239 if (handleReferenceCounts.find(tensorHandle) == handleReferenceCounts.end()) in AllocateDynamicBuffers()
241 handleReferenceCounts[tensorHandle] = numConnections; in AllocateDynamicBuffers()
[all …]
|
| D | LoadedNetwork.cpp | 618 auto tensorHandle = m_ConstantTensorHandles[pair.first]; in AllocateAndExecuteConstantWorkloads() local
619 tensorHandle->Allocate(); in AllocateAndExecuteConstantWorkloads()
638 ITensorHandle* tensorHandle = outSlot.GetOutputHandler().GetData(); in AllocateAndExecuteConstantWorkloadsAsync() local
640 m_ConstantTensorHandles[layer->GetGuid()] = tensorHandle; in AllocateAndExecuteConstantWorkloadsAsync()
641 tensorHandle->Allocate(); in AllocateAndExecuteConstantWorkloadsAsync()
646 memDesc.m_Outputs.push_back(tensorHandle); in AllocateAndExecuteConstantWorkloadsAsync()
802 std::unique_ptr<ITensorHandle> tensorHandle = in WorkloadData() local
806 m_InputTensorPins.emplace_back(std::move(tensorHandle), inputTensor.GetInfo(), layerId); in WorkloadData()
813 std::unique_ptr<ITensorHandle> tensorHandle = in WorkloadData() local
817 … m_OutputTensorPins.emplace_back(std::move(tensorHandle), outputTensor.GetInfo(), layerId); in WorkloadData()
[all …]
|
| /external/armnn/src/dynamic/sample/ |
| D | SampleDynamicAdditionWorkload.cpp | 14 inline const armnn::TensorInfo& GetTensorInfo(const armnn::ITensorHandle* tensorHandle) in GetTensorInfo() argument
18 static_cast<const SampleTensorHandle*>(tensorHandle); in GetTensorInfo()
24 const armnn::ITensorHandle* tensorHandle = data.m_Inputs[idx]; in GetInputTensorData() local
25 return reinterpret_cast<const float*>(tensorHandle->Map()); in GetInputTensorData()
30 armnn::ITensorHandle* tensorHandle = data.m_Outputs[idx]; in GetOutputTensorData() local
31 return reinterpret_cast<float*>(tensorHandle->Map()); in GetOutputTensorData()
|
| /external/armnn/src/backends/reference/workloads/ |
| D | RefWorkloadUtils.hpp | 27 inline const TensorInfo& GetTensorInfo(const ITensorHandle* tensorHandle) in GetTensorInfo() argument
31 PolymorphicDowncast<const TensorHandleType*>(tensorHandle); in GetTensorInfo()
38 const ITensorHandle* tensorHandle = data.m_Inputs[idx]; in GetInputTensorData() local
39 return reinterpret_cast<const DataType*>(tensorHandle->Map()); in GetInputTensorData()
45 ITensorHandle* tensorHandle = data.m_Outputs[idx]; in GetOutputTensorData() local
46 return reinterpret_cast<DataType*>(tensorHandle->Map()); in GetOutputTensorData()
50 DataType* GetOutputTensorData(ITensorHandle* tensorHandle) in GetOutputTensorData() argument
52 return reinterpret_cast<DataType*>(tensorHandle->Map()); in GetOutputTensorData()
|
| D | RefBaseWorkload.hpp | 25 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle() argument
27 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
31 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle() argument
33 this->m_Data.m_Outputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| /external/armnn/src/backends/tosaCommon/operatorMappings/ |
| D | TosaOperatorUtils.hpp | 280 …<uint8_t> ConvertConstantTensorDataToBuffer(const std::shared_ptr<ConstTensorHandle>& tensorHandle) in ConvertConstantTensorDataToBuffer() argument
284 auto tensorInfo = tensorHandle->GetTensorInfo(); in ConvertConstantTensorDataToBuffer()
291 memcpy(data.data(), tensorHandle->Map(true), tensorInfo.GetNumBytes()); in ConvertConstantTensorDataToBuffer()
299 memcpy(data.data(), tensorHandle->Map(true), tensorInfo.GetNumBytes()); in ConvertConstantTensorDataToBuffer()
308 memcpy(data.data(), tensorHandle->Map(true), tensorInfo.GetNumBytes()); in ConvertConstantTensorDataToBuffer()
315 memcpy(uint8Data.data(), tensorHandle->Map(true), tensorInfo.GetNumBytes()); in ConvertConstantTensorDataToBuffer()
321 memcpy(data.data(), tensorHandle->Map(true), tensorInfo.GetNumBytes()); in ConvertConstantTensorDataToBuffer()
329 memcpy(data.data(), tensorHandle->Map(true), tensorInfo.GetNumBytes()); in ConvertConstantTensorDataToBuffer()
345 tensorHandle->Unmap(); in ConvertConstantTensorDataToBuffer()
|
| /external/armnn/src/backends/tosaReference/workloads/ |
| D | TosaRefBaseWorkload.hpp | 27 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle() argument
29 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
33 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle() argument
35 this->m_Data.m_Outputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| D | TosaRefPreCompiledWorkload.hpp | 35 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle() argument
37 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
40 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle() argument
42 this->m_Data.m_Outputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| /external/armnn/src/backends/cl/ |
| D | ClTensorHandleFactory.cpp | 71 std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo); in CreateTensorHandle() local
76 tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup()); in CreateTensorHandle()
77 return tensorHandle; in CreateTensorHandle()
84 …std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLa… in CreateTensorHandle() local
89 tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup()); in CreateTensorHandle()
90 return tensorHandle; in CreateTensorHandle()
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonBaseWorkload.hpp | 21 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle() argument
24 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
38 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle() argument
41 this->m_Data.m_Outputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| D | NeonFloorFloatWorkload.cpp | 36 void NeonFloorFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slo… in ReplaceInputTensorHandle() argument
39 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
53 void NeonFloorFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int sl… in ReplaceOutputTensorHandle() argument
56 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| D | NeonConvertFp32ToFp16Workload.cpp | 44 void NeonConvertFp32ToFp16Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned … in ReplaceInputTensorHandle() argument
47 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
61 void NeonConvertFp32ToFp16Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceOutputTensorHandle() argument
64 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| D | NeonConvertFp16ToFp32Workload.cpp | 43 void NeonConvertFp16ToFp32Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned … in ReplaceInputTensorHandle() argument
46 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
60 void NeonConvertFp16ToFp32Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceOutputTensorHandle() argument
63 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| D | NeonL2NormalizationFloatWorkload.cpp | 63 void NeonL2NormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsign… in ReplaceInputTensorHandle() argument
66 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
80 void NeonL2NormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsig… in ReplaceOutputTensorHandle() argument
83 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClBaseWorkload.hpp | 21 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle() argument
24 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
38 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle() argument
41 this->m_Data.m_Outputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| D | ClFloorFloatWorkload.cpp | 44 void ClFloorFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) in ReplaceInputTensorHandle() argument
47 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
61 void ClFloorFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) in ReplaceOutputTensorHandle() argument
64 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| D | ClNormalizationFloatWorkload.cpp | 65 void ClNormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned i… in ReplaceInputTensorHandle() argument
68 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
82 void ClNormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned … in ReplaceOutputTensorHandle() argument
85 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| D | ClL2NormalizationFloatWorkload.cpp | 63 void ClL2NormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceInputTensorHandle() argument
66 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
80 void ClL2NormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigne… in ReplaceOutputTensorHandle() argument
83 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| D | ClConvertFp16ToFp32Workload.cpp | 64 void ClConvertFp16ToFp32Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned in… in ReplaceInputTensorHandle() argument
67 this->m_Data.m_Inputs[slot] = tensorHandle; in ReplaceInputTensorHandle()
81 void ClConvertFp16ToFp32Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned i… in ReplaceOutputTensorHandle() argument
84 this->m_Data.m_Outputs[slot] = tensorHandle; in ReplaceOutputTensorHandle()
|
| /external/armnn/src/backends/backendsCommon/ |
| D | TensorHandle.cpp | 66 ScopedTensorHandle::ScopedTensorHandle(const ConstTensorHandle& tensorHandle) in ScopedTensorHandle() argument
67 : ScopedTensorHandle(tensorHandle.GetTensorInfo()) in ScopedTensorHandle()
69 CopyFrom(tensorHandle.GetConstTensor<void>(), tensorHandle.GetTensorInfo().GetNumBytes()); in ScopedTensorHandle()
|