Searched defs:values_accessor (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SoftMax.cu | 159 PackedTensorAccessor<scalar_t, 2> values_accessor, in cuda_sparse_coo_softmax_backward_kernel()
346 auto values_accessor = in compute_pool_max() local
422 auto values_accessor = values_2.packed_accessor64<scalar_t, 2>(); in cuda_sparse_coo_softmax() local
536 auto values_accessor = values_2.packed_accessor64<scalar_t, 2>(); in cuda_sparse_coo_softmax_backward() local
|
| H A D | SparseCsrTensorMath.cu | 232 scalar_t* values_accessor = valuesBuffer.data_ptr<scalar_t>(); in add_out_dense_sparse_compressed_cuda() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | SortingKernel.cpp | 209 auto values_accessor = StridedRandomAccessor<scalar_t>( in sort_kernel() local
215 auto values_accessor = StridedRandomAccessor<scalar_t>( in sort_kernel() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SoftMax.cpp | 325 auto values_accessor = values_2.accessor<scalar_t, 2>(); in cpu_sparse_coo_softmax() local
463 auto values_accessor = values_2.accessor<scalar_t, 2>(); in cpu_sparse_coo_softmax_backward() local
|
| H A D | SparseCsrTensorMath.cpp | 545 auto values_accessor = values.accessor<scalar_t, 1>(); in addmm_out_sparse_csr_native_cpu() local
905 auto values_accessor = valuesBuffer.accessor<scalar_t, 2>(); in add_out_dense_sparse_compressed_cpu() local
|
| H A D | SparseTensorMath.cpp | 598 auto values_accessor = values.accessor<scalar_t, 1>(); in add_dense_sparse_worker_non_hybrid_cpu() local
1204 auto values_accessor = values.accessor<scalar_t, 1>(); in s_addmm_out_sparse_dense_worker() local
1563 auto values_accessor = values.accessor<scalar_t, 1>(); in _sspaddmm_out_cpu() local
|