Home
last modified time | relevance | path

Searched refs:base_shape (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/core/ir/
Danf_utils.cc31 auto base_shape = node->Shape(); in IsNodeOutputDynamicShape() local
32 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()
/third_party/mindspore/mindspore/core/abstract/
Dutils.cc381 AbstractBasePtr MakeAbstract(const BaseShapePtr &base_shape, const TypePtr &type) { in MakeAbstract() argument
382 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()
Dutils.h65 AbstractBasePtr MakeAbstract(const BaseShapePtr &base_shape, const TypePtr &type);
/third_party/mindspore/mindspore/core/ops/
Dneighborexchange.cc156 BaseShapePtr base_shape = std::make_shared<abstract::Shape>(each_shape); in InferShape() local
157 MS_EXCEPTION_IF_NULL(base_shape); in InferShape()
158 base_shape_list.push_back(base_shape); in InferShape()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc807 … const abstract::BaseShapePtr &base_shape, in GetOutputInferShape() argument
810 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 …]
Danf_runtime_algorithm.h160 …ector<size_t> GetOutputInferShape(const AnfNodePtr &node, const abstract::BaseShapePtr &base_shape,
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcontext.cc249 std::shared_ptr<abstract::BaseShape> base_shape = std::make_shared<abstract::Shape>(shape); in ParallelParameterContextRestoreShape() local
250 ptr->set_shape(base_shape); in ParallelParameterContextRestoreShape()
/third_party/mindspore/mindspore/core/utils/
Dcheck_convert_utils.cc422 auto base_shape = abstract->BuildShape(); in GetTensorInputShape() local
423 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()