Cross.h 260 B

1234567891011121314
  1. #pragma once
  2. #include <ATen/native/DispatchStub.h>
  3. namespace at {
  4. class Tensor;
  5. namespace native {
  6. using cross_fn = void(*)(const Tensor&, const Tensor&, const Tensor&, const int64_t d);
  7. DECLARE_DISPATCH(cross_fn, cross_stub);
  8. }} // namespace at::native