Searched refs:cpu_gather_expanded_index_kernel (Results 1 – 1 of 1) sorted by relevance
814 void cpu_gather_expanded_index_kernel(const Tensor& result, const Tensor& index, const Tensor& self… in cpu_gather_expanded_index_kernel() function872 cpu_gather_expanded_index_kernel<scalar_t>(result, index, self); in gather_expanded_index_kernel()