Searched refs:parallel_shape (Results 1 – 3 of 3) sorted by relevance
515 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in SetClonedTensorShapeForOptimizer() local516 MS_EXCEPTION_IF_NULL(parallel_shape); in SetClonedTensorShapeForOptimizer()517 cloned_abstract->set_shape(parallel_shape); in SetClonedTensorShapeForOptimizer()607 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in HandleAdaFactorOpt() local608 MS_EXCEPTION_IF_NULL(parallel_shape); in HandleAdaFactorOpt()609 cloned_abstract->set_shape(parallel_shape); in HandleAdaFactorOpt()
1693 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in SetParallelShape() local1694 MS_EXCEPTION_IF_NULL(parallel_shape); in SetParallelShape()1698 cloned_abstract->set_shape(parallel_shape); in SetParallelShape()2382 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in SplitSens() local2383 MS_EXCEPTION_IF_NULL(parallel_shape); in SplitSens()2386 cloned_abstract->set_shape(parallel_shape); in SplitSens()
683 BaseShapePtr parallel_shape = std::make_shared<abstract::Shape>(slice_shape); in InsertReceive() local686 abstract_clone->set_shape(parallel_shape); in InsertReceive()