Fill.h 397 B

123456789101112131415161718192021
  1. // Functions that fill Tensors with constants. Implementations are in Fill.cpp.
  2. #pragma once
  3. #include <ATen/native/DispatchStub.h>
  4. namespace c10 {
  5. class Scalar;
  6. }
  7. namespace at {
  8. class Tensor;
  9. struct TensorIterator;
  10. namespace native {
  11. DECLARE_DISPATCH(void(*)(TensorIterator&, const c10::Scalar&), fill_stub);
  12. Tensor& fill_out(Tensor& self, const Scalar& value);
  13. }} // namespace at::native