| 139 | class TransposeOpModel : public SingleOpModel { |
| 140 | public: |
| 141 | void SetInput(std::initializer_list<float> data) { |
| 142 | PopulateTensor<float>(input_, data); |
| 143 | } |
| 144 | |
| 145 | void SetPerm(std::initializer_list<int> data) { |
| 146 | PopulateTensor<int>(perm_, data); |