| 193 | }; |
| 194 | |
| 195 | class LargeGatherDataset final : public GatherDataset |
| 196 | { |
| 197 | public: |
| 198 | LargeGatherDataset() |
| 199 | { |
| 200 | // 2D input |
| 201 | add_config(TensorShape(150U, 150U), TensorShape(50U), 0); |
| 202 | add_config(TensorShape(150U, 150U), TensorShape(50U), 1); |
| 203 | add_config(TensorShape(150U, 150U), TensorShape(50U), -1); |
| 204 | |
| 205 | // 3D input |
| 206 | add_config(TensorShape(50U, 40U, 60U), TensorShape(33U), 0); |
| 207 | add_config(TensorShape(40U, 50U, 60U), TensorShape(24U), 1); |
| 208 | add_config(TensorShape(70U, 80U, 100U), TensorShape(50U), 2); |
| 209 | add_config(TensorShape(40U, 50U, 60U), TensorShape(24U), -1); |
| 210 | add_config(TensorShape(70U, 80U, 100U), TensorShape(50U), -2); |
| 211 | |
| 212 | // 4D input |
| 213 | add_config(TensorShape(30U, 40U, 20U, 20U), TensorShape(33U), 0); |
| 214 | add_config(TensorShape(23U, 10U, 60U, 20U), TensorShape(24U), 1); |
| 215 | add_config(TensorShape(14U, 20U, 10U, 31U), TensorShape(30U), 2); |
| 216 | add_config(TensorShape(34U, 10U, 40U, 20U), TensorShape(50U), 3); |
| 217 | add_config(TensorShape(23U, 10U, 60U, 20U), TensorShape(24U), -1); |
| 218 | add_config(TensorShape(14U, 20U, 10U, 31U), TensorShape(30U), -2); |
| 219 | add_config(TensorShape(34U, 10U, 40U, 20U), TensorShape(50U), -3); |
| 220 | } |
| 221 | }; |
| 222 | |
| 223 | } // namespace datasets |
| 224 | } // namespace test |
no outgoing calls
no test coverage detected