Searched defs:template_types (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | slice_grad.cc | 53 std::set<TypePtr> template_types = {kTensorType}; in SliceGradInferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | dynamic_broadcast_to.cc | 87 std::set<TypePtr> template_types = {kTensorType}; in DynamicBroadcastToInferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | broadcast_to.cc | 144 std::set<TypePtr> template_types = {kTensorType}; in InferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/utils/ |
| D | check_convert_utils.cc | 568 inline bool CheckType(const TypePtr &check_type, const std::set<TypePtr> &template_types) { in CheckType()
994 … const std::set<TypePtr> &template_types, const string &prim_name, in CheckTensorSubClass()
1035 … const 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/ |
| D | common_utils.cc | 687 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.py | 600 def check_subclass(arg_name, type_, template_types, prim_name, addition_error_info=None): argument
|