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