Searched refs:flatten_indices_cpu_kernel (Results 1 – 1 of 1) sorted by relevance
19 Tensor flatten_indices_cpu_kernel(const Tensor& indices, IntArrayRef size) { in flatten_indices_cpu_kernel() function25 REGISTER_ARCH_DISPATCH(flatten_indices_stub, DEFAULT, &flatten_indices_cpu_kernel);26 REGISTER_AVX512_DISPATCH(flatten_indices_stub, &flatten_indices_cpu_kernel);27 REGISTER_AVX2_DISPATCH(flatten_indices_stub, &flatten_indices_cpu_kernel);28 REGISTER_VSX_DISPATCH(flatten_indices_stub, &flatten_indices_cpu_kernel);29 REGISTER_ZVECTOR_DISPATCH(flatten_indices_stub, &flatten_indices_cpu_kernel);