Home
last modified time | relevance | path

Searched defs:template_types (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dslice_grad.cc53 std::set<TypePtr> template_types = {kTensorType}; in SliceGradInferType() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Ddynamic_broadcast_to.cc87 std::set<TypePtr> template_types = {kTensorType}; in DynamicBroadcastToInferType() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dbroadcast_to.cc144 std::set<TypePtr> template_types = {kTensorType}; in InferType() local
/third_party/mindspore/mindspore-src/source/mindspore/core/utils/
Dcheck_convert_utils.cc568 inline bool CheckType(const TypePtr &check_type, const std::set<TypePtr> &template_types) { in CheckType()
994const std::set<TypePtr> &template_types, const string &prim_name, in CheckTensorSubClass()
1035const std::set<TypePtr> &template_types, const std::string &prim_name) { in CheckSubClass()
1058 const std::set<TypePtr> &template_types, in CheckSubClassWithMoreInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/
Dcommon_utils.cc687 bool CheckType(const TypePtr &check_type, const std::set<TypePtr> &template_types) { in CheckType()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/
D_checkparam.py600 def check_subclass(arg_name, type_, template_types, prim_name, addition_error_info=None): argument