Data set containing pairs of small tensor shapes that are broadcast compatible in non-x dimensions. */
| 295 | |
| 296 | /** Data set containing pairs of small tensor shapes that are broadcast compatible in non-x dimensions. */ |
| 297 | class SmallShapesNonXBroadcast final : public framework::dataset::ZipDataset<ShapeDataset, ShapeDataset> |
| 298 | { |
| 299 | public: |
| 300 | SmallShapesNonXBroadcast() |
| 301 | : ZipDataset<ShapeDataset, ShapeDataset>( |
| 302 | ShapeDataset("Shape0", |
| 303 | {TensorShape{9U, 9U}, TensorShape{128U, 1U, 5U, 3U}, TensorShape{9U, 9U, 3U, 4U}, |
| 304 | TensorShape{1U, 16U, 10U, 2U, 128U}}), |
| 305 | ShapeDataset("Shape1", |
| 306 | {TensorShape{9U, 1U, 2U}, TensorShape{128U, 64U, 1U, 3U}, TensorShape{9U, 1U, 3U}, |
| 307 | TensorShape{1U, 1U, 1U, 1U, 128U}})) |
| 308 | { |
| 309 | } |
| 310 | }; |
| 311 | |
| 312 | class TemporaryLimitedSmallShapesBroadcast final : public framework::dataset::ZipDataset<ShapeDataset, ShapeDataset> |
| 313 | { |
no outgoing calls
no test coverage detected