Searched refs:chosen_device (Results 1 – 1 of 1) sorted by relevance
973 string chosen_device; in AreDevicesCompatible() local975 devices, /*allow_mixing_unknown_and_cpu=*/false, &chosen_device)); in AreDevicesCompatible()984 return resource_op_device.empty() || resource_op_device == chosen_device; in AreDevicesCompatible()1003 TF_RETURN_IF_ERROR(DeviceToDeviceType(chosen_device, &device_type)); in AreDevicesCompatible()1006 << "chosen device = " << chosen_device in AreDevicesCompatible()1028 string chosen_device; in ShouldCompileClusterImpl() local1030 devices, /*allow_mixing_unknown_and_cpu=*/false, &chosen_device)); in ShouldCompileClusterImpl()1034 TF_RETURN_IF_ERROR(DeviceToDeviceType(chosen_device, &device_type)); in ShouldCompileClusterImpl()1037 << "chosen device = " << chosen_device in ShouldCompileClusterImpl()1050 << " cluster with device " << chosen_device; in ShouldCompileClusterImpl()[all …]