Searched refs:BaseShape (Results 1 – 6 of 6) sorted by relevance
36 class BaseShape; variable37 using BaseShapePtr = std::shared_ptr<BaseShape>;40 class MS_CORE_API BaseShape : public Base {42 BaseShape() = default;43 ~BaseShape() override = default;45 MS_DECLARE_PARENT(BaseShape, Base)46 virtual bool operator==(const BaseShape &other) const;47 bool operator!=(const BaseShape &other) const;56 class MS_CORE_API NoShape : public BaseShape {58 MS_DECLARE_PARENT(NoShape, BaseShape) in MS_DECLARE_PARENT() argument[all …]
46 std::ostream &operator<<(std::ostream &os, const BaseShape &bs) { in operator <<()51 std::ostream &operator<<(std::ostream &os, const std::shared_ptr<BaseShape> bs) { in operator <<()57 bool BaseShape::operator==(const BaseShape &other) const { in operator ==()64 bool BaseShape::operator!=(const BaseShape &other) const { return !(*this == other); } in operator !=()96 bool Shape::operator==(const BaseShape &other) const { in operator ==()135 bool SequeueShape::SequeueEqual(const BaseShape &other) const { in SequeueEqual()152 template bool SequeueShape::SequeueEqual<TupleShape>(const BaseShape &) const;153 template bool SequeueShape::SequeueEqual<ListShape>(const BaseShape &) const;
515 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in SetClonedTensorShapeForOptimizer()607 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in HandleAdaFactorOpt()
249 std::shared_ptr<abstract::BaseShape> base_shape = std::make_shared<abstract::Shape>(shape); in ParallelParameterContextRestoreShape()
905 …std::shared_ptr<abstract::BaseShape> virtual_output_shape = std::make_shared<abstract::Shape>(shap… in InsertVirtualOutput()1693 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in SetParallelShape()2382 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in SplitSens()
51 class BaseShape; variable54 using BaseShapePtr = std::shared_ptr<abstract::BaseShape>;