LinearAlgebra.h 343 B

123456789101112131415161718
  1. #pragma once
  2. #include <ATen/native/DispatchStub.h>
  3. #include <c10/util/Optional.h>
  4. namespace c10 {
  5. class Scalar;
  6. }
  7. namespace at {
  8. struct TensorIterator;
  9. }
  10. namespace at { namespace native {
  11. using addr_fn = void (*)(TensorIterator &, const Scalar& beta, const Scalar& alpha);
  12. DECLARE_DISPATCH(addr_fn, addr_stub);
  13. }} // namespace at::native