| 219 | } |
| 220 | |
| 221 | string TfDriver::ReadOutput(const tensorflow::Tensor& tensor) { |
| 222 | switch (tensor.dtype()) { |
| 223 | case tensorflow::DT_FLOAT: |
| 224 | return TensorDataToCsvString<float>(tensor); |
| 225 | case tensorflow::DT_INT32: |
| 226 | return TensorDataToCsvString<int32_t>(tensor); |
| 227 | case tensorflow::DT_INT64: |
| 228 | return TensorDataToCsvString<tensorflow::int64>(tensor); |
| 229 | case tensorflow::DT_UINT8: |
| 230 | return TensorDataToCsvString<uint8_t>(tensor); |
| 231 | case tensorflow::DT_STRING: |
| 232 | return TensorDataToTfLiteHexString(tensor); |
| 233 | case tensorflow::DT_BOOL: |
| 234 | return TensorDataToCsvString<bool>(tensor); |
| 235 | default: |
| 236 | Invalidate(absl::StrCat("Unsupported tensor type ", |
| 237 | tensorflow::DataType_Name(tensor.dtype()), |
| 238 | " in ReadOutput")); |
| 239 | return ""; |
| 240 | } |
| 241 | } |
| 242 | |
| 243 | string TfDriver::ReadOutput(int id) { |
| 244 | if (!IsValid()) return ""; |