Searched refs:res_spec (Results 1 – 1 of 1) sorted by relevance
525 void CheckCustomPrimOutputInferResult(const PrimitivePtr &prim, const AbstractBasePtr &res_spec) { in CheckCustomPrimOutputInferResult() argument527 MS_EXCEPTION_IF_NULL(res_spec); in CheckCustomPrimOutputInferResult()537 if (res_spec->isa<AbstractTensor>() && output_num != 1) { in CheckCustomPrimOutputInferResult()540 << res_spec->ToString(); in CheckCustomPrimOutputInferResult()541 } else if (res_spec->isa<AbstractTuple>() && in CheckCustomPrimOutputInferResult()542 (res_spec->cast<AbstractTuplePtr>()->size() != LongToSize(output_num))) { in CheckCustomPrimOutputInferResult()544 << " not matches the infer result " << res_spec->ToString(); in CheckCustomPrimOutputInferResult()563 auto res_spec = FromValue(converted_ret); in PyInferRes2Abstract() local564 MS_EXCEPTION_IF_NULL(res_spec); in PyInferRes2Abstract()565 if (res_spec->isa<AbstractTensor>()) { in PyInferRes2Abstract()[all …]