| 81 | } |
| 82 | |
| 83 | inline Status FromGrpcStatus(const ::grpc::Status& s) { |
| 84 | if (s.ok()) { |
| 85 | return Status::OK(); |
| 86 | } else { |
| 87 | // Convert "UNKNOWN" stream removed errors into unavailable, to allow |
| 88 | // for retry upstream. |
| 89 | if (IsStreamRemovedError(s)) { |
| 90 | return Status(tensorflow::error::UNAVAILABLE, s.error_message()); |
| 91 | } |
| 92 | return Status(static_cast<tensorflow::error::Code>(s.error_code()), |
| 93 | s.error_message()); |
| 94 | } |
| 95 | } |
| 96 | |
| 97 | inline ::grpc::Status ToGrpcStatus(const ::tensorflow::Status& s) { |
| 98 | if (s.ok()) { |