24.04
|
Go to the documentation of this file.
25 #ifndef ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_GPUCKWCOMPONENTARGUMENT_H
26 #define ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_GPUCKWCOMPONENTARGUMENT_H
28 #include "compute_kernel_writer/include/ckw/TensorOperand.h"
29 #include "compute_kernel_writer/include/ckw/TensorSampler.h"
30 #include "compute_kernel_writer/include/ckw/TileOperand.h"
34 namespace experimental
36 namespace dynamic_fusion
72 ckw::TensorOperand &
tensor();
78 const ckw::TensorOperand &
tensor()
const;
91 ckw::TileOperand &
tile();
97 const ckw::TileOperand &
tile()
const;
112 ckw::TensorOperand _tensor{};
113 ckw::TileOperand _tile{};
114 ckw::TensorSampler _sampler{};
121 #endif // ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_GPUCKWCOMPONENTARGUMENT_H
GpuCkwComponentArgument()=default
Default constructor.
The argument of a dynamic fusion component which can be either user tensor or virtual tensor.
bool has_tile() const
Get whether the argument contains a tile.
GpuCkwComponentArgument & init_virtual_tensor(ckw::TileOperand &tile, const ckw::TensorSampler &sampler)
Bind the tile and sampler to the tensor argument.
bool has_tensor() const
Get whether the argument is a user tensor.
ckw::TensorOperand & tensor()
Get the tensor operand.
Copyright (c) 2017-2024 Arm Limited.
ckw::TensorSampler & tensor_sampler()
Get the tensor sampling information for the tile.
ckw::TileOperand & tile()
Get the tile operand.