Searched refs:_sparse_coo_tensor_with_dims_and_tensors (Results 1 – 14 of 14) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | SparseTensorUtils.cpp | 118 return at::_sparse_coo_tensor_with_dims_and_tensors( in zeros_like_with_indices()
|
| D | TensorShape.cpp | 1502 return _sparse_coo_tensor_with_dims_and_tensors( in permute_sparse_coo()
1766 return _sparse_coo_tensor_with_dims_and_tensors( in select_sparse()
1771 return _sparse_coo_tensor_with_dims_and_tensors( in select_sparse()
1944 return _sparse_coo_tensor_with_dims_and_tensors( in index_select_sparse_cpu()
2417 return _sparse_coo_tensor_with_dims_and_tensors( in index_select_sparse_cpu()
2494 return _sparse_coo_tensor_with_dims_and_tensors( in index_select_sparse_cpu()
3403 return _sparse_coo_tensor_with_dims_and_tensors( in unsqueeze_sparse()
3406 return _sparse_coo_tensor_with_dims_and_tensors( in unsqueeze_sparse()
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseUnaryOps.cpp | 86 Tensor result = at::_sparse_coo_tensor_with_dims_and_tensors( in coalesced_unary_ufunc()
|
| D | SparseTensor.cpp | 359 return at::_sparse_coo_tensor_with_dims_and_tensors( in sparse_coo_tensor()
886 return at::_sparse_coo_tensor_with_dims_and_tensors( in _pin_memory_sparse_coo()
|
| D | SparseTensorMath.cpp | 1708 …SparseTensor new_sparse = at::_sparse_coo_tensor_with_dims_and_tensors(new_sparse_dim, new_dense_d… in _sparse_sum()
1811 …return at::_sparse_coo_tensor_with_dims_and_tensors(input_sparse_dim, input_dense_dim, input_sizes… in _sparse_sum_backward_cpu()
1888 …return at::_sparse_coo_tensor_with_dims_and_tensors(input_sparse_dim, input_dense_dim, input_sizes… in _sparse_sum_backward_cpu()
|
| /external/pytorch/torch/masked/maskedtensor/ |
| D | _ops_refs.py | 494 @register_dispatch_func([torch.ops.aten._sparse_coo_tensor_with_dims_and_tensors])
495 def _sparse_coo_tensor_with_dims_and_tensors(func, *args, **kwargs): function
|
| /external/pytorch/torch/_subclasses/ |
| D | fake_impls.py | 255 @register_op_impl(aten._sparse_coo_tensor_with_dims_and_tensors.default)
256 def _sparse_coo_tensor_with_dims_and_tensors(fake_mode, func, *args, **kwargs): function
|
| D | fake_tensor.py | 2181 aten._sparse_coo_tensor_with_dims_and_tensors.default,
|
| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SparseCUDATensorMath.cu | 590 …return at::_sparse_coo_tensor_with_dims_and_tensors(input_sparse_dim, input_dense_dim, input_sizes… in _sparse_sum_backward_cuda()
666 …return at::_sparse_coo_tensor_with_dims_and_tensors(input_sparse_dim, input_dense_dim, input_sizes… in _sparse_sum_backward_cuda()
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_has_decomposition.expect | 526 aten::_sparse_coo_tensor_with_dims_and_tensors
527 aten::_sparse_coo_tensor_with_dims_and_tensors.out
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Indexing.cu | 1655 return at::_sparse_coo_tensor_with_dims_and_tensors( in index_select_sparse_cuda()
1799 return _sparse_coo_tensor_with_dims_and_tensors( in index_select_sparse_cuda()
|
| /external/pytorch/test/inductor/ |
| D | test_cudagraph_trees.py | 840 return aten._sparse_coo_tensor_with_dims_and_tensors(
|
| /external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/ |
| D | fambench_dlrm_training.txt | 251 Operator: aten._sparse_coo_tensor_with_dims_and_tensors.default
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 2363 make_fallback(aten._sparse_coo_tensor_with_dims_and_tensors)
|