Searched defs:enable_packed_var (Results 1 – 1 of 1) sorted by relevance
81 def get_tpu_strategy(enable_packed_var=False): argument258 def test_handle_in_cross_replica_context(self, enable_packed_var): argument271 def testStaticHashTableDatasetFnHostTrainingLoop(self, enable_packed_var): argument315 def test_function_compile_with_xla(self, enable_packed_var): argument333 def test_sequential_runs(self, enable_packed_var): argument363 def test_device_switch_case(self, enable_packed_var): argument390 def test_recover_from_compilation_failures(self, enable_packed_var): argument422 self, enable_packed_var): argument444 def test_computation_on_subset_cores(self, enable_packed_var): argument485 def test_worker_devices_on_subset_cores(self, enable_packed_var): argument[all …]