ViewFuncs.h 498 B

12345678910111213141516171819202122232425262728
  1. #pragma once
  2. // ${generated_comment}
  3. #include <torch/library.h>
  4. #include <torch/csrc/autograd/variable.h>
  5. #include <c10/core/SymIntArrayRef.h>
  6. #ifndef AT_PER_OPERATOR_HEADERS
  7. #include <ATen/Operators.h>
  8. #else
  9. $ops_headers
  10. #endif
  11. namespace torch::autograd::generated {
  12. using at::Scalar;
  13. using at::Tensor;
  14. using at::IntArrayRef;
  15. using at::ArrayRef;
  16. using at::Type;
  17. using at::ScalarType;
  18. using std::optional;
  19. using c10::fmap;
  20. ${view_func_declarations}
  21. } // namespace torch::autograd::generated