Home
last modified time | relevance | path

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

/external/armnn/src/backends/backendsCommon/test/
DTensorCopyUtils.cpp9 void CopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory) in CopyDataToITensorHandle()
14 void CopyDataFromITensorHandle(void* memory, const armnn::ITensorHandle* tensorHandle) in CopyDataFromITensorHandle()
19 void AllocateAndCopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory) in AllocateAndCopyDataToITensorHandle()
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.hpp26 inline const TensorInfo& GetTensorInfo(const ITensorHandle* tensorHandle) in GetTensorInfo()
37 const ITensorHandle* tensorHandle = data.m_Inputs[idx]; in GetInputTensorData() local
44 ITensorHandle* tensorHandle = data.m_Outputs[idx]; in GetOutputTensorData() local
/external/armnn/src/backends/cl/
DClTensorHandleFactory.cpp75 std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo); in CreateTensorHandle() local
88 …std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLa… in CreateTensorHandle() local
DClWorkloadFactory.cpp147 std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo); in CreateTensorHandle() local
158 …std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLa… in CreateTensorHandle() local
/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
DNeonWorkloadFactory.cpp93 auto tensorHandle = std::make_unique<NeonTensorHandle>(tensorInfo); in CreateTensorHandle() local
105 auto tensorHandle = std::make_unique<NeonTensorHandle>(tensorInfo, dataLayout); in CreateTensorHandle() local
/external/armnn/src/armnn/
DGraph.cpp201 { in AllocateDynamicBuffers()
213 … ITensorHandle *tensorHandle = TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()); in AllocateDynamicBuffers() local
231 … ITensorHandle *tensorHandle = TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()); in AllocateDynamicBuffers() local
257 ITensorHandle *tensorHandle = TraceSubTensorHandleAncestry( in AllocateDynamicBuffers() local
DLoadedNetwork.cpp433 std::unique_ptr<ITensorHandle> tensorHandle = in WorkloadData() local
444 std::unique_ptr<ITensorHandle> tensorHandle = in WorkloadData() local
550 void LoadedNetwork::EnqueueInput(const BindableLayer& layer, ITensorHandle* tensorHandle, const Ten… in EnqueueInput()
615 void LoadedNetwork::EnqueueOutput(const BindableLayer& layer, ITensorHandle* tensorHandle, const Te… in EnqueueOutput()
DNetworkQuantizer.cpp98 …o rangeTrackerCallback = [&](LayerGuid guid, unsigned int slotIndex, ITensorHandle *tensorHandle) { in Refine()
/external/armnn/src/backends/aclCommon/test/
DCreateWorkloadClNeon.hpp30 …st_tools::predicate_result CompareTensorHandleShape(IComputeTensorHandle* tensorHand… in CompareTensorHandleShape()
/external/armnn/src/backends/backendsCommon/
DCpuTensorHandle.cpp66 ScopedCpuTensorHandle::ScopedCpuTensorHandle(const ConstCpuTensorHandle& tensorHandle) in ScopedCpuTensorHandle()
/external/armnn/src/armnnUtils/
DTensorUtils.cpp58 std::pair<float, float> FindMinMax(ITensorHandle* tensorHandle) in FindMinMax()
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DEagerOperationBuilder.java226 private static native void addInput(long opHandle, long tensorHandle); in addInput()
252 private static native void setAttrTensor(long opHandle, String name, long tensorHandle); in setAttrTensor()
DGraphOperationBuilder.java334 private static native void setAttrTensor(long handle, String name, long tensorHandle); in setAttrTensor()
336 private static native void setAttrTensorList(long handle, String name, long[] tensorHandle); in setAttrTensorList()
DTensor.java641 NativeReference(long tensorHandle) { in NativeReference()
664 private long tensorHandle; field in Tensor.NativeReference
/external/armnn/src/backends/neon/test/
DNeonCreateWorkloadTests.cpp26 …ools::predicate_result CompareIAclTensorHandleShape(IAclTensorHandle* tensorHan… in CompareIAclTensorHandleShape()
/external/armnn/src/backends/cl/test/
DClCreateWorkloadTests.cpp22 …tools::predicate_result CompareIClTensorHandleShape(IClTensorHandle* tensorHand… in CompareIClTensorHandleShape()