| 123456789101112131415161718192021 |
- #define TORCH_ASSERT_NO_OPERATORS
- #include <ATen/native/ufunc/${name}.h>
- #include <ATen/Dispatch.h>
- #include <ATen/native/DispatchStub.h>
- #include <c10/core/Scalar.h>
- ${cuda_headers}
- namespace at {
- // NB: this is explicitly copied here (via codegen) rather than
- // included via NativeFunctions.h to avoid recompiling this file when
- // NativeFunctions.h changes
- namespace meta {
- ${meta_declaration}
- }
- namespace native {
- ${native_declaration}
- ${native_definitions}
- }} // namespace at::native
|
