| 212 | } |
| 213 | |
| 214 | void TfDriver::ReshapeTensor(int id, const string& values_as_string) { |
| 215 | input_shapes_[id] = Split<int64_t>(values_as_string, ","); |
| 216 | input_tensors_[input_names_[id]] = |
| 217 | CreateTensor(input_types_[id], input_shapes_[id]); |
| 218 | ResetTensor(id); |
| 219 | } |
| 220 | |
| 221 | string TfDriver::ReadOutput(const tensorflow::Tensor& tensor) { |
| 222 | switch (tensor.dtype()) { |