| 143 | } |
| 144 | |
| 145 | void SetPerm(std::initializer_list<int> data) { |
| 146 | PopulateTensor<int>(perm_, data); |
| 147 | } |
| 148 | |
| 149 | std::vector<float> GetOutput() { return ExtractVector<float>(output_); } |
| 150 | std::vector<int> GetOutputShape() { return GetTensorShape(output_); } |