| 448 | } |
| 449 | |
| 450 | StatusOr<Shape> Client::GetShape(const GlobalData& data) { |
| 451 | GetShapeRequest request; |
| 452 | *request.mutable_data() = data.handle(); |
| 453 | GetShapeResponse response; |
| 454 | |
| 455 | VLOG(1) << "making get shape request"; |
| 456 | Status s = stub_->GetShape(&request, &response); |
| 457 | VLOG(1) << "done with request"; |
| 458 | |
| 459 | if (!s.ok()) { |
| 460 | return s; |
| 461 | } |
| 462 | |
| 463 | return Shape(response.shape()); |
| 464 | } |
| 465 | |
| 466 | StatusOr<string> Client::ExecutionStatsAsString( |
| 467 | const XlaComputation& computation, const ExecutionProfile& profile) { |