21.02
|
Basic function to run kernels::CpuPermuteKernel. More...
#include <CpuPermute.h>
Public Member Functions | |
CpuPermute ()=default | |
Constructor. More... | |
void | configure (const ITensorInfo *src, ITensorInfo *dst, const PermutationVector &perm) |
Configure operator for a given list of arguments. More... | |
Public Member Functions inherited from INEOperator | |
INEOperator (IRuntimeContext *ctx=nullptr) | |
Constructor. More... | |
INEOperator (const INEOperator &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INEOperator (INEOperator &&)=default | |
Default move constructor. More... | |
INEOperator & | operator= (const INEOperator &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INEOperator & | operator= (INEOperator &&)=default |
Default move assignment operator. More... | |
~INEOperator () | |
Default destructor. More... | |
void | run (ITensorPack &tensors) override |
Run the kernels contained in the function. More... | |
void | prepare (ITensorPack &constants) override |
Prepare the function for executing. More... | |
MemoryRequirements | workspace () const override |
Return the memory requirements required by the workspace. More... | |
Public Member Functions inherited from IOperator | |
virtual | ~IOperator ()=default |
Destructor. More... | |
Static Public Member Functions | |
static Status | validate (const ITensorInfo *src, const ITensorInfo *dst, const PermutationVector &perm) |
Static function to check if given info will lead to a valid configuration of CpuPermute. More... | |
Basic function to run kernels::CpuPermuteKernel.
Definition at line 34 of file CpuPermute.h.
|
default |
Constructor.
void configure | ( | const ITensorInfo * | src, |
ITensorInfo * | dst, | ||
const PermutationVector & | perm | ||
) |
Configure operator for a given list of arguments.
[in] | src | Source tensor to permute. Data types supported: All |
[out] | dst | Destintation tensor. Data types supported: Same as src |
[in] | perm | Permutation vector |
Definition at line 32 of file CpuPermute.cpp.
Referenced by CpuSoftmaxGeneric< IS_LOG >::configure().
|
static |
Static function to check if given info will lead to a valid configuration of CpuPermute.
[in] | src | Source tensor to permute. Data types supported: All |
[in] | dst | Destination tensor. Data types supported: Same as dst |
[in] | perm | Permutation vector |
Definition at line 39 of file CpuPermute.cpp.
References CpuPermuteKernel::validate().
Referenced by NEPermute::validate(), and CpuSoftmaxGeneric< IS_LOG >::validate().