Searched refs:cross_kernel_impl (Results 1 – 1 of 1) sorted by relevance
71 static void cross_kernel_impl(const Tensor& result, const Tensor& a, const Tensor& b, const int64_t… in cross_kernel_impl() function79 REGISTER_DISPATCH(cross_stub, &cross_kernel_impl);