Lines Matching refs:test_shape
5534 def test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners): function
5614 test_shape(N, C, H, W, H, W, mode, padding_mode, align_corners)
5623 test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners)
5632 test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners)
5641 test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners)
5649 test_shape(N, C, IH, IW, 0, W, mode, padding_mode, align_corners)
5657 test_shape(N, 0, IH, IW, H, W, mode, padding_mode, align_corners)
5665 test_shape(0, C, IH, IW, H, W, mode, padding_mode, align_corners)
5895 def test_shape(N, C, ID, IH, IW, D, H, W, mode, padding_mode, align_corners): function
5931 test_shape(N, C, D, H, W, D, H, W, mode, padding_mode, align_corners)
5942 test_shape(N, C, ID, IH, IW, D, H, W, mode, padding_mode, align_corners)
5953 test_shape(N, C, ID, IH, IW, D, H, W, mode, padding_mode, align_corners)
5963 test_shape(N, C, ID, IH, IW, D, H, W, mode, padding_mode, align_corners)
5973 test_shape(N, C, ID, IH, IW, D, 0, W, mode, padding_mode, align_corners)
5983 test_shape(N, 0, ID, IH, IW, D, H, W, mode, padding_mode, align_corners)
5993 test_shape(0, C, ID, IH, IW, D, H, W, mode, padding_mode, align_corners)