| 70 | } |
| 71 | |
| 72 | void SetStringInput(std::initializer_list<string> data) { |
| 73 | PopulateStringTensor(input_, data); |
| 74 | } |
| 75 | |
| 76 | std::vector<T> GetOutput() { return ExtractVector<T>(output_); } |
| 77 | std::vector<int> GetOutputShape() { return GetTensorShape(output_); } |