(self, _, version)
| 624 | |
| 625 | @parameterized.expand(all_versions_for("Col2Im")) |
| 626 | def test_col2im_5d(self, _, version) -> None: |
| 627 | graph = self._make_graph( |
| 628 | [ |
| 629 | ("input", TensorProto.FLOAT, (1, 10, 12)), |
| 630 | ("output_shape", TensorProto.INT64, (3,)), |
| 631 | ("kernel_shape", TensorProto.INT64, (3,)), |
| 632 | ], |
| 633 | [ |
| 634 | make_node( |
| 635 | "Col2Im", ["input", "output_shape", "kernel_shape"], ["output"] |
| 636 | ) |
| 637 | ], |
| 638 | [], |
| 639 | initializer=[ |
| 640 | make_tensor("output_shape", TensorProto.INT64, (3,), (3, 4, 5)), |
| 641 | make_tensor("kernel_shape", TensorProto.INT64, (3,), (1, 1, 5)), |
| 642 | ], |
| 643 | ) |
| 644 | self._assert_inferred( |
| 645 | graph, |
| 646 | [make_tensor_value_info("output", TensorProto.FLOAT, (1, 2, 3, 4, 5))], |
| 647 | opset_imports=[helper.make_opsetid(ONNX_DOMAIN, version)], |
| 648 | ) |
| 649 | |
| 650 | @parameterized.expand(all_versions_for("Concat")) |
| 651 | def test_concat(self, _, version) -> None: |
nothing calls this directly
no test coverage detected