Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_concat_op_gpu.cu.cc64 const int64 input_nz = nz - nnz_scan[input_num]; in SparseConcatKernel() local
103 const int64 input_nz = permuted_nz - nnz_scan[input_num]; in SparseConcatPermuteKernel() local
H A Dsparse_slice_op_gpu.cu.cc98 int64_t input_nz = selected_nonzeros[output_nz]; in SparseSliceGatherKernel() local
H A Dsparse_split_op_gpu.cu.cc174 Index input_nz = sort_permutation[sorted_input_nz]; in SparseSplitScatterKernel() local