Home
last modified time | relevance | path

Searched refs:SparseConcatPermuteKernel (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_concat_op_gpu.cu.cc84 __global__ void SparseConcatPermuteKernel( in SparseConcatPermuteKernel() function
229 output_nnz, device, &SparseConcatPermuteKernel<T>, in operator ()()
233 GpuLaunchKernel(SparseConcatPermuteKernel<T>, config.block_count, in operator ()()