Searched refs:use_parallel (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/tests/st/hcom/ |
| D | hcom_sparsetensor.py | 37 random_offset=0, use_parallel=False, fakedata_mode=FakeDataInitMode.RandomInit): argument
53 if use_parallel:
146 def train_mindspore_impl(self, indices, epoch, batch_size, use_parallel=True): argument
147 …ds = FakeData(size=8, batch_size=batch_size, num_class=8, image_size=(), use_parallel=use_parallel)
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | optimizer_parallel.py | 86 num_classes=10, random_offset=0, use_parallel=False, argument
102 if use_parallel is True:
316 use_parallel=True,
322 use_parallel=True,
|
| D | parallel_strategy_search.py | 89 num_classes=10, random_offset=0, use_parallel=False, argument
105 if use_parallel is True:
352 image_size=(3, 224, 224), use_parallel=True,
379 image_size=(3, 224, 224), use_parallel=True,
|
| D | multifieldembeddinglookup_parallel.py | 84 num_classes=10, random_offset=0, use_parallel=False, argument
100 if use_parallel is True:
311 parallel_dataset = FakeData(size=64, batch_size=8, image_size=(64,), use_parallel=True)
|
| /third_party/mindspore/tests/st/mix_precision/ |
| D | utils.py | 68 num_classes=10, random_offset=0, use_parallel=False, argument
84 if use_parallel is True:
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | optimizer.py | 210 self.use_parallel = True
217 self.use_parallel = False
219 self.use_parallel = False
220 if self.use_parallel:
|
| D | lamb.py | 333 if self.use_parallel:
|
| D | adam.py | 508 if self.use_parallel:
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | adam.py | 316 if self.use_parallel:
418 if self.use_parallel:
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | adam.py | 316 if self.use_parallel:
418 if self.use_parallel:
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | adam.py | 315 if self.use_parallel:
417 if self.use_parallel:
|