20.02
|
#include <OutputHandler.hpp>
Public Member Functions | |
void | SetTensorInfo (const TensorInfo &tensorInfo) |
| |
void | CreateTensorHandles (const IWorkloadFactory &factory, const bool IsMemoryManaged=true) |
| |
void | CreateTensorHandles (const ITensorHandleFactory &factory, const bool IsMemoryManaged=true) |
const TensorInfo & | GetTensorInfo () const |
| |
ITensorHandle * | GetData () const |
| |
void | CollectWorkloadOutputs (WorkloadDataCollector &dataCollector) const |
Fill the outputs for a given queue descriptor. More... | |
void | SetData (std::unique_ptr< ITensorHandle > data) |
bool | IsTensorInfoSet () const |
Returns true if SetTensorInfo() has been called at least once on this. More... | |
Definition at line 30 of file OutputHandler.hpp.
void CollectWorkloadOutputs | ( | WorkloadDataCollector & | dataCollector | ) | const |
Fill the outputs for a given queue descriptor.
Definition at line 33 of file OutputHandler.cpp.
References WorkloadDataCollector::Push().
Referenced by OutputHandler::GetData().
void CreateTensorHandles | ( | const IWorkloadFactory & | factory, |
const bool | IsMemoryManaged = true |
||
) |
Does not allocate memory.
factory | - Factory to be used for handler creation. |
Definition at line 23 of file OutputHandler.cpp.
References IWorkloadFactory::CreateTensorHandle().
Referenced by Layer::CreateTensorHandles().
void CreateTensorHandles | ( | const ITensorHandleFactory & | factory, |
const bool | IsMemoryManaged = true |
||
) |
Definition at line 28 of file OutputHandler.cpp.
References ITensorHandleFactory::CreateTensorHandle().
|
inline |
Definition at line 48 of file OutputHandler.hpp.
References OutputHandler::CollectWorkloadOutputs().
Referenced by ConcatLayer::CreateWorkload(), and SplitterLayer::CreateWorkload().
|
inline |
Definition at line 44 of file OutputHandler.hpp.
Referenced by ConcatLayer::CreateWorkload(), SplitterLayer::CreateWorkload(), PermuteAsReshapeImpl::Run(), TransposeAsReshapeImpl::Run(), OptimizeConsecutiveReshapesImpl::Run(), and FoldPadIntoConvolution2dImpl::Run().
|
inline |
Returns true if SetTensorInfo() has been called at least once on this.
Definition at line 56 of file OutputHandler.hpp.
|
inline |
Definition at line 53 of file OutputHandler.hpp.
Referenced by ConcatLayer::CreateWorkload().
void SetTensorInfo | ( | const TensorInfo & | tensorInfo | ) |
tensorInfo | - TensorInfo for the output. |
Definition at line 17 of file OutputHandler.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), InputSlot::Insert(), TransposeAsReshapeImpl::Run(), PermuteAsReshapeImpl::Run(), FoldPadIntoConvolution2dImpl::Run(), OptimizeConsecutiveReshapesImpl::Run(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), MoveTransposeUpImpl::Run(), and MovePermuteUpImpl::Run().