Searched refs:_segment_reduce_lengths_cpu_kernel (Results 1 – 1 of 1) sorted by relevance
130 Tensor _segment_reduce_lengths_cpu_kernel( in _segment_reduce_lengths_cpu_kernel() function464 &_segment_reduce_lengths_cpu_kernel);465 REGISTER_AVX2_DISPATCH(_segment_reduce_lengths_stub, &_segment_reduce_lengths_cpu_kernel);466 REGISTER_AVX512_DISPATCH(_segment_reduce_lengths_stub, &_segment_reduce_lengths_cpu_kernel);467 REGISTER_VSX_DISPATCH(_segment_reduce_lengths_stub, &_segment_reduce_lengths_cpu_kernel);468 REGISTER_ZVECTOR_DISPATCH(_segment_reduce_lengths_stub, &_segment_reduce_lengths_cpu_kernel);