/usr/local/lib64/python3.6/site-packages/torch/include/ATen/native
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/Cross.h (262B)
#pragma once
#include
#include
namespace at { namespace native {
using cross_fn = void(*)(Tensor&, const Tensor&, const Tensor&, const int64_t d);
DECLARE_DISPATCH(cross_fn, cross_stub);
}} // namespace at::native