Home
last modified time | relevance | path

Searched defs:parameter_stage (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc782 auto parameter_stage = parameter_color_map_[parameter]; in HandleSharedParameter() local
833 …neTransformer::FillParameterStage(const CNodePtr &node, std::set<int64_t> *const parameter_stage) { in FillParameterStage()
849 std::set<int64_t> *const parameter_stage) { in GetStageByArgument()
889 std::set<int64_t> parameter_stage; in ParameterColoring() local
Dfold_pipeline_transformer.cc607 auto parameter_stage = parameter_color_map_[parameter]; in HandleSharedParameter() local
Dpipeline_interleave.cc434 std::set<int64_t> *const parameter_stage) { in GetStageByArgument()
477 std::set<int64_t> parameter_stage; in ParameterColoring() local