Lines Matching refs:test_shape
289 def test_shape(sparse_dims, nnz, with_size): function
309 test_shape(3, 10, 100)
310 test_shape(3, 10, [100, 100, 100])
311 test_shape(3, 10, [100, 100, 100, 5, 5, 5, 0])
312 test_shape(3, 0, [0, 0, 100, 5, 5, 5, 0])
811 def test_shape(sparse_dims, nnz, with_size): function
822 test_shape(4, 20, 5)
823 test_shape(3, 10, [100, 100, 100, 5, 5, 5, 0])
824 test_shape(3, 0, [0, 0, 100, 5, 5, 5, 0])
930 def test_shape(sparse_dims, nnz, with_size): function
943 test_shape(4, 6, 3)
944 test_shape(4, 3, [7, 7, 7, 3, 3, 3, 0])
945 test_shape(4, 0, [0, 0, 7, 3, 3, 3, 0])
964 def test_shape(sparse_dims, nnz, with_size): function
995 test_shape(2, 3, [2, 3, 4, 5])
996 test_shape(2, 3, [2, 2, 0])
999 test_shape(3, 0, [0, 0, 2])
1005 def test_shape(di, dj, dk, nnz): function
1020 test_shape(10, 20, 30, 20)
1021 test_shape(0, 20, 30, 0)
1022 test_shape(10, 0, 30, 0)
1023 test_shape(10, 20, 0, 0)
1024 test_shape(10, 20, 0, 20)
1058 def test_shape(sparse_dims, nnz, sizes): function
1066 test_shape(1, 20, [1])
1067 test_shape(4, 20, [3, 17, 19, 5])
1068 test_shape(2, 20, [3, 17, 19, 5])
1069 test_shape(2, 20, [3, 17, 19, 0])
1128 def test_shape(sparse_dims, nnz, sizes, unsqueeze_dim, fail_message=None): function
1139 test_shape(3, 10, [5, 7, 11], 0)
1142 test_shape(3, 10, [5, 7, 11, 13, 17], 0)
1143 test_shape(3, 10, [5, 7, 11, 13, 17], 3)
1146 test_shape(3, 10, [5, 7, 11, 13, 17], 4)
1147 test_shape(3, 10, [5, 7, 11, 13, 17], 5)
1150 test_shape(3, 10, [5, 7, 11, 13, 17], -1)
1151 test_shape(3, 10, [5, 7, 11, 13, 17], -6)
1154 test_shape(3, 10, [5, 7, 11, 13, 17], -7, "Dimension out of range")
1155 test_shape(3, 10, [5, 7, 11, 13, 17], 6, "Dimension out of range")
1160 def test_shape(sparse_dims, nnz, sizes, select_dim, select_index, fail_message=None): function
1176 test_shape(1, 10, sizes, 0, i)
1177 test_shape(1, 10, sizes, 0, sizes[0] + 1, r'select[(][)][:] index \d out of range.*')
1182 test_shape(3, 10, sizes, d, i)
1187 test_shape(1, 10, sizes, d, i)
1207 def test_shape(sparse_dims, nnz, sizes, select_dim, select_index, fail_message=None): function
1226 test_shape(1, 10, sizes, d, index)
1227 test_shape(len(sizes) // 2, 10, sizes, d, index)
1228 test_shape(len(sizes), 10, sizes, d, index)
1334 def test_shape(di, dj, dk, nnz): function
1353 test_shape(10, 100, 100, 20)
1354 test_shape(100, 1000, 200, 20)
1355 test_shape(64, 10000, 300, 20)
1356 test_shape(0, 100, 100, 0)
1357 test_shape(10, 0, 100, 0)
1358 test_shape(10, 100, 0, 0)
1359 test_shape(10, 100, 0, 20)
1368 def test_shape(num_mats, dim_i, dim_j, dim_k, nnz): function
1389 test_shape(10, 10, 100, 99, 20)
1390 test_shape(10, 100, 1000, 200, 20)
1391 test_shape(10, 64, 10000, 300, 20)
1392 test_shape(10, 0, 100, 99, 0)
1393 test_shape(10, 10, 0, 100, 0)
1394 test_shape(10, 10, 100, 0, 0)
1395 test_shape(10, 10, 100, 0, 20)
1396 test_shape(10, 10, 100, 0, 20)
1424 def test_shape(num_mats, dim_i, dim_j, dim_k, nnz): function
1447 test_shape(10, 10, 100, 99, 20)
1448 test_shape(10, 100, 1000, 200, 20)
1449 test_shape(10, 64, 10000, 300, 20)
1450 test_shape(10, 0, 100, 99, 0)
1451 test_shape(10, 10, 0, 100, 0)
1452 test_shape(10, 10, 100, 0, 0)
1453 test_shape(10, 10, 100, 0, 20)
1454 test_shape(10, 10, 100, 0, 20)
1474 def test_shape(di, dj, dk, nnz): function
1493 test_shape(7, 5, 3, 20)
1494 test_shape(1000, 100, 100, 20)
1495 test_shape(3000, 64, 300, 20)
1496 test_shape(0, 100, 100, 0)
1497 test_shape(1000, 0, 100, 0)
1498 test_shape(1000, 100, 0, 0)
1508 def test_shape(di, dj, dk, nnz): function
1523 test_shape(7, 5, 3, 20)
1524 test_shape(1000, 100, 100, 20)
1525 test_shape(3000, 64, 300, 20)
1526 test_shape(0, 100, 100, 0)
1527 test_shape(1000, 0, 100, 0)
1528 test_shape(1000, 100, 0, 0)
1562 def test_shape(m, n, p, nnz, broadcast, alpha_beta=None): function
1588 test_shape(7, 8, 9, 20, False, None)
1589 test_shape(7, 8, 9, 20, True, None)
1590 test_shape(7, 8, 9, 20, False, (1, 0))
1591 test_shape(7, 8, 9, 20, True, (1, 0))
1592 test_shape(7, 8, 9, 20, False, (1, 1))
1593 test_shape(7, 8, 9, 20, True, (1, 1))
1599 def test_shape(d1, d2, d3, nnz, transposed): function
1614 test_shape(7, 8, 9, 20, False)
1615 test_shape(7, 8, 9, 20, True)
1627 def test_shape(sparse_dims, nnz, with_shape): function
1643 def test_shape(di, dj, dk, nnz): function
1651 test_shape(7, 5, 3, 20)
1652 test_shape(1000, 100, 100, 20)
1653 test_shape(3000, 64, 300, 20)
1654 test_shape(0, 100, 100, 0)
1655 test_shape(1000, 0, 100, 0)
1656 test_shape(1000, 100, 0, 0)
1657 test_shape(1000, 100, 0, 20)
1662 def test_shape(di, dj, dk, nnz): function
1670 test_shape(7, 5, 3, 20)
1671 test_shape(1000, 100, 100, 20)
1672 test_shape(3000, 64, 300, 20)
1673 test_shape(0, 100, 100, 0)
1674 test_shape(1000, 0, 100, 0)
1675 test_shape(1000, 100, 0, 0)
1676 test_shape(1000, 100, 0, 20)
1768 def test_shape(sparse_dims, nnz, with_size): function
1773 test_shape(3, 10, 100)
1774 test_shape(4, 10, [100, 100, 100, 5, 5, 5, 0])
1775 test_shape(4, 0, [0, 0, 100, 5, 5, 5, 0])
1967 def test_shape(dense_size, sparse_dims_shape, dense_dims_shape, sparse_size): function
1977 test_shape([3, 4], [1, 4], [4, 4, 4], [3, 4, 4])
1978 test_shape([3, 4, 0], [1, 4], [4, 4, 4, 0], [3, 4, 4, 0])
2181 def test_shape(i_shapes, v_shapes, shape, nnzs): function
2185 test_shape([2, 3, 4], [3, 4, 5, 6], [2, 3, 4], [9, 12])
2186 test_shape([0, 3, 4], [3, 4, 5, 6], [2, 3, 4], [0])
2187 test_shape([2, 3, 4], [0, 4, 5, 6], [2, 3, 4], [9, 12])
2188 test_shape([2, 3, 4], [3, 4, 5, 6], [2, 3, 0], [9, 12])
2189 test_shape([0, 3, 4], [3, 4, 5, 6], [2, 3, 0], [0])
2190 test_shape([2, 3, 4], [0, 4, 5, 6], [2, 3, 0], [9, 12])
2207 def test_shape(i_shapes, v_shapes, nnzs): function
2211 test_shape([2, 3, 4], [3, 4, 5, 6], [9, 12])
2212 test_shape([0, 3, 4], [3, 4, 5, 6], [0])
2213 test_shape([2, 3, 4], [0, 4, 5, 6], [9, 12])
2214 test_shape([2, 3, 4], [3, 4, 5, 6], [9, 12])
2215 test_shape([0, 3, 4], [3, 4, 5, 6], [0])
2216 test_shape([2, 3, 4], [0, 4, 5, 6], [9, 12])
2567 def test_shape(di, dj, dk, nnz): function
2575 test_shape(10, 100, 100, 20)
2576 test_shape(100, 1000, 1000, 20)
2577 test_shape(64, 10000, 10000, 20)
2578 test_shape(0, 100, 100, 0)
2579 test_shape(10, 0, 0, 0)
2580 test_shape(10, 100, 100, 0)
2581 test_shape(10, 100, 100, 20)
2584 test_shape(10, 100, 10, 20)
2670 def test_shape(sparse_dims, nnz, with_size): function
2680 test_shape(3, 10, 100)
2681 test_shape(3, 0, [100, 100, 0])
2794 def test_shape(i_shape, v_shape, size, expected_size): function
2807 test_shape([1, 0], [0, 2, 4, 0], None, [0, 2, 4, 0])
2808 test_shape([3, 0], [0, 2, 4, 0], None, [0, 0, 0, 2, 4, 0])
2809 test_shape([1, 0], [0, 2, 4, 0], [0, 2, 4, 0], [0, 2, 4, 0])
2810 test_shape([3, 0], [0, 2, 4, 0], [0, 0, 0, 2, 4, 0], [0, 0, 0, 2, 4, 0])
2811 test_shape([3, 0], [0, 2, 4, 0], [1, 2, 3, 2, 4, 0], [1, 2, 3, 2, 4, 0])