Home
last modified time | relevance | path

Searched defs:tensorHandle (Results 1 – 25 of 43) sorted by relevance

12

/external/armnn/src/backends/cl/test/
DClImportTensorHandleFactoryTests.cpp37 auto tensorHandle = factory.CreateTensorHandle(tensorInfo); variable
63 auto tensorHandle = factory.CreateTensorHandle(tensorInfo); variable
82 auto tensorHandle = factory.CreateTensorHandle(tensorInfo); variable
101 auto tensorHandle = factory.CreateTensorHandle(tensorInfo); variable
119 auto tensorHandle = factory.CreateTensorHandle(tensorInfo); variable
/external/armnn/src/armnnTestUtils/
DTensorCopyUtils.cpp10 void CopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory) in CopyDataToITensorHandle()
15 void CopyDataFromITensorHandle(void* memory, const armnn::ITensorHandle* tensorHandle) in CopyDataFromITensorHandle()
20 void AllocateAndCopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory) in AllocateAndCopyDataToITensorHandle()
/external/armnn/include/armnnTestUtils/
DWorkloadTestUtils.hpp26 armnn::ITensorHandle* tensorHandle) in AddInputToWorkload()
36 armnn::ITensorHandle* tensorHandle) in AddOutputToWorkload()
47 armnn::ITensorHandle* tensorHandle) in SetWorkloadInput()
58 armnn::ITensorHandle* tensorHandle) in SetWorkloadOutput()
/external/armnn/src/dynamic/sample/
DSampleDynamicAdditionWorkload.cpp14 inline const armnn::TensorInfo& GetTensorInfo(const armnn::ITensorHandle* tensorHandle) in GetTensorInfo()
24 const armnn::ITensorHandle* tensorHandle = data.m_Inputs[idx]; in GetInputTensorData() local
30 armnn::ITensorHandle* tensorHandle = data.m_Outputs[idx]; in GetOutputTensorData() local
/external/armnn/src/backends/reference/workloads/
DRefWorkloadUtils.hpp27 inline const TensorInfo& GetTensorInfo(const ITensorHandle* tensorHandle) in GetTensorInfo()
38 const ITensorHandle* tensorHandle = data.m_Inputs[idx]; in GetInputTensorData() local
45 ITensorHandle* tensorHandle = data.m_Outputs[idx]; in GetOutputTensorData() local
50 DataType* GetOutputTensorData(ITensorHandle* tensorHandle) in GetOutputTensorData()
DRefBaseWorkload.hpp25 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle()
31 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle()
/external/armnn/src/backends/tosaReference/workloads/
DTosaRefPreCompiledWorkload.hpp35 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle()
40 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle()
DTosaRefBaseWorkload.hpp27 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle()
33 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle()
/external/armnn/src/backends/neon/workloads/
DNeonBaseWorkload.hpp21 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle()
38 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle()
DNeonFloorFloatWorkload.cpp36 void NeonFloorFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slo… in ReplaceInputTensorHandle()
53 void NeonFloorFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int sl… in ReplaceOutputTensorHandle()
DNeonConvertFp32ToFp16Workload.cpp44 void NeonConvertFp32ToFp16Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned … in ReplaceInputTensorHandle()
61 void NeonConvertFp32ToFp16Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceOutputTensorHandle()
DNeonConvertFp16ToFp32Workload.cpp43 void NeonConvertFp16ToFp32Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned … in ReplaceInputTensorHandle()
60 void NeonConvertFp16ToFp32Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceOutputTensorHandle()
DNeonL2NormalizationFloatWorkload.cpp63 void NeonL2NormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsign… in ReplaceInputTensorHandle()
80 void NeonL2NormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsig… in ReplaceOutputTensorHandle()
DNeonNormalizationFloatWorkload.cpp113 void NeonNormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceInputTensorHandle()
130 void NeonNormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigne… in ReplaceOutputTensorHandle()
/external/armnn/src/backends/cl/workloads/
DClBaseWorkload.hpp21 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle()
38 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle()
DClFloorFloatWorkload.cpp44 void ClFloorFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) in ReplaceInputTensorHandle()
61 void ClFloorFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) in ReplaceOutputTensorHandle()
DClConvertFp16ToFp32Workload.cpp64 void ClConvertFp16ToFp32Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned in… in ReplaceInputTensorHandle()
81 void ClConvertFp16ToFp32Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned i… in ReplaceOutputTensorHandle()
DClConvertFp32ToFp16Workload.cpp64 void ClConvertFp32ToFp16Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned in… in ReplaceInputTensorHandle()
81 void ClConvertFp32ToFp16Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned i… in ReplaceOutputTensorHandle()
DClNormalizationFloatWorkload.cpp65 void ClNormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned i… in ReplaceInputTensorHandle()
82 void ClNormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned … in ReplaceOutputTensorHandle()
DClL2NormalizationFloatWorkload.cpp63 void ClL2NormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceInputTensorHandle()
80 void ClL2NormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigne… in ReplaceOutputTensorHandle()
DClBatchNormalizationFloatWorkload.cpp127 void ClBatchNormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsig… in ReplaceInputTensorHandle()
144 void ClBatchNormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsi… in ReplaceOutputTensorHandle()
/external/armnn/src/backends/cl/
DClTensorHandleFactory.cpp71 std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo); in CreateTensorHandle() local
84 …std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLa… in CreateTensorHandle() local
DClImportTensorHandleFactory.cpp58 …std::unique_ptr<ClImportTensorHandle> tensorHandle = std::make_unique<ClImportTensorHandle>(tensor… in CreateTensorHandle() local
66 …std::unique_ptr<ClImportTensorHandle> tensorHandle = std::make_unique<ClImportTensorHandle>(tensor… in CreateTensorHandle() local
/external/armnn/src/armnn/
DLoadedNetwork.cpp618 auto tensorHandle = m_ConstantTensorHandles[pair.first]; in AllocateAndExecuteConstantWorkloads() local
638 ITensorHandle* tensorHandle = outSlot.GetOutputHandler().GetData(); in AllocateAndExecuteConstantWorkloadsAsync() local
802 std::unique_ptr<ITensorHandle> tensorHandle = in WorkloadData() local
813 std::unique_ptr<ITensorHandle> tensorHandle = in WorkloadData() local
1031 void LoadedNetwork::EnqueueInput(const BindableLayer& layer, ITensorHandle* tensorHandle, const Ten… in EnqueueInput()
1096 void LoadedNetwork::EnqueueOutput(const BindableLayer& layer, ITensorHandle* tensorHandle, const Te… in EnqueueOutput()
1311 std::unique_ptr<ITensorHandle> tensorHandle = in EnqueueInput() local
1332 std::unique_ptr<ITensorHandle> tensorHandle = in EnqueueInput() local
1356 std::unique_ptr<ITensorHandle> tensorHandle = in ImportOutputTensor() local
1384 std::unique_ptr<ITensorHandle> tensorHandle = in CopyToOutputTensor() local
[all …]
/external/armnn/src/backends/neon/
DNeonTensorHandleFactory.cpp61 auto tensorHandle = std::make_unique<NeonTensorHandle>(tensorInfo); in CreateTensorHandle() local
77 auto tensorHandle = std::make_unique<NeonTensorHandle>(tensorInfo, dataLayout); in CreateTensorHandle() local

12