| 63 | } |
| 64 | |
| 65 | ParsedTensor::Kind get_kind(const std::string &tok) |
| 66 | { |
| 67 | if (tok == "input") |
| 68 | return ParsedTensor::Kind::Input; |
| 69 | else if (tok == "output") |
| 70 | return ParsedTensor::Kind::Output; |
| 71 | else |
| 72 | throw oops::UserExn("Unrecognizable token", "token", tok); |
| 73 | } |
| 74 | |
| 75 | TF_DataType get_dtype(const std::string &tok) |
| 76 | { |