Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsegment_reduction_ops_gpu.cu.h142 __global__ void SegmentMeanNormalizeKernel( in SegmentMeanNormalizeKernel() function
162 ninner, nsegments, d, SegmentMeanNormalizeKernel<SegmentId, Index, T>, in LaunchSegmentMeanNormalizeKernel()
164 return GpuLaunchKernel(SegmentMeanNormalizeKernel<SegmentId, Index, T>, in LaunchSegmentMeanNormalizeKernel()