Searched refs:device_assignment_proto (Results 1 – 1 of 1) sorted by relevance
419 xla::DeviceAssignmentProto device_assignment_proto; in GetGeneralTPUExecutionDeviceAssignment() local420 TF_RETURN_IF_ERROR(device_assignment.Serialize(&device_assignment_proto)); in GetGeneralTPUExecutionDeviceAssignment()423 std::move(devices_and_hosts), std::move(device_assignment_proto)); in GetGeneralTPUExecutionDeviceAssignment()