Searched refs:gather_cpu_kernel (Results 1 – 1 of 1) sorted by relevance
876 void gather_cpu_kernel(const Tensor& result, const Tensor& self, int64_t dim, const Tensor& index) { in gather_cpu_kernel() function958 REGISTER_DISPATCH(gather_stub, &gather_cpu_kernel);