Searched defs:sparse_dims (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 289 def test_shape(sparse_dims, nnz, with_size): argument
811 def test_shape(sparse_dims, nnz, with_size): argument
930 def test_shape(sparse_dims, nnz, with_size): argument
964 def test_shape(sparse_dims, nnz, with_size): argument
1058 def test_shape(sparse_dims, nnz, sizes): argument
1128 def test_shape(sparse_dims, nnz, sizes, unsqueeze_dim, fail_message=None): argument
1160 def test_shape(sparse_dims, nnz, sizes, select_dim, select_index, fail_message=None): argument
1207 def test_shape(sparse_dims, nnz, sizes, select_dim, select_index, fail_message=None): argument
1627 def test_shape(sparse_dims, nnz, with_shape): argument
1768 def test_shape(sparse_dims, nnz, with_size): argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | reverse_op.cc | 241 const Tensor& sparse_dims = context->input(1); in Compute() local
|