Searched refs:_test_variable_initialization (Results 1 – 1 of 1) sorted by relevance
211 def _test_variable_initialization(self, task_type, task_id, num_gpus): member in CollectiveAllReduceStrategyTestBase361 self._test_variable_initialization,444 self._test_variable_initialization,