Searched defs:use_tpu (Results 1 – 2 of 2) sorted by relevance
1209 def test_optimizer_with_slot_creation_fn(self, use_tpu): argument
2613 bool use_tpu; in BuildTPUReplicateOps() local