xref: /aosp_15_r20/external/pytorch/aten/src/ATen/native/Fill.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 // Functions that fill Tensors with constants. Implementations are in Fill.cpp.
2 
3 #pragma once
4 
5 #include <ATen/native/DispatchStub.h>
6 
7 namespace c10 {
8 class Scalar;
9 }
10 
11 namespace at {
12 class Tensor;
13 struct TensorIterator;
14 
15 namespace native {
16 
17 DECLARE_DISPATCH(void(*)(TensorIterator&, const c10::Scalar&), fill_stub);
18 
19 Tensor& fill_out(Tensor& self, const Scalar& value);
20 
21 }} // namespace at::native
22