Home
last modified time | relevance | path

Searched defs:col_indices_ptr (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/mkl/
DSparseDescriptors.h97 auto col_indices_ptr = col_indices_->data_ptr<MKL_INT>(); in MklSparseCsrDescriptor() local
/external/pytorch/torch/sparse/
D_triton_ops.py1271 col_indices_ptr, argument
1395 col_indices_ptr, argument
2261 col_indices_ptr, argument
/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSparseCsrTensorMath.cu502 index_t* col_indices_ptr = col_indices.data_ptr<index_t>(); in reduce_sparse_csr_dim0_cuda_template() local