aboutsummaryrefslogtreecommitdiff
path: root/compute_kernel_writer/prototype/include/ckw/TensorInfo.h
diff options
context:
space:
mode:
Diffstat (limited to 'compute_kernel_writer/prototype/include/ckw/TensorInfo.h')
-rw-r--r--compute_kernel_writer/prototype/include/ckw/TensorInfo.h10
1 files changed, 9 insertions, 1 deletions
diff --git a/compute_kernel_writer/prototype/include/ckw/TensorInfo.h b/compute_kernel_writer/prototype/include/ckw/TensorInfo.h
index 00bb60a444..807158896b 100644
--- a/compute_kernel_writer/prototype/include/ckw/TensorInfo.h
+++ b/compute_kernel_writer/prototype/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 /* CKW_PROTOTYPE_INCLUDE_CKW_TENSORINFO_H */