Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/armnn/src/backends/cl/test/
H A 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
/aosp_15_r20/external/armnn/src/armnn/
H A 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 …]
H A DGraph.cpp204 { in AllocateDynamicBuffers()
216 … ITensorHandle *tensorHandle = TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()); in AllocateDynamicBuffers() local
234 … ITensorHandle *tensorHandle = TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()); in AllocateDynamicBuffers() local
260 ITensorHandle *tensorHandle = TraceSubTensorHandleAncestry( in AllocateDynamicBuffers() local
/aosp_15_r20/external/armnn/include/armnnTestUtils/
H A 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()
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A 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()
H A DRefBaseWorkload.hpp25 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle()
31 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle()
/aosp_15_r20/external/armnn/src/armnnTestUtils/
H A 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()
/aosp_15_r20/external/armnn/src/dynamic/sample/
H A 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
/aosp_15_r20/external/armnn/src/backends/tosaReference/workloads/
H A DTosaRefPreCompiledWorkload.hpp35 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle()
40 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle()
H A DTosaRefBaseWorkload.hpp27 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle()
33 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle()
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonBaseWorkload.hpp21 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle()
38 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle()
H A DNeonFloorFloatWorkload.cpp36 void NeonFloorFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slo… in ReplaceInputTensorHandle()
53 void NeonFloorFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int sl… in ReplaceOutputTensorHandle()
H A DNeonConvertFp32ToFp16Workload.cpp44 void NeonConvertFp32ToFp16Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned … in ReplaceInputTensorHandle()
61 void NeonConvertFp32ToFp16Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceOutputTensorHandle()
H A DNeonConvertFp16ToFp32Workload.cpp43 void NeonConvertFp16ToFp32Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned … in ReplaceInputTensorHandle()
60 void NeonConvertFp16ToFp32Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceOutputTensorHandle()
H A DNeonL2NormalizationFloatWorkload.cpp63 void NeonL2NormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsign… in ReplaceInputTensorHandle()
80 void NeonL2NormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsig… in ReplaceOutputTensorHandle()
H A DNeonNormalizationFloatWorkload.cpp113 void NeonNormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceInputTensorHandle()
130 void NeonNormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigne… in ReplaceOutputTensorHandle()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClBaseWorkload.hpp21 void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceInputTensorHandle()
38 void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override in ReplaceOutputTensorHandle()
H A DClFloorFloatWorkload.cpp44 void ClFloorFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) in ReplaceInputTensorHandle()
61 void ClFloorFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) in ReplaceOutputTensorHandle()
H A DClNormalizationFloatWorkload.cpp65 void ClNormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned i… in ReplaceInputTensorHandle()
82 void ClNormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned … in ReplaceOutputTensorHandle()
H A DClConvertFp32ToFp16Workload.cpp64 void ClConvertFp32ToFp16Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned in… in ReplaceInputTensorHandle()
81 void ClConvertFp32ToFp16Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned i… in ReplaceOutputTensorHandle()
H A DClConvertFp16ToFp32Workload.cpp64 void ClConvertFp16ToFp32Workload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned in… in ReplaceInputTensorHandle()
81 void ClConvertFp16ToFp32Workload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned i… in ReplaceOutputTensorHandle()
H A DClL2NormalizationFloatWorkload.cpp63 void ClL2NormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned… in ReplaceInputTensorHandle()
80 void ClL2NormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigne… in ReplaceOutputTensorHandle()
H A DClBatchNormalizationFloatWorkload.cpp127 void ClBatchNormalizationFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsig… in ReplaceInputTensorHandle()
144 void ClBatchNormalizationFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsi… in ReplaceOutputTensorHandle()
/aosp_15_r20/external/armnn/src/backends/cl/
H A 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
H A 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

12