Searched refs:SetNodeDeviceForTPUCommunication (Results 1 – 1 of 1) sorted by relevance
295 Status SetNodeDeviceForTPUCommunication(DeviceNameUtils::ParsedName device, in SetNodeDeviceForTPUCommunication() function3481 SetNodeDeviceForTPUCommunication(tpu_device, DEVICE_CPU, image)); in CopyOutsideCompilationNodes()3542 SetNodeDeviceForTPUCommunication(tpu_device, DEVICE_CPU, node)); in ReplicateOutsideCompilationNodes()