21.02
|
Tensor packing service. More...
#include <ITensorPack.h>
Public Member Functions | |
ITensorPack ()=default | |
Default Constructor. More... | |
void | add_tensor (int id, ITensor *tensor) |
Add tensor to the pack. More... | |
void | add_tensor (int id, const ITensor *tensor) |
Add const tensor to the pack. More... | |
ITensor * | get_tensor (int id) |
Get tensor of a given id from the pac. More... | |
const ITensor * | get_const_tensor (int id) const |
Get constant tensor of a given id. More... | |
size_t | size () const |
Pack size accessor. More... | |
bool | empty () const |
Checks if pack is empty. More... | |
Tensor packing service.
Definition at line 37 of file ITensorPack.h.
|
default |
Default Constructor.
void add_tensor | ( | int | id, |
ITensor * | tensor | ||
) |
Add tensor to the pack.
[in] | id | ID/type of the tensor to add |
[in] | tensor | Tensor to add |
Definition at line 30 of file ITensorPack.cpp.
Referenced by NEFill::run(), NEElementwiseUnaryLayer< op >::run(), NEReshapeLayer::run(), NECopy::run(), NEFloor::run(), CpuConcatenate::run(), NEElementwiseMax::run(), ClConcatenate::run(), CLReshapeLayer::run(), NEPermute::run(), NESoftmaxLayerGeneric< IS_LOG >::run(), CLFloor::run(), CLRsqrtLayer::run(), CpuPoolingAssemblyDispatch::run(), CLFill::run(), NEArithmeticSubtraction::run(), CLLogicalNot::run(), CLCopy::run(), NEConcatenateLayer::run(), CLPoolingLayer::run(), NEPoolingLayer::run(), CpuSoftmaxGeneric< IS_LOG >::run(), NEActivationLayer::run(), CLPermute::run(), NEArithmeticAddition::run(), CLActivationLayer::run(), CLCrop::run(), NEPReluLayer::run(), CLConcatenateLayer::run(), CLStridedSlice::run(), CLLogicalAnd::run(), CLLogicalOr::run(), NESlice::run(), CLPReluLayer::run(), NEElementwiseMin::run(), CLExpLayer::run(), NEStridedSlice::run(), CLArithmeticAddition::run(), CLSlice::run(), NEElementwiseSquaredDiff::run(), CLNegLayer::run(), NEPixelWiseMultiplication::run(), CLSinLayer::run(), NEElementwiseDivision::run(), CLArithmeticSubtraction::run(), NEComplexPixelWiseMultiplication::run(), CLPixelWiseMultiplication::run(), CLLogLayer::run(), NEElementwisePower::run(), CLArithmeticDivision::run(), CLAbsLayer::run(), CLComplexPixelWiseMultiplication::run(), NEElementwiseComparison::run(), CLRoundLayer::run(), NEElementwiseComparisonStatic< op >::run(), CLElementwiseMax::run(), CLElementwiseMin::run(), CLElementwiseSquaredDiff::run(), and CLElementwisePower::run().
void add_tensor | ( | int | id, |
const ITensor * | tensor | ||
) |
Add const tensor to the pack.
[in] | id | ID/type of the tensor to add |
[in] | tensor | Tensor to add |
Definition at line 35 of file ITensorPack.cpp.
bool empty | ( | ) | const |
Checks if pack is empty.
Definition at line 61 of file ITensorPack.cpp.
Referenced by CLTuner::add_tuning_params(), CLScheduler::init(), IScheduler::num_threads_hint(), ICLOperator::run(), INEOperator::run(), ClPooling::run(), CpuConcatenate::run(), ClConcatenate::run(), CpuPoolingAssemblyDispatch::run(), CpuSoftmaxGeneric< IS_LOG >::run(), NELogicalKernel::run_op(), CpuActivationKernel::run_op(), CpuFloorKernel::run_op(), CpuAddKernel::run_op(), and CpuPoolingAssemblyWrapperKernel::run_op().
const ITensor * get_const_tensor | ( | int | id | ) | const |
Get constant tensor of a given id.
[in] | id | ID of tensor to extract |
Definition at line 40 of file ITensorPack.cpp.
Referenced by CpuConcatenate::run(), ClConcatenate::run(), CpuSoftmaxGeneric< IS_LOG >::run(), ClElementwiseKernel::run_op(), CpuLogits1DMaxKernel::run_op(), CpuReshapeKernel::run_op(), CpuCopyKernel::run_op(), ClReshapeKernel::run_op(), ClFloorKernel::run_op(), ClCopyKernel::run_op(), ClElementWiseUnaryKernel::run_op(), CpuConcatenateHeightKernel::run_op(), CpuConcatenateWidthKernel::run_op(), NELogicalKernel::run_op(), CpuActivationKernel::run_op(), CpuElementwiseKernel::run_op(), CpuPermuteKernel::run_op(), ClWidthConcatenate2TensorsKernel::run_op(), CpuConcatenateBatchKernel::run_op(), ClHeightConcatenateKernel::run_op(), ClWidthConcatenateKernel::run_op(), ClActivationKernel::run_op(), CpuFloorKernel::run_op(), ClPoolingKernel::run_op(), CpuPoolingKernel::run_op(), ClBatchConcatenateKernel::run_op(), ClDepthConcatenateKernel::run_op(), ClFillKernel::run_op(), CpuConcatenateDepthKernel::run_op(), ClWidthConcatenate4TensorsKernel::run_op(), CpuElementwiseUnaryKernel::run_op(), CpuAddKernel::run_op(), CLStridedSliceKernel::run_op(), ClPermuteKernel::run_op(), CpuPoolingAssemblyWrapperKernel::run_op(), CLFillBorderKernel::run_op(), ClCropKernel::run_op(), CpuSubKernel::run_op(), NEStridedSliceKernel::run_op(), CpuLogits1DSoftmaxKernel< IS_LOG >::run_op(), NEPixelWiseMultiplicationKernel::run_op(), CLPixelWiseMultiplicationKernel::run_op(), NEComplexPixelWiseMultiplicationKernel::run_op(), and CLComplexPixelWiseMultiplicationKernel::run_op().
ITensor * get_tensor | ( | int | id | ) |
Get tensor of a given id from the pac.
[in] | id | ID of tensor to extract |
Definition at line 50 of file ITensorPack.cpp.
Referenced by CpuConcatenate::run(), ClConcatenate::run(), CpuSoftmaxGeneric< IS_LOG >::run(), CpuFillKernel::run_op(), ClElementwiseKernel::run_op(), CpuLogits1DMaxKernel::run_op(), CpuReshapeKernel::run_op(), CpuCopyKernel::run_op(), ClReshapeKernel::run_op(), ClFloorKernel::run_op(), ClCopyKernel::run_op(), ClElementWiseUnaryKernel::run_op(), CpuConcatenateHeightKernel::run_op(), CpuConcatenateWidthKernel::run_op(), CpuActivationKernel::run_op(), NELogicalKernel::run_op(), CpuElementwiseKernel::run_op(), CpuPermuteKernel::run_op(), ClWidthConcatenate2TensorsKernel::run_op(), CpuConcatenateBatchKernel::run_op(), ClWidthConcatenateKernel::run_op(), ClActivationKernel::run_op(), CpuFloorKernel::run_op(), ClHeightConcatenateKernel::run_op(), ClPoolingKernel::run_op(), CpuPoolingKernel::run_op(), ClBatchConcatenateKernel::run_op(), ClDepthConcatenateKernel::run_op(), ClWidthConcatenate4TensorsKernel::run_op(), CpuConcatenateDepthKernel::run_op(), CpuAddKernel::run_op(), CLStridedSliceKernel::run_op(), CpuElementwiseUnaryKernel::run_op(), ClPermuteKernel::run_op(), NEFillBorderKernel::run_op(), CpuPoolingAssemblyWrapperKernel::run_op(), ClCropKernel::run_op(), CpuSubKernel::run_op(), NEStridedSliceKernel::run_op(), CpuLogits1DSoftmaxKernel< IS_LOG >::run_op(), NEPixelWiseMultiplicationKernel::run_op(), CLPixelWiseMultiplicationKernel::run_op(), NEComplexPixelWiseMultiplicationKernel::run_op(), and CLComplexPixelWiseMultiplicationKernel::run_op().
size_t size | ( | ) | const |
Pack size accessor.
Definition at line 56 of file ITensorPack.cpp.
Referenced by CpuConcatenate::run(), and ClConcatenate::run().