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

fix bug of softplus infer

This commit is contained in:
lianliguang 2021-06-08 14:40:41 +08:00
parent df491ffe53
commit 872e4b97fa
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/softplus.cc
View File

@ -40,8 +40,8 @@ TypePtr SoftplusInferType(const PrimitivePtr &prim, const std::vector<AbstractBa
auto prim_name = prim->name();
// check
std::set<TypePtr> valid_index_types = {kFloat16, kFloat32, kFloat64};
auto x_type =
CheckAndConvertUtils::CheckTensorTypeValid("x", input_args[0]->BuildType(), valid_index_types, prim_name);
auto x_type = input_args[0]->BuildType();
(void)CheckAndConvertUtils::CheckTensorTypeValid("x", x_type, valid_index_types, prim_name);
return x_type;
}
} // namespace