p19647583
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!37175 Correct wrong info for ClipByNorm ops 

Merge pull request !37175 from JoyLvliang/correct_wrong_info
This commit is contained in:
i-robot 2022-07-04 13:32:36 +00:00 committed by Gitee
parent 1ac22ee375 6d0afab7ca
commit 6f0f6aae38
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/core/ops/clip_by_norm.cc
View File

@ -84,8 +84,8 @@ TypePtr ClipByNormInferType(const PrimitivePtr &primitive, const std::vector<Abs
auto x_type = CheckAndConvertUtils::GetTensorInputType(kNameClipByNorm, input_args_abs, 0);
auto clip_norm_type = CheckAndConvertUtils::GetTensorInputType(kNameClipByNorm, input_args_abs, 1);
const std::set<TypePtr> supported_types{kFloat16, kFloat32};
(void)CheckAndConvertUtils::CheckSubClass("data type of x", x_type, supported_types, kNameClipByNorm);
(void)CheckAndConvertUtils::CheckSubClass("data type of clip_norm", clip_norm_type, supported_types, kNameClipByNorm);
(void)CheckAndConvertUtils::CheckSubClass("x", x_type, supported_types, kNameClipByNorm);
(void)CheckAndConvertUtils::CheckSubClass("clip_norm", clip_norm_type, supported_types, kNameClipByNorm);
return kFloat32;
}
} // namespace