| 150 | } |
| 151 | |
| 152 | std::vector<SliceArgs<OutputType, Dims>> GenerateArgs( |
| 153 | const InListCPU<InputType, Dims>& input_tlv) { |
| 154 | ArgsGenerator generator; |
| 155 | std::vector<SliceArgs<OutputType, Dims>> slice_args; |
| 156 | for (int i = 0; i < NumSamples; i++) { |
| 157 | auto shape = input_tlv.tensor_shape(i); |
| 158 | slice_args.push_back(generator.Get(shape)); |
| 159 | } |
| 160 | return slice_args; |
| 161 | } |
| 162 | |
| 163 | virtual void Run() = 0; |
| 164 | }; |
no test coverage detected