24.02.1
|
A table of all the variables used in the kernel. More...
#include <GpuKernelVariableTable.h>
Data Structures | |
struct | TensorVariable |
A tensor variable whose main purposes are: More... | |
Public Types | |
using | VariableList = std::vector< TensorVariable > |
Public Member Functions | |
void | declare_variable (const GpuKernelComponentGroup &comp_group, const ITensorInfo *tensor, GpuKernelArgumentInfo argument_info, const std::string &alias="unnamed") |
Declare a TensorVariable for a corresponding tensor info. More... | |
TensorVariable | get_variable (const ITensorInfo *tensor) const |
Get the TensorVariable associated with tensor . More... | |
VariableList | get_variable_list (const std::vector< const ITensorInfo * > &tensors) const |
Get the TensorVariable list associated with tensors . More... | |
A table of all the variables used in the kernel.
Each kernel has exactly one variable table.
Definition at line 49 of file GpuKernelVariableTable.h.
using VariableList = std::vector<TensorVariable> |
Definition at line 70 of file GpuKernelVariableTable.h.
void declare_variable | ( | const GpuKernelComponentGroup & | comp_group, |
const ITensorInfo * | tensor, | ||
GpuKernelArgumentInfo | argument_info, | ||
const std::string & | alias = "unnamed" |
||
) |
Declare a TensorVariable for a corresponding tensor info.
[in] | comp_group | Component group the tensor belongs to |
[in] | tensor | Tensor info with which the new variable is associated |
[in] | argument_info | Kernel argument information |
[in] | alias | Alias for the variable. Will be used as part of the variable name |
Definition at line 37 of file GpuKernelVariableTable.cpp.
References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_ERROR_ON_MSG, GpuKernelComponentGroup::get_tile_for_tensor(), arm_compute::test::validation::ss(), and tensor.
Referenced by ClTemplateStore::declare_variables(), ClTemplateCast::declare_variables(), ClTemplateReshape::declare_variables(), ClTemplateLogits1DNorm::declare_variables(), ClTemplateLogits1DMaxShiftExpSum::declare_variables(), ClTemplateDepthwiseConv2d::declare_variables(), ClTemplateDirectConv2d::declare_variables(), ClTemplateResize::declare_variables(), ClTemplateActivation::declare_variables(), and ClTemplatePool2d::declare_variables().
GpuKernelVariableTable::TensorVariable get_variable | ( | const ITensorInfo * | tensor | ) | const |
Get the TensorVariable associated with tensor
.
[in] | tensor | Tensor info to be queried |
Definition at line 76 of file GpuKernelVariableTable.cpp.
References tensor.
Referenced by ClTemplateStore::get_tag_lut(), ClTemplateCast::get_tag_lut(), ClTemplateLogits1DNorm::get_tag_lut(), ClTemplateReshape::get_tag_lut(), ClTemplateLogits1DMaxShiftExpSum::get_tag_lut(), ClTemplateDepthwiseConv2d::get_tag_lut(), ClTemplateDirectConv2d::get_tag_lut(), ClTemplateResize::get_tag_lut(), ClTemplateActivation::get_tag_lut(), ClTemplatePool2d::get_tag_lut(), and GpuKernelVariableTable::get_variable_list().
GpuKernelVariableTable::VariableList get_variable_list | ( | const std::vector< const ITensorInfo * > & | tensors | ) | const |
Get the TensorVariable list associated with tensors
.
[in] | tensors | List of tensor infos to be queried |
Definition at line 83 of file GpuKernelVariableTable.cpp.
References GpuKernelVariableTable::get_variable(), and tensor.