Searched refs:CheckTypeValid (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | scatter_nd.cc | 47 …(void)CheckAndConvertUtils::CheckTypeValid("update type", update_type, update_valid_types, prim->n… in InferType()
|
| D | scatter_nd_update.cc | 46 …(void)CheckAndConvertUtils::CheckTypeValid("update type", update_type, update_valid_types, prim->n… in InferType()
|
| D | fill.cc | 42 (void)CheckAndConvertUtils::CheckTypeValid("output datatype", dtype, valid_types, prim_name); in FillInfer()
|
| D | ctc_loss_v2.cc | 68 …auto type = CheckAndConvertUtils::CheckTypeValid("log_probs", input_args[0]->BuildType(), valid_ty… in CTCLossV2InferType()
|
| D | one_hot.cc | 79 (void)CheckAndConvertUtils::CheckTypeValid("depth", input_args[kInputIndex1]->BuildType(), in OneHotInferType()
|
| D | gather.cc | 45 …(void)CheckAndConvertUtils::CheckTypeValid("axis_type", input_args[kInputIndex2]->BuildType(), int… in GatherInfer()
|
| D | conv2d.cc | 243 …auto out_type = CheckAndConvertUtils::CheckTypeValid("x", input_args[0]->BuildType(), valid_types,… in Conv2dInferType()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.cc | 578 return CheckTypeValid(args.begin()->first, arg_, valid_values, prim_name); in CheckScalarOrTensorTypesSame()
629 TypePtr CheckAndConvertUtils::CheckTypeValid(const std::string &arg_name, const TypePtr &arg_type, in CheckTypeValid() function in mindspore::CheckAndConvertUtils
668 (void)CheckTypeValid("name", name->BuildType(), {kString}, class_name); in CheckSummaryParam()
674 (void)CheckTypeValid("value", value->BuildType(), {kTensorType}, class_name); in CheckSummaryParam()
|
| D | check_convert_utils.h | 293 static TypePtr CheckTypeValid(const std::string &arg_name, const TypePtr &arg_type,
|