Home
last modified time | relevance | path

Searched defs:uniform_split (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dget_parallel_info.cc281 bool uniform_split = tensor_layout->uniform_split(); in GetParameterLayoutFromGraph() local
309 bool uniform_split = layout->get_uniform_split(); in GetParameterLayoutFromResource() local
/third_party/mindspore/mindspore-src/source/mindspore/core/load_mindir/
Dload_model.h42 void set_uniform_split(bool uniform_split) { uniform_split_ = uniform_split; } in set_uniform_split()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_layout.h64 bool uniform_split() const { return uniform_split_; } in uniform_split() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dcompile_cache_manager.cc61 bool uniform_split = tensor_layout->uniform_split(); in BuildLayout() local
/third_party/mindspore/mindspore-src/source/mindspore/core/proto/
Dmind_ir.proto235 optional bool uniform_split = 6; field
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc498 bool uniform_split = tensor_layout->uniform_split(); in SliceParameterObj() local
553 bool uniform_split = tensor_layout->uniform_split(); in SliceTensorObj() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/train/
Dserialization.py1693 def _check_param_for_integrate_save(pipeline_stages, uniform_split): argument