Home
last modified time | relevance | path

Searched refs:CheckSubClass (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dshape.cc40 …(void)CheckAndConvertUtils::CheckSubClass("shape type", input_args[0]->BuildType(), valid_params_t… in ShapeInfer()
56 …(void)CheckAndConvertUtils::CheckSubClass("shape type", input_args[0]->BuildType(), valid_params_t… in ShapeInferValue()
Dexpand_dims.cc54 (void)CheckAndConvertUtils::CheckSubClass("x_type", x_type, valid_x_type, prim_name); in ExpandDimsInfer()
Dselect.cc75 (void)CheckAndConvertUtils::CheckSubClass("x_type", x_type, {kTensorType}, prim_name); in SelectInferType()
76 (void)CheckAndConvertUtils::CheckSubClass("y_type", y_type, {kTensorType}, prim_name); in SelectInferType()
Dflatten.cc45 …(void)CheckAndConvertUtils::CheckSubClass("infer type", input_args[0]->BuildType(), valid_types, p… in InferType()
Dexp.cc56 …(void)CheckAndConvertUtils::CheckSubClass("x_type", input_args[0]->BuildType(), valid_params_types… in InferType()
Dreciprocal.cc57 …(void)CheckAndConvertUtils::CheckSubClass("x_type", input_args[0]->BuildType(), valid_params_types… in InferType()
Dneg.cc57 …(void)CheckAndConvertUtils::CheckSubClass("x_type", input_args[0]->BuildType(), valid_params_types… in InferType()
Dlog.cc57 …(void)CheckAndConvertUtils::CheckSubClass("x_type", input_args[0]->BuildType(), valid_params_types… in InferType()
Dunpack.cc30 …(void)CheckAndConvertUtils::CheckSubClass("x", input_args[0]->BuildType(), {TypeIdToType(kObjectTy… in UnpackInfer()
Dzeros.cc50 return CheckAndConvertUtils::CheckSubClass("dtype", output_type, valid_types, prim_name); in ZerosInferType()
Dassert.cc74 (void)CheckAndConvertUtils::CheckSubClass("input", dtype, template_types, op_name); in AssertInfer()
Dones.cc48 return CheckAndConvertUtils::CheckSubClass("dtype", output_type, valid_types, prim_name); in OnesInferType()
Dmasked_fill.cc65 …(void)CheckAndConvertUtils::CheckSubClass("value", input_args[kInputIndex2]->BuildType(), {kFloat}… in InferType()
Dpad.cc52 …return CheckAndConvertUtils::CheckSubClass("infer type", input_args[0]->BuildType(), valid_types, … in InferType()
Dlerp.cc64 …(void)CheckAndConvertUtils::CheckSubClass("weight", input_args[kInputIndex2]->BuildType(), {kFloat… in InferType()
Dgather_d.cc70 …(void)CheckAndConvertUtils::CheckSubClass("dim", input_args[kInputIndex1]->BuildType(), valid_type… in GatherDInfer()
Dpack.cc25 …(void)CheckAndConvertUtils::CheckSubClass("input_x[0]", x_types[0], {TypeIdToType(kObjectTypeTenso… in _get_pack_shape()
Dbroadcast_to.cc71 (void)CheckAndConvertUtils::CheckSubClass("x_dtype", x_dtype, template_types, prim->name()); in BroadcastToInferType()
Dtranspose.cc86 …return CheckAndConvertUtils::CheckSubClass("x", input_args[0]->BuildType(), {kTensorType}, prim->n… in InferType()
Dgather.cc40 …(void)CheckAndConvertUtils::CheckSubClass("params_type", input_args[kInputIndex0]->BuildType(), va… in GatherInfer()
/third_party/mindspore/mindspore/core/utils/
Dcheck_convert_utils.cc491 return CheckSubClass(input_names, check_type, check_list, prim_name); in CheckTensorTypeSame()
512 return CheckSubClass(type_name, type, check_list, prim_name); in CheckTensorTypeValid()
548 TypePtr CheckAndConvertUtils::CheckSubClass(const std::string &type_name, const TypePtr &type, in CheckSubClass() function in mindspore::CheckAndConvertUtils
638 return CheckSubClass(arg_name, arg_type, valid_type, prim_name); in CheckTypeValid()
Dcheck_convert_utils.h288 static TypePtr CheckSubClass(const std::string &type_name, const TypePtr &type,