Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/
Dstrategy_checkpoint_info.cc51 auto param_split_shape = layout_j.value().at("param_split_shape").get<std::vector<int64_t>>(); in FromJson() local
162 straspb::ParamSplitShape *param_split_shape = parallel_layouts->add_param_split_shape(); in to_protobuf() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dnode_strategy.proto54 repeated ParamSplitShape param_split_shape = 3; field
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dgather_info.cc57 auto param_split_shape = static_cast<int64_t>(GetValue<int64_t>(ele)); in GetManualSplitWithoutOffsetAttr() local