Home
last modified time | relevance | path

Searched refs:fakedata_mode (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/tests/st/mix_precision/
Dutils.py69 fakedata_mode=FakeDataInitMode.RandomInit): argument
82 self.fakedata_mode = fakedata_mode
119 if self.fakedata_mode == FakeDataInitMode.OnesInit:
121 elif self.fakedata_mode == FakeDataInitMode.ZerosInit:
123 elif self.fakedata_mode == FakeDataInitMode.UniqueInit:
Dtest_mix_precision.py119 fakedata_mode=FakeDataInitMode.OnesInit)
153 fakedata_mode=FakeDataInitMode.OnesInit)
158 fakedata_mode=FakeDataInitMode.OnesInit)
/third_party/mindspore/tests/st/hcom/
Dhcom_sparsetensor.py37 random_offset=0, use_parallel=False, fakedata_mode=FakeDataInitMode.RandomInit): argument
51 self.fakedata_mode = fakedata_mode
87 if self.fakedata_mode == FakeDataInitMode.OnesInit:
89 elif self.fakedata_mode == FakeDataInitMode.ZerosInit:
91 elif self.fakedata_mode == FakeDataInitMode.UniqueInit:
/third_party/mindspore/tests/st/auto_parallel/
Doptimizer_parallel.py87 fakedata_mode=FakeDataInitMode.RandomInit): argument
100 self.fakedata_mode = fakedata_mode
137 if self.fakedata_mode == FakeDataInitMode.OnesInit:
139 elif self.fakedata_mode == FakeDataInitMode.ZerosInit:
141 elif self.fakedata_mode == FakeDataInitMode.UniqueInit:
Dmultifieldembeddinglookup_parallel.py85 fakedata_mode=FakeDataInitMode.RandomInit): argument
98 self.fakedata_mode = fakedata_mode
135 if self.fakedata_mode == FakeDataInitMode.OnesInit:
137 elif self.fakedata_mode == FakeDataInitMode.ZerosInit:
139 elif self.fakedata_mode == FakeDataInitMode.UniqueInit:
Dparallel_strategy_search.py90 fakedata_mode=FakeDataInitMode.RandomInit): argument
103 self.fakedata_mode = fakedata_mode
140 if self.fakedata_mode == FakeDataInitMode.OnesInit:
142 elif self.fakedata_mode == FakeDataInitMode.ZerosInit:
144 elif self.fakedata_mode == FakeDataInitMode.UniqueInit: