Home
last modified time | relevance | path

Searched refs:dataCollector (Results 1 – 4 of 4) sorted by relevance

/external/armnn/src/armnn/
DLayer.hpp375 WorkloadDataCollector dataCollector(descriptor.m_Inputs, info.m_InputTensorInfos); in CollectQueueDescriptorInputs() local
376 CollectWorkloadInputs(dataCollector); in CollectQueueDescriptorInputs()
382 WorkloadDataCollector dataCollector(descriptor.m_Outputs, info.m_OutputTensorInfos); in CollectQueueDescriptorOutputs() local
383 CollectWorkloadOutputs(dataCollector); in CollectQueueDescriptorOutputs()
422 void CollectWorkloadInputs(WorkloadDataCollector& dataCollector) const;
423 void CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const;
DOutputHandler.cpp33 void OutputHandler::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const in CollectWorkloadOutputs()
35 dataCollector.Push(m_TensorHandle.get(), m_TensorInfo); in CollectWorkloadOutputs()
DLayer.cpp255 void Layer::CollectWorkloadInputs(WorkloadDataCollector& dataCollector) const in CollectWorkloadInputs()
262 dataCollector.Push(outputHandler.GetData(), outputHandler.GetTensorInfo()); in CollectWorkloadInputs()
266 void Layer::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const in CollectWorkloadOutputs()
270 outputHandler.CollectWorkloadOutputs(dataCollector); in CollectWorkloadOutputs()
DOutputHandler.hpp49 void CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const;