Searched refs:_sparse_coo_tensor_unsafe_symint (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | Embedding.cpp | 102 return at::_sparse_coo_tensor_unsafe_symint(at::empty({1, 0}, indices_.options().dtype(kLong)), in embedding_sparse_backward()
109 return at::_sparse_coo_tensor_unsafe_symint(index.to(kLong), values, weight_size); in embedding_sparse_backward()
|
| D | TensorAdvancedIndexing.cpp | 2228 …if (self.ndimension() == 0) return at::_sparse_coo_tensor_unsafe_symint(at::empty_symint({0,grad.s… in _gather_sparse_backward()
2229 …if (grad.ndimension() == 0) return at::_sparse_coo_tensor_unsafe_symint(index.view({1,1}), grad, s… in _gather_sparse_backward()
2246 return at::_sparse_coo_tensor_unsafe_symint(sparse_ind, grad.reshape(-1), self.sym_sizes()); in _gather_sparse_backward()
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseTensor.cpp | 486 …return at::native::_sparse_coo_tensor_unsafe_symint(indices, values_, c10::fromIntArrayRefSlow(siz… in _sparse_coo_tensor_unsafe()
495 Tensor _sparse_coo_tensor_unsafe_symint(const Tensor& indices, const Tensor& values_, c10::SymIntAr… in _sparse_coo_tensor_unsafe_symint() function
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 7163 return at::_sparse_coo_tensor_unsafe_symint( in values_backward()
|