Searched refs:gather_expanded_index_kernel (Results 1 – 1 of 1) sorted by relevance
869 void gather_expanded_index_kernel(const Tensor& result, const Tensor& self, const Tensor& index) { in gather_expanded_index_kernel() function969 REGISTER_DISPATCH(gather_expanded_index_stub, &gather_expanded_index_kernel);