Searched refs:split_compile_and_shard (Results 1 – 2 of 2) sorted by relevance
1085 def split_compile_and_shard(computation, function1303 return split_compile_and_shard(
3180 (compile_op, loss,) = tpu.split_compile_and_shard(3205 (compile_op, loss,) = tpu.split_compile_and_shard(3235 (compile_op, dummy_predict_op,) = tpu.split_compile_and_shard(