| 26 | void TF_DeleteStatus(TF_Status* s) { delete s; } |
| 27 | |
| 28 | void TF_SetStatus(TF_Status* s, TF_Code code, const char* msg) { |
| 29 | if (code == TF_OK) { |
| 30 | s->status = Status::OK(); |
| 31 | return; |
| 32 | } |
| 33 | s->status = Status(static_cast<Code>(code), tensorflow::StringPiece(msg)); |
| 34 | } |
| 35 | |
| 36 | TF_Code TF_GetCode(const TF_Status* s) { |
| 37 | return static_cast<TF_Code>(s->status.code()); |