blob: 24ce84256010225a10402f19342ab063b5350c85 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
#include "tensorflow/core/public/status.h"
#include <stdio.h>
namespace tensorflow {
Status::Status(tensorflow::error::Code code, StringPiece msg) {
assert(code != tensorflow::error::OK);
state_ = new State;
state_->code = code;
state_->msg = msg.ToString();
}
Status::~Status() { delete state_; }
void Status::Update(const Status& new_status) {
if (ok()) {
*this = new_status;
}
}
void Status::SlowCopyFrom(const State* src) {
delete state_;
if (src == nullptr) {
state_ = nullptr;
} else {
state_ = new State(*src);
}
}
const string& Status::empty_string() {
static string* empty = new string;
return *empty;
}
string Status::ToString() const {
if (state_ == NULL) {
return "OK";
} else {
char tmp[30];
const char* type;
switch (code()) {
case tensorflow::error::CANCELLED:
type = "Cancelled";
break;
case tensorflow::error::UNKNOWN:
type = "Unknown";
break;
case tensorflow::error::INVALID_ARGUMENT:
type = "Invalid argument";
break;
case tensorflow::error::DEADLINE_EXCEEDED:
type = "Deadline exceeded";
break;
case tensorflow::error::NOT_FOUND:
type = "Not found";
break;
case tensorflow::error::ALREADY_EXISTS:
type = "Already exists";
break;
case tensorflow::error::PERMISSION_DENIED:
type = "Permission denied";
break;
case tensorflow::error::UNAUTHENTICATED:
type = "Unauthenticated";
break;
case tensorflow::error::RESOURCE_EXHAUSTED:
type = "Resource exhausted";
break;
case tensorflow::error::FAILED_PRECONDITION:
type = "Failed precondition";
break;
case tensorflow::error::ABORTED:
type = "Aborted";
break;
case tensorflow::error::OUT_OF_RANGE:
type = "Out of range";
break;
case tensorflow::error::UNIMPLEMENTED:
type = "Unimplemented";
break;
case tensorflow::error::INTERNAL:
type = "Internal";
break;
case tensorflow::error::UNAVAILABLE:
type = "Unavailable";
break;
case tensorflow::error::DATA_LOSS:
type = "Data loss";
break;
default:
snprintf(tmp, sizeof(tmp), "Unknown code(%d)",
static_cast<int>(code()));
type = tmp;
break;
}
string result(type);
result += ": ";
result += state_->msg;
return result;
}
}
std::ostream& operator<<(std::ostream& os, const Status& x) {
os << x.ToString();
return os;
}
} // namespace tensorflow
|
