Searched refs:is_tpu (Results 1 – 3 of 3) sorted by relevance
56 is_tpu = backend.is_tpu_strategy(strategy)57 if ((not is_tpu) and strategy and ds_context.in_cross_replica_context()):
418 combinations.combine(is_tpu=[False])) + combinations.combine(422 is_tpu=[True]))423 def testRunStepsWithOutputContext(self, distribution, optimizer_fn, is_tpu): argument
291 const bool is_tpu = device != nullptr && device->device_type() == "TPU"; in GetDeviceForInput() local294 is_tpu ? MTypeFromDTypeIntsOnDevice(tensor_handle->dtype) in GetDeviceForInput()