Searched refs:get_barrier (Results 1 – 8 of 8) sorted by relevance
215 multi_process_runner.get_barrier().wait()229 multi_process_runner.get_barrier().wait()317 multi_process_runner.get_barrier().wait()325 multi_process_runner.get_barrier().wait()
77 multi_process_runner.get_barrier().wait()110 multi_process_runner.get_barrier().wait()
57 return multi_process_runner.get_barrier()299 multi_process_runner.get_barrier()
159 multi_process_runner.get_barrier().wait()
1304 def get_barrier(): function
56 multi_process_runner.get_barrier().wait()58 multi_process_runner.get_barrier().wait()86 multi_process_runner.get_barrier().wait()
20 name: "get_barrier"
174 tf.__internal__.distribute.multi_process_runner.get_barrier().wait()195 tf.__internal__.distribute.multi_process_runner.get_barrier().wait()