Searched defs:is_tpu (Results 1 – 2 of 2) sorted by relevance
221 const bool is_tpu = all_reduce.device_type().endswith("TPU"); in LowerAllReduceOpImpl() local
347 const bool is_tpu = device != nullptr && device->device_type() == "TPU"; in GetDeviceForInput() local