Searched defs:tpu_device_assignment (Results 1 – 3 of 3) sorted by relevance
330 const auto& tpu_device_assignment = status_or.ValueOrDie(); in TEST() local433 const auto& tpu_device_assignment = status_or.ValueOrDie(); in TEST() local546 auto& tpu_device_assignment = status_or.ValueOrDie(); in TEST() local
536 auto& tpu_device_assignment = status_or_tpu_device_assignment.ValueOrDie(); in GetHostDeviceOutsideComputation() local
604 auto& tpu_device_assignment = status_or_tpu_device_assignment.ValueOrDie(); in Rewrite() local