Searched refs:write_if_core_0 (Results 1 – 1 of 1) sorted by relevance
1911 def write_if_core_0(step, replica_id, tt_summary): function1919 write_op = tpu.outside_compilation(write_if_core_0, step=step,