UfuncCPU.cpp 445 B

12345678910111213141516171819
  1. #define TORCH_ASSERT_NO_OPERATORS
  2. #include <ATen/native/DispatchStub.h>
  3. #include <ATen/TensorIterator.h>
  4. #include <ATen/TensorMeta.h>
  5. namespace at {
  6. // NB: this is explicitly copied here (via codegen) rather than
  7. // included via NativeFunctions.h to avoid recompiling this file when
  8. // NativeFunctions.h changes
  9. namespace meta {
  10. ${meta_declaration}
  11. }
  12. namespace native {
  13. ${native_declaration}
  14. ${native_definitions}
  15. }} // namespace at::native