Home
last modified time | relevance | path

Searched defs:dense_ptr (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dcsr_sparse_matrix_to_dense_op.cc92 auto dense_ptr = dense_t.flat<T>().data(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_to_dense_op_gpu.cu.cc152 T* dense_ptr = dense->flat<T>().data(); in operator ()() local
/aosp_15_r20/external/pytorch/torch/sparse/
H A D_triton_ops.py1400 dense_ptr, argument
2247 def _bsr_strided_addmm_kernel(
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensorMath.cpp1205 scalar_t* dense_ptr = dense.data_ptr<scalar_t>(); in s_addmm_out_sparse_dense_worker() local
1564 scalar_t* dense_ptr = dense.data_ptr<scalar_t>(); in _sspaddmm_out_cpu() local
H A DSparseCsrTensorMath.cpp546 scalar_t* dense_ptr = dense.data_ptr<scalar_t>(); in addmm_out_sparse_csr_native_cpu() local