LinearAlgebra.h 299 B

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