Home
last modified time | relevance | path

Searched refs:sparse_compressed_tensor (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensor.cpp579 Tensor sparse_compressed_tensor( in sparse_compressed_tensor() function
609 Tensor sparse_compressed_tensor( in sparse_compressed_tensor() function
652 …return at::native::sparse_compressed_tensor(compressed_indices, plain_indices, values, dtype, layo…
666 …return at::native::sparse_compressed_tensor(compressed_indices, plain_indices, values, size, dtype…
/aosp_15_r20/external/pytorch/torch/utils/
H A D_device.py43 torch.sparse_compressed_tensor,
/aosp_15_r20/external/pytorch/torch/sparse/
H A D__init__.py658 a = torch.sparse_compressed_tensor(
H A D_triton_ops.py227 return torch.sparse_compressed_tensor(
1861 return torch.sparse_compressed_tensor(
/aosp_15_r20/external/pytorch/docs/source/
H A Dsparse.rst1084 :func:`torch.sparse_compressed_tensor` function that have the same
1091 layout parameter to the :func:`torch.sparse_compressed_tensor`
1097 …>>> csr = torch.sparse_compressed_tensor(compressed_indices, plain_indices, values, layout=torch.s…
1103 …>>> csc = torch.sparse_compressed_tensor(compressed_indices, plain_indices, values, layout=torch.s…
1287 sparse_compressed_tensor
/aosp_15_r20/external/pytorch/torch/multiprocessing/
H A Dreductions.py473 return torch.sparse_compressed_tensor(
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse_csr.py283 sparse = torch.sparse_compressed_tensor(
287 sparse = torch.sparse_compressed_tensor(
441 x = torch.sparse_compressed_tensor(compressed_indices,
906 … torch.sparse_compressed_tensor(compressed_indices, plain_indices, values, size, layout=layout)
908 … torch.sparse_compressed_tensor(compressed_indices, plain_indices, values, layout=layout)
957 …sparse = torch.sparse_compressed_tensor(compressed_indices, plain_indices, values, size, layout=la…
3814 input_broadcasted_clone = torch.sparse_compressed_tensor(
4077 … return torch.sparse_compressed_tensor(compressed_indices, plain_indices, nc_copy(t.values()),
H A Dtest_sparse.py4283 r = torch.sparse_compressed_tensor(
4343 x = torch.sparse_compressed_tensor(
4536 …return torch.sparse_compressed_tensor(compressed_indices, invalid_plain_indices, values, shape, la…
4538 …return torch.sparse_compressed_tensor(compressed_indices, invalid_plain_indices, values, shape, la…
4733 return torch.sparse_compressed_tensor(*args, **kwargs)
4972 t = torch.sparse_compressed_tensor(compressed_indices, plain_indices, range(20),
5251 ones = torch.sparse_compressed_tensor(mask.crow_indices(), mask.col_indices(),
5257 ones = torch.sparse_compressed_tensor(mask.ccol_indices(), mask.row_indices(),
5462 return torch.sparse_compressed_tensor(*args, **kwargs)
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_new.cpp997 return at::sparse_compressed_tensor( in sparse_compressed_tensor_ctor_worker()
1053 return at::sparse_compressed_tensor( in sparse_compressed_tensor_ctor_worker()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_torch_functions_manual.cpp193 sparse_compressed_tensor,
/aosp_15_r20/external/pytorch/torch/
H A D_utils.py315 result = torch.sparse_compressed_tensor(
H A D_torch_docs.py9900 torch.sparse_compressed_tensor,
H A Doverrides.py227 torch.sparse_compressed_tensor,
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect1179 aten::sparse_compressed_tensor.comp_plain_value
1180 aten::sparse_compressed_tensor.comp_plain_value_size
/aosp_15_r20/external/pytorch/torch/autograd/
H A Dgradcheck.py1689 vec = torch.sparse_compressed_tensor(
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py3341 return torch.sparse_compressed_tensor(compressed_indices, plain_indices,
3456 yield torch.sparse_compressed_tensor(*args, **kwargs)
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dutils.py827 return torch.sparse_compressed_tensor(