diff options
Diffstat (limited to 'compute_kernel_writer/src/types/ConstantData.cpp')
-rw-r--r-- | compute_kernel_writer/src/types/ConstantData.cpp | 49 |
1 files changed, 24 insertions, 25 deletions
diff --git a/compute_kernel_writer/src/types/ConstantData.cpp b/compute_kernel_writer/src/types/ConstantData.cpp index d2155cf55a..67b1103860 100644 --- a/compute_kernel_writer/src/types/ConstantData.cpp +++ b/compute_kernel_writer/src/types/ConstantData.cpp @@ -30,52 +30,51 @@ namespace ckw { namespace { - template<typename T> - inline typename std::enable_if<std::is_same<T, float>::value, std::string>::type to_str(T value) - { - std::stringstream ss; - ss << std::scientific << std::setprecision(std::numeric_limits<T>::max_digits10) << value; - return ss.str(); - } +template <typename T> +inline typename std::enable_if<std::is_same<T, float>::value, std::string>::type to_str(T value) +{ + std::stringstream ss; + ss << std::scientific << std::setprecision(std::numeric_limits<T>::max_digits10) << value; + return ss.str(); +} - template<typename T> - inline typename std::enable_if<!std::is_same<T, float>::value && !std::is_same<T, bool>::value, std::string>::type to_str(T value) - { - return std::to_string(value); - } +template <typename T> +inline typename std::enable_if<!std::is_same<T, float>::value && !std::is_same<T, bool>::value, std::string>::type +to_str(T value) +{ + return std::to_string(value); +} - template<typename T> - inline typename std::enable_if<std::is_same<T, bool>::value, std::string>::type to_str(T value) - { - return std::to_string((int) value); - } +template <typename T> +inline typename std::enable_if<std::is_same<T, bool>::value, std::string>::type to_str(T value) +{ + return std::to_string((int)value); } +} // namespace -template<typename T> +template <typename T> ConstantData::ConstantData(std::initializer_list<std::initializer_list<T>> values, DataType data_type) : _data_type(data_type) { CKW_ASSERT(validate<T>(data_type)); CKW_ASSERT(values.size() > 0); - for(auto value_arr: values) + for (auto value_arr : values) { // Each row must have the same number of elements CKW_ASSERT(value_arr.size() == (*values.begin()).size()); StringVector vec; - std::transform(value_arr.begin(), value_arr.end(), - std::back_inserter(vec), - [](T val) { return to_str(val); }); + std::transform(value_arr.begin(), value_arr.end(), std::back_inserter(vec), [](T val) { return to_str(val); }); _values.push_back(std::move(vec)); } } -template<typename T> +template <typename T> bool ConstantData::validate(DataType data_type) { - switch(data_type) + switch (data_type) { case DataType::Fp32: case DataType::Fp16: @@ -107,7 +106,7 @@ template bool ConstantData::validate<uint32_t>(DataType); template bool ConstantData::validate<bool>(DataType); template bool ConstantData::validate<float>(DataType); -const std::vector<std::vector<std::string>>& ConstantData::values() const +const std::vector<std::vector<std::string>> &ConstantData::values() const { return _values; } |