RangeFactories.h 356 B

123456789101112
  1. #include <ATen/native/DispatchStub.h>
  2. #include <c10/core/Scalar.h>
  3. namespace at {
  4. struct TensorIterator;
  5. namespace native {
  6. DECLARE_DISPATCH(void(*)(TensorIterator&, const Scalar&, const Scalar&, const Scalar&), arange_stub);
  7. DECLARE_DISPATCH(void(*)(TensorIterator&, const Scalar&, const Scalar&, int64_t), linspace_stub);
  8. }} // namespace at::native