Searched refs:m_TensorHandle (Results 1 – 7 of 7) sorted by relevance
193 , m_TensorHandle(std::move(ptr)) {}; in ManagedConstTensorHandle()198 if (m_TensorHandle) in Map()200 auto pRet = m_TensorHandle->Map(blocking); in Map()223 if (m_TensorHandle) in ~ManagedConstTensorHandle()232 if (m_Mapped && m_TensorHandle) in Unmap()234 m_TensorHandle->Unmap(); in Unmap()241 return m_TensorHandle->GetTensorInfo(); in GetTensorInfo()251 std::shared_ptr<ConstTensorHandle> m_TensorHandle; member in armnn::ManagedConstTensorHandle
24 m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo, IsMemoryManaged); in CreateTensorHandles() local30 m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo, IsMemoryManaged); in CreateTensorHandles()35 dataCollector.Push(m_TensorHandle.get(), m_TensorInfo); in CollectWorkloadOutputs()43 m_AllocatedTensorHandle = std::move(m_TensorHandle); in SetAllocatedData()
46 ITensorHandle* GetData() const { return m_TensorHandle.get(); } in GetData()51 void SetData(std::unique_ptr<ITensorHandle> data) { m_TensorHandle = std::move(data); } in SetData()55 void UseAllocatedData() { m_TensorHandle = m_AllocatedTensorHandle; } in UseAllocatedData()60 std::shared_ptr<ITensorHandle> m_TensorHandle; member in armnn::OutputHandler
179 , m_TensorHandle(std::move(tensorHandle)) in ImportedTensorHandlePin()186 if (m_TensorHandle) in ~ImportedTensorHandlePin()188 m_TensorHandle->Unimport(); in ~ImportedTensorHandlePin()193 std::unique_ptr<ITensorHandle> m_TensorHandle; member
753 : m_TensorHandle(std::move(handle)) in TensorPin()759 ITensorHandle* GetTensorHandle() const { return m_TensorHandle.get(); } in GetTensorHandle()764 std::unique_ptr<ITensorHandle> m_TensorHandle; member in armnn::__anonf444cf270511::TensorPin880 … auto outputTensorHandle = m_PreImportedInputHandles[inputIndex].m_TensorHandle.get(); in EnqueueWorkload()933 … ITensorHandle* inputTensorHandle = m_PreImportedOutputHandles[outputIndex].m_TensorHandle.get(); in EnqueueWorkload()1439 auto outputTensorHandle = m_PreImportedInputHandles[inputIndex].m_TensorHandle.get(); in ImportInputs()1529 ITensorHandle* tensorHandle = importedTensorHandlePin.m_TensorHandle.get(); in ImportInputs()1581 auto inputTensorHandle = m_PreImportedOutputHandles[outputIndex].m_TensorHandle.get(); in ImportOutputs()1661 ITensorHandle* tensorHandle = importedTensorHandlePin.m_TensorHandle.get(); in ImportOutputs()1694 auto& importedTensorHandle = m_PreImportedInputHandles[id].m_TensorHandle; in ClearImportedInputs()[all …]
20 m_TensorHandle = descriptor.m_Inputs[0]; in SyncMemGenericWorkload()26 m_TensorHandle->Map(true); in Execute()27 m_TensorHandle->Unmap(); in Execute()
25 ITensorHandle* m_TensorHandle; member in armnn::SyncMemGenericWorkload