Data set containing pairs of large tensor shapes that are broadcast compatible. */
| 425 | |
| 426 | /** Data set containing pairs of large tensor shapes that are broadcast compatible. */ |
| 427 | class LargeShapesBroadcast final : public framework::dataset::ZipDataset<ShapeDataset, ShapeDataset> |
| 428 | { |
| 429 | public: |
| 430 | LargeShapesBroadcast() |
| 431 | : ZipDataset<ShapeDataset, ShapeDataset>( |
| 432 | ShapeDataset( |
| 433 | "Shape0", |
| 434 | {TensorShape{1921U, 541U}, TensorShape{1U, 485U, 2U, 3U}, TensorShape{4159U, 1U}, TensorShape{799U}}), |
| 435 | ShapeDataset("Shape1", |
| 436 | {TensorShape{1921U, 1U, 2U}, TensorShape{641U, 1U, 2U, 3U}, TensorShape{1U, 127U, 25U}, |
| 437 | TensorShape{799U, 595U, 1U, 4U}})) |
| 438 | { |
| 439 | } |
| 440 | }; |
| 441 | |
| 442 | /** Data set containing large 1D tensor shapes. */ |
| 443 | class Large1DShapes final : public ShapeDataset |
no outgoing calls
no test coverage detected