Home
last modified time | relevance | path

Searched defs:full_batch (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dvirtual_output_info.cc58 bool full_batch = ParallelContext::GetInstance()->full_batch(); in GenerateStrategies() local
Dvirtual_dataset_info.cc160 bool full_batch = ParallelContext::GetInstance()->full_batch(); in GenerateStrategies() local
Dget_next_info.cc125 bool full_batch = ParallelContext::GetInstance()->full_batch(); in CheckStrategy() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcontext.h66 bool full_batch() const { return full_batch_; } in full_batch() function
Dcontext.cc92 void ParallelContext::set_full_batch(bool full_batch) { full_batch_ = full_batch; } in set_full_batch()
Dparameter_manager.cc339 bool full_batch = ParallelContext::GetInstance()->full_batch(); in HandleNoUsedParameter() local
Dstep_parallel.cc1736 bool full_batch = ParallelContext::GetInstance()->full_batch(); in SetVirtualDatasetStrategy() local
/third_party/mindspore/mindspore/parallel/
D_auto_parallel_context.py249 def set_full_batch(self, full_batch): argument
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc1608 bool full_batch = parallel::ParallelContext::GetInstance()->full_batch(); in GetFirstPSEmbeddingCache() local