| 632 | } |
| 633 | |
| 634 | TF_Tensor* TF_CheckpointReaderGetTensor(TF_CheckpointReader* reader, |
| 635 | const char* name, TF_Status* status) { |
| 636 | std::unique_ptr<tensorflow::Tensor> tensor; |
| 637 | reader->GetTensor(name, &tensor, status); |
| 638 | if (!status->status.ok()) return nullptr; |
| 639 | return tensorflow::TF_TensorFromTensor(*tensor, status); |
| 640 | } |
| 641 | |
| 642 | void TF_CheckpointReaderGetVariableShape(TF_CheckpointReader* reader, |
| 643 | const char* name, int64_t* dims, |
nothing calls this directly
no test coverage detected