diff options
author | Nikolaj Jensen <nikolaj.jensen@arm.com> | 2023-07-03 09:44:42 +0100 |
---|---|---|
committer | Nikolaj Jensen <nikolaj.jensen@arm.com> | 2023-07-07 08:51:11 +0000 |
commit | acea4071a7f457bab696dc3c895ba47d60345541 (patch) | |
tree | 66a5f659f995ea110bcb8e217dc6a94e1f7ac47d /compute_kernel_writer/include/ckw/TensorInfo.h | |
parent | e86f992d26a79cad76244c4444d113e45afa9b88 (diff) | |
download | ComputeLibrary-acea4071a7f457bab696dc3c895ba47d60345541.tar.gz |
Fix code formatting in CKW
Signed-off-by: Nikolaj Jensen <nikolaj.jensen@arm.com>
Change-Id: I8064b345c1efd243f8bded12ed5d561afe7c339a
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9854
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Jakub Sujak <jakub.sujak@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'compute_kernel_writer/include/ckw/TensorInfo.h')
-rw-r--r-- | compute_kernel_writer/include/ckw/TensorInfo.h | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/compute_kernel_writer/include/ckw/TensorInfo.h b/compute_kernel_writer/include/ckw/TensorInfo.h index 44846bc94c..41abe60f35 100644 --- a/compute_kernel_writer/include/ckw/TensorInfo.h +++ b/compute_kernel_writer/include/ckw/TensorInfo.h @@ -117,20 +117,28 @@ public: * - less than 0: bind a virtual tensor (tile) */ TensorInfo(DataType dt, const TensorShape &shape, TensorDataLayout dl, int32_t id); + /** Set shape */ TensorInfo &shape(const TensorShape &shape); + /** Get shape */ TensorShape shape() const; + /** Set data type */ TensorInfo &data_type(DataType dt); + /** Get data type */ DataType data_type() const; + /** Set data layout */ TensorInfo &data_layout(TensorDataLayout dl); + /** Get data layout */ TensorDataLayout data_layout() const; + /** Set id */ TensorInfo &id(int32_t id); + /** Get layout */ int32_t id() const; @@ -140,6 +148,6 @@ private: TensorDataLayout _dl{ TensorDataLayout::Unknown }; int32_t _id{ -1 }; }; -} // namespace kw +} // namespace ckw #endif /* COMPUTE_KERNEL_WRITER_INCLUDE_CKW_TENSORINFO_H */ |