Home
last modified time | relevance | path

Searched defs:nnz_per_batch (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsparse_tensor_to_csr_sparse_matrix_op.cc199 auto nnz_per_batch = nnz_per_batch_host.tensor().vec<int32>(); in ComputeAsync() local
H A Ddense_to_csr_sparse_matrix_op.cc229 auto nnz_per_batch = nnz_per_batch_host.tensor().vec<int32>(); in ComputeAsync() local
H A Dkernels_gpu.cu.cc55 TTypes<int32>::Vec nnz_per_batch) { in operator ()()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorConversions.cpp698 auto nnz_per_batch = values.size(1); in sparse_compressed_to_dense() local