24.02.1
|
Go to the documentation of this file.
33 namespace experimental
35 namespace dynamic_fusion
40 const std::string &alias)
45 auto it = _vars.find(
tensor->id());
47 if (it != _vars.end())
58 it = _vars.find(target->id());
62 _vars[
tensor->id()] = it->second;
68 ss << alias <<
"_t" << abs(
tensor->id());
69 const auto uniq_name =
ss.str();
72 _vars.emplace(
tensor->id(), var);
78 const auto var = _vars.at(
tensor->id());
86 for (
const auto &
tensor : tensors)
88 if (!
tensor->has_valid_id())
A tensor variable whose main purposes are:
std::vector< TensorVariable > VariableList
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
std::stringstream ss(mlgo_str)
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
Contain information required to set up a kernel argument at run time.
CLTensor * tensor
Pointer to the auxiliary tensor.
TensorVariable get_variable(const ITensorInfo *tensor) const
Get the TensorVariable associated with tensor.
std::string get_cl_type_from_data_type(const DataType &dt)
Translates a tensor data type to the appropriate OpenCL type.
const ITensorInfo * get_tile_for_tensor(const ITensorInfo *tensor) const
Get the shared tile that can be used to store temporary data of the specified tensor.
Copyright (c) 2017-2024 Arm Limited.
A group of gpu kernel components to be fused together PRECONDITIONS:
Store the tensor's metadata.
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.
VariableList get_variable_list(const std::vector< const ITensorInfo * > &tensors) const
Get the TensorVariable list associated with tensors.
DataType
Available data types.
TagVal()=default
Default constructor.