Searched refs:_sparse_coo_tensor_unsafe (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | NonSymbolicBC.h | 12 TORCH_API at::Tensor _sparse_coo_tensor_unsafe(const at::Tensor & indices, const at::Tensor & value…
|
| D | TensorConversions.cpp | 285 return at::_sparse_coo_tensor_unsafe( in _to_copy()
1907 …return at::native::_sparse_coo_tensor_unsafe(indices, values, self.sizes())._coalesced_(coalesced); in sparse_compressed_to_sparse()
|
| /external/pytorch/torch/csrc/autograd/functions/ |
| D | accumulate_grad.h | 161 update_grad(at::_sparse_coo_tensor_unsafe( in accumulateGrad()
|
| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SparseCUDATensor.cu | 163 …SparseTensor dst = ::at::native::_sparse_coo_tensor_unsafe(newIndices, newValues, self.sizes())._c… in _coalesce_sparse_cuda()
|
| /external/pytorch/torch/csrc/utils/ |
| D | tensor_flatten.cpp | 119 at::_sparse_coo_tensor_unsafe(indices[i], values[i], ref_t.sizes()); in unflatten_sparse_tensors()
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseTensor.cpp | 466 return at::native::_sparse_coo_tensor_unsafe( in sparse_coo_tensor()
477 Tensor _sparse_coo_tensor_unsafe(const Tensor& indices, const Tensor& values_, at::IntArrayRef size, in _sparse_coo_tensor_unsafe() function
|
| D | SparseCsrTensor.cpp | 1230 return at::_sparse_coo_tensor_unsafe(indices, values, new_sizes)._coalesced_(true); in select_sparse_csr_worker()
|
| /external/pytorch/torch/csrc/jit/serialization/ |
| D | unpickler.cpp | 907 result = at::_sparse_coo_tensor_unsafe( in rebuildSparseTensor()
|