Home
last modified time | relevance | path

Searched defs:CheckAxis (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dlayer_norm.cc61 int64_t CheckAxis(const std::string &op, const std::string &args_name, int64_t axis_value, int64_t … in CheckAxis() function
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Dparam_validator.cc216 int64_t CheckAxis(const std::string &op, const std::string &args_name, const ValuePtr &axis, int64_… in CheckAxis() function