| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 289 def test_shape(sparse_dims, nnz, with_size): function
811 def test_shape(sparse_dims, nnz, with_size): function
930 def test_shape(sparse_dims, nnz, with_size): function
964 def test_shape(sparse_dims, nnz, with_size): function
1005 def test_shape(di, dj, dk, nnz): function
1058 def test_shape(sparse_dims, nnz, sizes): function
1128 def test_shape(sparse_dims, nnz, sizes, unsqueeze_dim, fail_message=None): function
1160 def test_shape(sparse_dims, nnz, sizes, select_dim, select_index, fail_message=None): function
1207 def test_shape(sparse_dims, nnz, sizes, select_dim, select_index, fail_message=None): function
1334 def test_shape(di, dj, dk, nnz): function
[all …]
|
| H A D | test_sparse_csr.py | 1839 def test_shape(di, dj, dk, nnz0=None, nnz1=None): function
1941 def test_shape(d1, d2, d3, nnz, transposed, index_dtype): function
1960 def test_shape(m, n, p, nnz, broadcast, index_dtype, alpha_beta=None): function
|
| H A D | test_nn.py | 5534 def test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners): function
5895 def test_shape(N, C, ID, IH, IW, D, H, W, mode, padding_mode, align_corners): function
|
| H A D | test_mps.py | 10607 def test_shape(N, C, IH, IW, H, W, mode, padding_mode, align_corners): function
|
| H A D | test_autograd.py | 3917 def test_shape(self): member in TestAutograd
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_utils_test.cc | 352 for (const TensorShape& test_shape : kTensorShapes) { in TEST_P() local
455 for (const TensorShape& test_shape : kTensorShapes) { in TEST_P() local
495 for (const TensorShape& test_shape : kTensorShapes) { in TEST_P() local
|
| /aosp_15_r20/external/harfbuzz_ng/test/api/ |
| H A D | test-shape.c | 112 test_shape (void) in test_shape() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/structured/ |
| H A D | structured_array_ops_test.py | 274 def test_shape(self): member in StructuredArrayOpsTest
|
| /aosp_15_r20/external/pytorch/test/torch_np/ |
| H A D | test_basic.py | 263 def test_shape(self, func): member in TestShapeLikeToArray
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | array_test.py | 91 def test_shape(self): member in ArrayTest
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | sub_test.cc | 424 const std::vector<int> test_shape = {18}; in TEST() local
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | test_serialize.py | 750 def test_shape(self): member in TestDeserialize
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_eager_transforms.py | 305 def test_shape(self, device): member in TestGradTransform
|