diff options
Diffstat (limited to 'reference_model/src/ops/type_conversion.h')
-rw-r--r-- | reference_model/src/ops/type_conversion.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/reference_model/src/ops/type_conversion.h b/reference_model/src/ops/type_conversion.h index 5f197cf..b0de30c 100644 --- a/reference_model/src/ops/type_conversion.h +++ b/reference_model/src/ops/type_conversion.h @@ -137,11 +137,11 @@ private: }; template <DType InDtype> -class CastHelper<InDtype, DType_FLOAT> +class CastHelper<InDtype, DType_FP32> { public: using InEigenType = typename GetEigenType<InDtype>::type; - using OutEigenType = typename GetEigenType<DType_FLOAT>::type; + using OutEigenType = typename GetEigenType<DType_FP32>::type; using FcnType = std::function<OutEigenType(InEigenType)>; CastHelper(); const FcnType& get_fcn() const @@ -154,10 +154,10 @@ private: }; template <DType OutDtype> -class CastHelper<DType_FLOAT, OutDtype> +class CastHelper<DType_FP32, OutDtype> { public: - using InEigenType = typename GetEigenType<DType_FLOAT>::type; + using InEigenType = typename GetEigenType<DType_FP32>::type; using OutEigenType = typename GetEigenType<OutDtype>::type; using FcnType = std::function<OutEigenType(InEigenType)>; static constexpr int32_t OutMin = GetQMin<OutDtype>::value; |