diff options
Diffstat (limited to 'compute_kernel_writer/src/KernelWriter.cpp')
| -rw-r--r-- | compute_kernel_writer/src/KernelWriter.cpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/compute_kernel_writer/src/KernelWriter.cpp b/compute_kernel_writer/src/KernelWriter.cpp index 7b83eade6f..ce34a1c2d6 100644 --- a/compute_kernel_writer/src/KernelWriter.cpp +++ b/compute_kernel_writer/src/KernelWriter.cpp @@ -28,6 +28,8 @@ #include "ckw/types/TargetArchitecture.h" #include "ckw/types/TargetLanguage.h" #include "src/cl/CLKernelWriter.h" +#include "src/cl/CLTensorArgument.h" +#include "src/cl/CLTile.h" namespace ckw { @@ -69,4 +71,14 @@ ITile &KernelWriter::get_tile(const TileOperand &operand) return operand._tile; } +TensorOperand KernelWriter::create_tensor_operand(ITensor &tensor) +{ + return TensorOperand(tensor); +} + +ITensor &KernelWriter::get_tensor(const TensorOperand &operand) +{ + return operand._tensor; +} + } // namespace ckw |