Home
last modified time | relevance | path

Searched refs:device_and_host (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc412 auto& device_and_host = devices_and_hosts[replica][logical_core]; in GetGeneralTPUExecutionDeviceAssignment() local
414 device_and_host.device = DeviceNameUtils::ParsedNameToString(tpu_device); in GetGeneralTPUExecutionDeviceAssignment()
415 device_and_host.host = GetCPUHostDeviceForTPUDevice(tpu_device); in GetGeneralTPUExecutionDeviceAssignment()