Searched defs:is_tpu (Results 1 – 2 of 2) sorted by relevance
423 def testRunStepsWithOutputContext(self, distribution, optimizer_fn, is_tpu): argument
291 const bool is_tpu = device != nullptr && device->device_type() == "TPU"; in GetDeviceForInput() local