forked from nudt_dsp/netrans
14 lines
323 B
C++
14 lines
323 B
C++
// Functions that fill Tensors with constants. Implementations are in Fill.cpp.
|
|
|
|
#pragma once
|
|
|
|
#include <ATen/ATen.h>
|
|
#include <ATen/native/DispatchStub.h>
|
|
#include <ATen/native/TensorIterator.h>
|
|
|
|
namespace at { namespace native {
|
|
|
|
DECLARE_DISPATCH(void(*)(TensorIterator&, Scalar), fill_stub);
|
|
|
|
}} // namespace at::native
|