Searched refs:base_shape (Results 1 – 8 of 8) sorted by relevance
31 auto base_shape = node->Shape(); in IsNodeOutputDynamicShape() local32 if (base_shape == nullptr) { in IsNodeOutputDynamicShape()36 …if (base_shape->isa<abstract::Shape>() && IsShapeDynamic(base_shape->cast<abstract::ShapePtr>())) { in IsNodeOutputDynamicShape()38 } else if (base_shape->isa<abstract::TupleShape>()) { in IsNodeOutputDynamicShape()39 auto tuple_shape = base_shape->cast<abstract::TupleShapePtr>(); in IsNodeOutputDynamicShape()
381 AbstractBasePtr MakeAbstract(const BaseShapePtr &base_shape, const TypePtr &type) { in MakeAbstract() argument382 MS_EXCEPTION_IF_NULL(base_shape); in MakeAbstract()384 if ((base_shape->isa<Shape>())) { in MakeAbstract()385 auto shape = base_shape->cast<ShapePtr>(); in MakeAbstract()394 } else if (base_shape->isa<TupleShape>() && type->isa<Tuple>()) { in MakeAbstract()395 auto shape_tuple = base_shape->cast<TupleShapePtr>(); in MakeAbstract()404 } else if (base_shape->isa<ListShape>() && type->isa<List>()) { in MakeAbstract()405 auto shape_list = base_shape->cast<ListShapePtr>(); in MakeAbstract()414 } else if (base_shape->isa<NoShape>() && type->isa<TypeNone>()) { in MakeAbstract()429 …MS_LOG(EXCEPTION) << "evaluator return invalid shape " << base_shape->ToString() << "or type. " <<… in MakeAbstract()
65 AbstractBasePtr MakeAbstract(const BaseShapePtr &base_shape, const TypePtr &type);
156 BaseShapePtr base_shape = std::make_shared<abstract::Shape>(each_shape); in InferShape() local157 MS_EXCEPTION_IF_NULL(base_shape); in InferShape()158 base_shape_list.push_back(base_shape); in InferShape()
807 … const abstract::BaseShapePtr &base_shape, in GetOutputInferShape() argument810 MS_EXCEPTION_IF_NULL(base_shape); in GetOutputInferShape()811 if (base_shape->isa<abstract::Shape>()) { in GetOutputInferShape()813 return TransShapeToSizet(base_shape->cast<abstract::ShapePtr>()); in GetOutputInferShape()818 } else if (base_shape->isa<abstract::TupleShape>()) { in GetOutputInferShape()819 auto tuple_shape = base_shape->cast<abstract::TupleShapePtr>(); in GetOutputInferShape()833 … << " should be a NoShape , ArrayShape or a TupleShape, but it is " << base_shape->ToString() in GetOutputInferShape()837 } else if (base_shape->isa<abstract::NoShape>()) { in GetOutputInferShape()841 << base_shape->ToString() << " node : " << node->DebugString() in GetOutputInferShape()1166 auto base_shape = node->Shape(); in GetOutputDetailShape() local[all …]
160 …ector<size_t> GetOutputInferShape(const AnfNodePtr &node, const abstract::BaseShapePtr &base_shape,
249 std::shared_ptr<abstract::BaseShape> base_shape = std::make_shared<abstract::Shape>(shape); in ParallelParameterContextRestoreShape() local250 ptr->set_shape(base_shape); in ParallelParameterContextRestoreShape()
422 auto base_shape = abstract->BuildShape(); in GetTensorInputShape() local423 MS_EXCEPTION_IF_NULL(base_shape); in GetTensorInputShape()424 if (!base_shape->isa<abstract::Shape>()) { in GetTensorInputShape()427 auto shape = base_shape->cast<abstract::ShapePtr>(); in GetTensorInputShape()