24.02.1
|
Go to the documentation of this file.
35 namespace experimental
37 namespace dynamic_fusion
42 const std::string &alias)
47 auto it = _vars.find(
tensor->id());
49 if (it != _vars.end())
57 auto &&inserted = _vars.emplace(
tensor->id(), var);
58 return &(inserted.first->second);
64 ss << alias <<
"_t" << abs(
tensor->id());
65 const auto uniq_name =
ss.str();
67 auto &&inserted = _vars.emplace(
tensor->id(), var);
68 return &(inserted.first->second);
The argument of a dynamic fusion component which can be either user tensor or virtual tensor.
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.
bool is_intermediate_tensor(const ITensorInfo *tensor) const
Check if a ITensorInfo is an "intermediate" tensor of the group.
std::stringstream ss(mlgo_str)
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
ckw::DataType to_ckw(DataType dt)
Convert the Compute Library data type to Compute Kernel Writer data type.
CLTensor * tensor
Pointer to the auxiliary tensor.
Copyright (c) 2017-2024 Arm Limited.
A group of gpu kernel components to be fused together PRECONDITIONS:
Store the tensor's metadata.
Helper to automatically manage kernel writer ID space.