StackKernel.h 322 B

123456789101112
  1. // Copyright 2004-present Facebook. All Rights Reserved.
  2. #pragma once
  3. #include <ATen/core/Tensor.h>
  4. #include <ATen/native/DispatchStub.h>
  5. namespace at { namespace native {
  6. using stack_serial_fn = void(*)(Tensor &, TensorList, int64_t);
  7. DECLARE_DISPATCH(stack_serial_fn, stack_serial_stub);
  8. }} // namespace at::native