Searched refs:_to_sparse (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| D | sparse.py | 507 def _to_sparse(tensor, **kwargs): function
548 lhs_sparse = _to_sparse(lhs, **to_sparse_kwargs)
549 rhs_sparse = _to_sparse(rhs, **to_sparse_kwargs)
|
| /external/pytorch/torch/masked/maskedtensor/ |
| D | _ops_refs.py | 426 @register_dispatch_func([torch.ops.aten._to_sparse])
427 def _to_sparse(func, *args, **kwargs): function
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_has_decomposition.expect | 591 aten::_to_sparse
592 aten::_to_sparse.out
593 aten::_to_sparse.sparse_dim
594 aten::_to_sparse.sparse_dim_out
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorConversions.cpp | 1967 return self._to_sparse(sparse_dim); in to_sparse()
1976 return self._to_sparse(layout, blocksize, dense_dim_opt); in to_sparse()
|
| /external/pytorch/test/ |
| D | test_meta.py | 835 aten._to_sparse.default : {c64, f16, i8, f64, c128, i64, bf16, f32, i32, b8, i16, u8},
836 aten._to_sparse.sparse_dim : {c64, f16, i8, f64, c128, i64, bf16, f32, i32, b8, i16, u8},
|
| D | test_sparse_csr.py | 555 def _to_sparse(x): function
565 sparse_sample = sample.transform(_to_sparse)
|
| /external/pytorch/torch/ |
| D | overrides.py | 349 Tensor._to_sparse,
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 2365 make_fallback(aten._to_sparse)
|