Home
last modified time | relevance | path

Searched defs:parallel_shape (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dinference_weight_preprocess_utils.cc74 auto parallel_shape = common::AnfAlgo::GetOutputInferShape(weight_node, kIndex0); in CheckFusionValid() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dfold_pipeline_transformer.cc393 BaseShapePtr parallel_shape = std::make_shared<abstract::Shape>(slice_shape); in InsertReceive() local
Dpipeline_transformer.cc1110 BaseShapePtr parallel_shape = std::make_shared<abstract::Shape>(slice_shape); in InsertReceive() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc730 std::shared_ptr<abstract::BaseShape> parallel_shape = nullptr; in SetClonedTensorShapeForOptimizer() local
1511 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(opt_shard_… in HandleCameAndAdaFactorOpt() local
Dstep_parallel.cc2542 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in SplitSens() local