diff options
Diffstat (limited to 'compute_kernel_writer/include/ckw/types')
| -rw-r--r-- | compute_kernel_writer/include/ckw/types/ConstantData.h | 4 | ||||
| -rw-r--r-- | compute_kernel_writer/include/ckw/types/Operators.h | 1 |
2 files changed, 5 insertions, 0 deletions
diff --git a/compute_kernel_writer/include/ckw/types/ConstantData.h b/compute_kernel_writer/include/ckw/types/ConstantData.h index 7708818ca8..ea95049c9e 100644 --- a/compute_kernel_writer/include/ckw/types/ConstantData.h +++ b/compute_kernel_writer/include/ckw/types/ConstantData.h @@ -53,6 +53,10 @@ public: template <typename T> ConstantData(std::initializer_list<std::initializer_list<T>> values, DataType data_type); + /** Templated constructor */ + template <typename T> + ConstantData(const std::vector<std::vector<T>> &values, DataType data_type); + private: /** Validate the given data type and the template type * diff --git a/compute_kernel_writer/include/ckw/types/Operators.h b/compute_kernel_writer/include/ckw/types/Operators.h index 1e5f9bd542..77b0519422 100644 --- a/compute_kernel_writer/include/ckw/types/Operators.h +++ b/compute_kernel_writer/include/ckw/types/Operators.h @@ -43,6 +43,7 @@ enum class UnaryOp : int32_t Fabs = 0x0014, Log = 0x0015, Round = 0x0016, + Floor = 0x0017, }; /** Assignment operators. */ |