Searched refs:template_types (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | assert.cc | 73 std::set<TypePtr> template_types = {kTensorType}; in AssertInfer() local
74 (void)CheckAndConvertUtils::CheckSubClass("input", dtype, template_types, op_name); in AssertInfer()
|
| D | broadcast_to.cc | 70 std::set<TypePtr> template_types = {kTensorType}; in BroadcastToInferType() local
71 (void)CheckAndConvertUtils::CheckSubClass("x_dtype", x_dtype, template_types, prim->name()); in BroadcastToInferType()
|
| D | tile.cc | 89 std::set<TypePtr> template_types = {kTensorType}; in TileInferType() local
90 return CheckAndConvertUtils::CheckTensorTypeValid("x_dtype", x_dtype, template_types, prim_name); in TileInferType()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.cc | 549 … const std::set<TypePtr> &template_types, const std::string &prim_name) { in CheckSubClass() argument
551 …bool ok = std::any_of(template_types.begin(), template_types.end(), [check_type](const TypePtr &ac… in CheckSubClass()
561 ok = std::any_of(template_types.begin(), template_types.end(), in CheckSubClass()
568 buffer << GetErrorTypeString(template_types, type) << ", but got " << type->ToString(); in CheckSubClass()
|
| D | check_convert_utils.h | 289 … const std::set<TypePtr> &template_types, const std::string &prim_name);
|
| /third_party/mindspore/mindspore/ |
| D | _checkparam.py | 480 def check_subclass(arg_name, type_, template_types, prim_name, addition_error_info=None): argument
482 if not isinstance(template_types, Iterable):
483 template_types = (template_types,)
485 for template_type in template_types:
|