Operator.h 425 B

123456789101112131415161718
  1. #pragma once
  2. // ${generated_comment}
  3. #include <tuple>
  4. #include <vector>
  5. // Forward declarations of any types needed in the operator signatures.
  6. // We can't directly include these classes because it will cause circular include dependencies.
  7. // This file is included by TensorBody.h, which defines the Tensor class.
  8. #include <ATen/core/ATen_fwd.h>
  9. namespace at {
  10. namespace _ops {
  11. ${declarations}
  12. }} // namespace at::_ops