21.05
|
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... | |
void | add_const_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_const_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 40 of file ITensorPack.cpp.
References ITensorPack::add_tensor().
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 ITensorPack::add_const_tensor(), TensorPack::add_tensor(), ClPRelu::run(), NEFill::run(), CpuConcatenate::run(), ClConcatenate::run(), NEReshapeLayer::run(), NETranspose::run(), CpuPoolingAssemblyDispatch::run(), NECopy::run(), NEFloor::run(), NEQuantizationLayer::run(), NEElementwiseUnaryLayer< op >::run(), NEConvertFullyConnectedWeights::run(), NEDequantizationLayer::run(), CLReshapeLayer::run(), NEPReluLayer::run(), CLTranspose::run(), NEPermute::run(), CpuSoftmaxGeneric< IS_LOG >::run(), CLRsqrtLayer::run(), NEScale::run(), CLFloor::run(), CLFill::run(), NESoftmaxLayerGeneric< IS_LOG >::run(), CLLogicalNot::run(), NEElementwiseMax::run(), CLCopy::run(), CLDequantizationLayer::run(), NEArithmeticAddition::run(), NEConcatenateLayer::run(), CLPReluLayer::run(), CLPermute::run(), CLPoolingLayer::run(), NEPoolingLayer::run(), CLConvertFullyConnectedWeights::run(), CLQuantizationLayer::run(), NEActivationLayer::run(), NEArithmeticSubtraction::run(), CLScale::run(), CLSoftmaxLayerGeneric< IS_LOG >::run(), CpuDirectConvolution::run(), CLActivationLayer::run(), CLCrop::run(), NEDirectConvolutionLayer::run(), CLConcatenateLayer::run(), CLDirectConvolutionLayer::run(), CLStridedSlice::run(), CLPixelWiseMultiplication::run(), CLLogicalAnd::run(), CLLogicalOr::run(), NEPixelWiseMultiplication::run(), NESlice::run(), CLArithmeticAddition::run(), NEStridedSlice::run(), CLExpLayer::run(), CLSlice::run(), NEElementwiseMin::run(), NEComplexPixelWiseMultiplication::run(), CLComplexPixelWiseMultiplication::run(), CLNegLayer::run(), NEElementwiseSquaredDiff::run(), CLLSTMLayer::run(), CLArithmeticSubtraction::run(), CLSinLayer::run(), NEElementwiseDivision::run(), CLLogLayer::run(), CLArithmeticDivision::run(), NEElementwisePower::run(), CLAbsLayer::run(), NEElementwiseComparison::run(), CLElementwiseMax::run(), CLRoundLayer::run(), NEElementwiseComparisonStatic< op >::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 66 of file ITensorPack.cpp.
Referenced by TensorPack::empty(), ClDequantization::run(), ICLOperator::run(), CpuDequantization::run(), ClQuantization::run(), INEOperator::run(), CpuScale::run(), CpuQuantization::run(), ClScale::run(), ClPooling::run(), CpuConcatenate::run(), ClConcatenate::run(), CpuMul::run(), CpuPoolingAssemblyDispatch::run(), CpuSoftmaxGeneric< IS_LOG >::run(), CpuComplexMul::run(), CpuActivationKernel::run_op(), NELogicalKernel::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 45 of file ITensorPack.cpp.
Referenced by CpuScale::prepare(), CpuDepthwiseConvolutionAssemblyDispatch::prepare(), CpuElementwiseBase::run(), CpuElementwiseUnary::run(), ClPRelu::run(), ClSoftmax::run(), CpuConcatenate::run(), ClConcatenate::run(), CpuSoftmaxGeneric< IS_LOG >::run(), CpuDirectConvolution::run(), ClElementwiseKernel::run_op(), CpuTransposeKernel::run_op(), CpuLogits1DMaxKernel::run_op(), CpuDequantizationKernel::run_op(), CpuReshapeKernel::run_op(), CpuCopyKernel::run_op(), ClTransposeKernel::run_op(), ClFloorKernel::run_op(), ClReshapeKernel::run_op(), ClElementWiseUnaryKernel::run_op(), ClCopyKernel::run_op(), CpuConcatenateHeightKernel::run_op(), CpuConcatenateWidthKernel::run_op(), ClDequantizationKernel::run_op(), CpuPermuteKernel::run_op(), CpuActivationKernel::run_op(), NELogicalKernel::run_op(), CpuElementwiseKernel::run_op(), CpuConcatenateBatchKernel::run_op(), CpuQuantizationKernel::run_op(), ClWidthConcatenate2TensorsKernel::run_op(), ClActivationKernel::run_op(), ClHeightConcatenateKernel::run_op(), ClWidthConcatenateKernel::run_op(), CpuFloorKernel::run_op(), ClPoolingKernel::run_op(), CpuPoolingKernel::run_op(), ClQuantizationKernel::run_op(), ClDepthConcatenateKernel::run_op(), ClFillKernel::run_op(), ClBatchConcatenateKernel::run_op(), CpuConvertFullyConnectedWeightsKernel::run_op(), CpuConcatenateDepthKernel::run_op(), ClPermuteKernel::run_op(), ClWidthConcatenate4TensorsKernel::run_op(), ClConvertFullyConnectedWeightsKernel::run_op(), ClScaleKernel::run_op(), CpuScaleKernel::run_op(), CpuElementwiseUnaryKernel::run_op(), CpuAddKernel::run_op(), CLStridedSliceKernel::run_op(), CpuDirectConvolutionKernel::run_op(), CpuDirectConvolutionOutputStageKernel::run_op(), CpuMulKernel::run_op(), ClMulKernel::run_op(), CpuPoolingAssemblyWrapperKernel::run_op(), CpuDepthwiseConvolutionNativeKernel::run_op(), ClDirectConvolutionKernel::run_op(), CLFillBorderKernel::run_op(), ClCropKernel::run_op(), ClLogits1DMaxShiftExpSumKernel::run_op(), CpuSubKernel::run_op(), NEStridedSliceKernel::run_op(), CpuLogits1DSoftmaxKernel< IS_LOG >::run_op(), ClComplexMulKernel::run_op(), ClLogits1DNormKernel::run_op(), and CpuComplexMulKernel::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 55 of file ITensorPack.cpp.
Referenced by TensorPack::get_tensor(), CpuScale::prepare(), CpuDepthwiseConvolutionAssemblyDispatch::prepare(), ClPRelu::run(), ClSoftmax::run(), CpuConcatenate::run(), ClConcatenate::run(), CpuSoftmaxGeneric< IS_LOG >::run(), CpuDepthwiseConvolutionAssemblyDispatch::run(), CpuDirectConvolution::run(), CpuFillKernel::run_op(), ClElementwiseKernel::run_op(), CpuTransposeKernel::run_op(), CpuReshapeKernel::run_op(), CpuLogits1DMaxKernel::run_op(), CpuDequantizationKernel::run_op(), CpuCopyKernel::run_op(), ClTransposeKernel::run_op(), ClFloorKernel::run_op(), ClReshapeKernel::run_op(), ClCopyKernel::run_op(), ClElementWiseUnaryKernel::run_op(), CpuConcatenateWidthKernel::run_op(), CpuConcatenateHeightKernel::run_op(), CpuActivationKernel::run_op(), ClDequantizationKernel::run_op(), CpuPermuteKernel::run_op(), NELogicalKernel::run_op(), CpuElementwiseKernel::run_op(), CpuConcatenateBatchKernel::run_op(), CpuQuantizationKernel::run_op(), ClWidthConcatenate2TensorsKernel::run_op(), CpuFloorKernel::run_op(), ClHeightConcatenateKernel::run_op(), ClWidthConcatenateKernel::run_op(), ClActivationKernel::run_op(), ClPoolingKernel::run_op(), ClQuantizationKernel::run_op(), CpuPoolingKernel::run_op(), ClBatchConcatenateKernel::run_op(), ClDepthConcatenateKernel::run_op(), CpuConvertFullyConnectedWeightsKernel::run_op(), ClPermuteKernel::run_op(), CpuConcatenateDepthKernel::run_op(), ClWidthConcatenate4TensorsKernel::run_op(), ClConvertFullyConnectedWeightsKernel::run_op(), ClScaleKernel::run_op(), CpuScaleKernel::run_op(), CpuElementwiseUnaryKernel::run_op(), CpuAddKernel::run_op(), CpuDirectConvolutionKernel::run_op(), CLStridedSliceKernel::run_op(), CpuDirectConvolutionOutputStageKernel::run_op(), CpuMulKernel::run_op(), NEFillBorderKernel::run_op(), ClMulKernel::run_op(), CpuPoolingAssemblyWrapperKernel::run_op(), CpuDepthwiseConvolutionNativeKernel::run_op(), ClDirectConvolutionKernel::run_op(), ClCropKernel::run_op(), CpuSubKernel::run_op(), ClLogits1DMaxShiftExpSumKernel::run_op(), NEStridedSliceKernel::run_op(), CpuLogits1DSoftmaxKernel< IS_LOG >::run_op(), ClComplexMulKernel::run_op(), ClLogits1DNormKernel::run_op(), and CpuComplexMulKernel::run_op().
size_t size | ( | ) | const |
Pack size accessor.
Definition at line 61 of file ITensorPack.cpp.
Referenced by CpuConcatenate::run(), ClConcatenate::run(), and TensorPack::size().