Searched refs:_segment_reduce_offsets_cpu_kernel (Results 1 – 1 of 1) sorted by relevance
156 Tensor _segment_reduce_offsets_cpu_kernel( in _segment_reduce_offsets_cpu_kernel() function474 &_segment_reduce_offsets_cpu_kernel);475 REGISTER_AVX2_DISPATCH(_segment_reduce_offsets_stub, &_segment_reduce_offsets_cpu_kernel);476 REGISTER_AVX512_DISPATCH(_segment_reduce_offsets_stub, &_segment_reduce_offsets_cpu_kernel);477 REGISTER_VSX_DISPATCH(_segment_reduce_offsets_stub, &_segment_reduce_offsets_cpu_kernel);478 REGISTER_ZVECTOR_DISPATCH(_segment_reduce_offsets_stub, &_segment_reduce_offsets_cpu_kernel);