24.04
|
A table of all the variables used in the kernel. More...
#include <GpuCkwVariableTable.h>
Public Member Functions | |
GpuCkwComponentArgument * | declare_variable (const GpuKernelComponentGroup &comp_group, GpuCkwScopedKernelWriter &writer, const ITensorInfo *tensor, const std::string &alias="unnamed") |
Declare a kernel component variable(argument) for the corresponding tensor info. More... | |
A table of all the variables used in the kernel.
It determines whether we create an virtual tensor var or a user tensor var It avoids duplicating variables for the same tensors (Tensors with the same id) Each kernel has exactly one variable table.
Definition at line 48 of file GpuCkwVariableTable.h.
GpuCkwComponentArgument * declare_variable | ( | const GpuKernelComponentGroup & | comp_group, |
GpuCkwScopedKernelWriter & | writer, | ||
const ITensorInfo * | tensor, | ||
const std::string & | alias = "unnamed" |
||
) |
Declare a kernel component variable(argument) for the corresponding tensor info.
[in] | comp_group | Component group the tensor belongs to |
[in] | writer | Compute Kernel Writer |
[in] | tensor | Tensor info with which the new variable is associated |
[in] | alias | Alias for the variable. Will be used as part of the variable name |
Definition at line 39 of file GpuCkwVariableTable.cpp.
References ARM_COMPUTE_ERROR_ON_MSG, GpuKernelComponentGroup::is_intermediate_tensor(), arm_compute::test::validation::ss(), tensor, and arm_compute::experimental::dynamic_fusion::to_ckw().
Referenced by GpuCkwStore::write_component_code(), GpuCkwActivation::write_component_code(), GpuCkwCast::write_component_code(), GpuCkwElementwiseBinary::write_component_code(), GpuCkwPool2d::write_component_code(), GpuCkwDepthwiseConv2d::write_component_code(), GpuCkwDirectConv2d::write_component_code(), and GpuCkwMatMul::write_component_code().