(self)
| 1636 | class ConvolutionTest(test_lib.TestCase): |
| 1637 | |
| 1638 | def testUnknownSize(self): |
| 1639 | # explicilty use float32 for ROCm, as MIOpen does not yet support float64 |
| 1640 | # np.ones defaults to using float64 when dtype is not explicitly specified |
| 1641 | dtype = np.float32 if test_lib.is_built_with_rocm() else np.float64 |
| 1642 | x = tensor_spec.TensorSpec(None, dtypes.float32, name="x") |
| 1643 | k = np.ones([3, 6, 6, 5], dtype=dtype) |
| 1644 | |
| 1645 | @def_function.function |
| 1646 | def F(value): |
| 1647 | return nn_ops.convolution(value, k, "SAME") |
| 1648 | |
| 1649 | F.get_concrete_function(x) |
| 1650 | |
| 1651 | |
| 1652 | class ConvTransposeTest(test_lib.TestCase): |
nothing calls this directly
no test coverage detected