UfuncCPUKernel.cpp 350 B

1234567891011121314
  1. #define TORCH_ASSERT_NO_OPERATORS
  2. #include <ATen/native/ufunc/${name}.h>
  3. #include <ATen/native/DispatchStub.h>
  4. #include <ATen/TensorIterator.h>
  5. #include <ATen/native/cpu/Loops.h>
  6. #include <ATen/cpu/vec/vec.h>
  7. #include <ATen/Dispatch.h>
  8. #include <c10/core/Scalar.h>
  9. namespace at {
  10. namespace native {
  11. ${native_definitions}
  12. }} // namespace at::native