NativeFunctions.h 1.1 KB

123456789101112131415161718192021222324252627282930313233
  1. #pragma once
  2. // ${generated_comment}
  3. #ifdef TORCH_ASSERT_NO_OPERATORS
  4. #error This change adds a dependency on native_functions.yaml, \
  5. meaning the file will need to be re-compiled every time an operator \
  6. is changed or added. Consider if your change would be better placed in \
  7. another file, or if a more specific header might achieve the same goal. \
  8. See NOTE: [Tensor vs. TensorBase]
  9. #endif
  10. #if defined(AT_PER_OPERATOR_HEADERS) && defined(TORCH_ASSERT_ONLY_METHOD_OPERATORS)
  11. #error This change adds a dependency on all pytorch operators, meaning the \
  12. file will need to be re-compiled every time an operator is changed or added. \
  13. Consider including a specific operator from <ATen/ops/{my_operator}_native.h> \
  14. and see NOTE [TORCH_ASSERT_ONLY_METHOD_OPERATORS].
  15. #endif
  16. #include <c10/core/Scalar.h>
  17. #include <c10/core/Storage.h>
  18. #include <c10/core/TensorOptions.h>
  19. #include <c10/util/Deprecated.h>
  20. #include <c10/util/Optional.h>
  21. #include <c10/core/QScheme.h>
  22. #include <ATen/core/Reduction.h>
  23. #include <ATen/core/Tensor.h>
  24. #include <tuple>
  25. #include <vector>
  26. ${NativeFunctions_includes}
  27. ${NativeFunctions_declarations}