123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377 |
- #pragma once
- // @generated by torchgen/gen.py from Function.h
- #include <ATen/Context.h>
- #include <ATen/DeviceGuard.h>
- #include <ATen/TensorUtils.h>
- #include <ATen/TracerMode.h>
- #include <ATen/core/Generator.h>
- #include <ATen/core/Reduction.h>
- #include <ATen/core/Tensor.h>
- #include <c10/core/Scalar.h>
- #include <c10/core/Storage.h>
- #include <c10/core/TensorOptions.h>
- #include <c10/util/Deprecated.h>
- #include <c10/util/Optional.h>
- #include <ATen/ops/randint_ops.h>
- namespace at {
- // aten::randint(int high, SymInt[] size, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint(int64_t high, at::IntArrayRef size, at::TensorOptions options=at::kLong) {
- return at::_ops::randint::call(high, c10::fromIntArrayRefSlow(size), optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor randint(int64_t high, at::IntArrayRef size, at::TensorOptions options=at::kLong) {
- return at::_ops::randint::call(high, c10::fromIntArrayRefSlow(size), optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- }
- // aten::randint(int high, SymInt[] size, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint(int64_t high, at::IntArrayRef size, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint::call(high, c10::fromIntArrayRefSlow(size), dtype, layout, device, pin_memory);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor randint(int64_t high, at::IntArrayRef size, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint::call(high, c10::fromIntArrayRefSlow(size), dtype, layout, device, pin_memory);
- }
- }
- // aten::randint(int high, SymInt[] size, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint_symint(int64_t high, c10::SymIntArrayRef size, at::TensorOptions options=at::kLong) {
- return at::_ops::randint::call(high, size, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor randint(int64_t high, c10::SymIntArrayRef size, at::TensorOptions options=at::kLong) {
- return at::_ops::randint::call(high, size, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- }
- // aten::randint(int high, SymInt[] size, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint_symint(int64_t high, c10::SymIntArrayRef size, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint::call(high, size, dtype, layout, device, pin_memory);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor randint(int64_t high, c10::SymIntArrayRef size, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint::call(high, size, dtype, layout, device, pin_memory);
- }
- }
- // aten::randint.generator(int high, SymInt[] size, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint(int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_generator::call(high, c10::fromIntArrayRefSlow(size), generator, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor randint(int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_generator::call(high, c10::fromIntArrayRefSlow(size), generator, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- }
- // aten::randint.generator(int high, SymInt[] size, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint(int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_generator::call(high, c10::fromIntArrayRefSlow(size), generator, dtype, layout, device, pin_memory);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor randint(int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_generator::call(high, c10::fromIntArrayRefSlow(size), generator, dtype, layout, device, pin_memory);
- }
- }
- // aten::randint.generator(int high, SymInt[] size, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint_symint(int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_generator::call(high, size, generator, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor randint(int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_generator::call(high, size, generator, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- }
- // aten::randint.generator(int high, SymInt[] size, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint_symint(int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_generator::call(high, size, generator, dtype, layout, device, pin_memory);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor randint(int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_generator::call(high, size, generator, dtype, layout, device, pin_memory);
- }
- }
- // aten::randint.low(int low, int high, SymInt[] size, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint(int64_t low, int64_t high, at::IntArrayRef size, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_low::call(low, high, c10::fromIntArrayRefSlow(size), optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor randint(int64_t low, int64_t high, at::IntArrayRef size, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_low::call(low, high, c10::fromIntArrayRefSlow(size), optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- }
- // aten::randint.low(int low, int high, SymInt[] size, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint(int64_t low, int64_t high, at::IntArrayRef size, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_low::call(low, high, c10::fromIntArrayRefSlow(size), dtype, layout, device, pin_memory);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor randint(int64_t low, int64_t high, at::IntArrayRef size, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_low::call(low, high, c10::fromIntArrayRefSlow(size), dtype, layout, device, pin_memory);
- }
- }
- // aten::randint.low(int low, int high, SymInt[] size, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint_symint(int64_t low, int64_t high, c10::SymIntArrayRef size, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_low::call(low, high, size, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor randint(int64_t low, int64_t high, c10::SymIntArrayRef size, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_low::call(low, high, size, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- }
- // aten::randint.low(int low, int high, SymInt[] size, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint_symint(int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_low::call(low, high, size, dtype, layout, device, pin_memory);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor randint(int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_low::call(low, high, size, dtype, layout, device, pin_memory);
- }
- }
- // aten::randint.low_generator(int low, int high, SymInt[] size, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint(int64_t low, int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_low_generator::call(low, high, c10::fromIntArrayRefSlow(size), generator, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor randint(int64_t low, int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_low_generator::call(low, high, c10::fromIntArrayRefSlow(size), generator, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- }
- // aten::randint.low_generator(int low, int high, SymInt[] size, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint(int64_t low, int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_low_generator::call(low, high, c10::fromIntArrayRefSlow(size), generator, dtype, layout, device, pin_memory);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor randint(int64_t low, int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_low_generator::call(low, high, c10::fromIntArrayRefSlow(size), generator, dtype, layout, device, pin_memory);
- }
- }
- // aten::randint.low_generator(int low, int high, SymInt[] size, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint_symint(int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_low_generator::call(low, high, size, generator, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor randint(int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, at::TensorOptions options=at::kLong) {
- return at::_ops::randint_low_generator::call(low, high, size, generator, optTypeMetaToScalarType(options.dtype_opt()), options.layout_opt(), options.device_opt(), options.pinned_memory_opt());
- }
- }
- // aten::randint.low_generator(int low, int high, SymInt[] size, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
- inline at::Tensor randint_symint(int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_low_generator::call(low, high, size, generator, dtype, layout, device, pin_memory);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor randint(int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
- return at::_ops::randint_low_generator::call(low, high, size, generator, dtype, layout, device, pin_memory);
- }
- }
- // aten::randint.out(int high, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_out(at::Tensor & out, int64_t high, at::IntArrayRef size) {
- return at::_ops::randint_out::call(high, c10::fromIntArrayRefSlow(size), out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor & randint_out(at::Tensor & out, int64_t high, at::IntArrayRef size) {
- return at::_ops::randint_out::call(high, c10::fromIntArrayRefSlow(size), out);
- }
- }
- // aten::randint.out(int high, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_outf(int64_t high, at::IntArrayRef size, at::Tensor & out) {
- return at::_ops::randint_out::call(high, c10::fromIntArrayRefSlow(size), out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor & randint_outf(int64_t high, at::IntArrayRef size, at::Tensor & out) {
- return at::_ops::randint_out::call(high, c10::fromIntArrayRefSlow(size), out);
- }
- }
- // aten::randint.out(int high, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_symint_out(at::Tensor & out, int64_t high, c10::SymIntArrayRef size) {
- return at::_ops::randint_out::call(high, size, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor & randint_out(at::Tensor & out, int64_t high, c10::SymIntArrayRef size) {
- return at::_ops::randint_out::call(high, size, out);
- }
- }
- // aten::randint.out(int high, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_symint_outf(int64_t high, c10::SymIntArrayRef size, at::Tensor & out) {
- return at::_ops::randint_out::call(high, size, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor & randint_outf(int64_t high, c10::SymIntArrayRef size, at::Tensor & out) {
- return at::_ops::randint_out::call(high, size, out);
- }
- }
- // aten::randint.generator_out(int high, SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_out(at::Tensor & out, int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator) {
- return at::_ops::randint_generator_out::call(high, c10::fromIntArrayRefSlow(size), generator, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor & randint_out(at::Tensor & out, int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator) {
- return at::_ops::randint_generator_out::call(high, c10::fromIntArrayRefSlow(size), generator, out);
- }
- }
- // aten::randint.generator_out(int high, SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_outf(int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, at::Tensor & out) {
- return at::_ops::randint_generator_out::call(high, c10::fromIntArrayRefSlow(size), generator, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor & randint_outf(int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, at::Tensor & out) {
- return at::_ops::randint_generator_out::call(high, c10::fromIntArrayRefSlow(size), generator, out);
- }
- }
- // aten::randint.generator_out(int high, SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_symint_out(at::Tensor & out, int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator) {
- return at::_ops::randint_generator_out::call(high, size, generator, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor & randint_out(at::Tensor & out, int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator) {
- return at::_ops::randint_generator_out::call(high, size, generator, out);
- }
- }
- // aten::randint.generator_out(int high, SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_symint_outf(int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, at::Tensor & out) {
- return at::_ops::randint_generator_out::call(high, size, generator, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor & randint_outf(int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, at::Tensor & out) {
- return at::_ops::randint_generator_out::call(high, size, generator, out);
- }
- }
- // aten::randint.low_out(int low, int high, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_out(at::Tensor & out, int64_t low, int64_t high, at::IntArrayRef size) {
- return at::_ops::randint_low_out::call(low, high, c10::fromIntArrayRefSlow(size), out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor & randint_out(at::Tensor & out, int64_t low, int64_t high, at::IntArrayRef size) {
- return at::_ops::randint_low_out::call(low, high, c10::fromIntArrayRefSlow(size), out);
- }
- }
- // aten::randint.low_out(int low, int high, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_outf(int64_t low, int64_t high, at::IntArrayRef size, at::Tensor & out) {
- return at::_ops::randint_low_out::call(low, high, c10::fromIntArrayRefSlow(size), out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor & randint_outf(int64_t low, int64_t high, at::IntArrayRef size, at::Tensor & out) {
- return at::_ops::randint_low_out::call(low, high, c10::fromIntArrayRefSlow(size), out);
- }
- }
- // aten::randint.low_out(int low, int high, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_symint_out(at::Tensor & out, int64_t low, int64_t high, c10::SymIntArrayRef size) {
- return at::_ops::randint_low_out::call(low, high, size, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor & randint_out(at::Tensor & out, int64_t low, int64_t high, c10::SymIntArrayRef size) {
- return at::_ops::randint_low_out::call(low, high, size, out);
- }
- }
- // aten::randint.low_out(int low, int high, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_symint_outf(int64_t low, int64_t high, c10::SymIntArrayRef size, at::Tensor & out) {
- return at::_ops::randint_low_out::call(low, high, size, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor & randint_outf(int64_t low, int64_t high, c10::SymIntArrayRef size, at::Tensor & out) {
- return at::_ops::randint_low_out::call(low, high, size, out);
- }
- }
- // aten::randint.low_generator_out(int low, int high, SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_out(at::Tensor & out, int64_t low, int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator) {
- return at::_ops::randint_low_generator_out::call(low, high, c10::fromIntArrayRefSlow(size), generator, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor & randint_out(at::Tensor & out, int64_t low, int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator) {
- return at::_ops::randint_low_generator_out::call(low, high, c10::fromIntArrayRefSlow(size), generator, out);
- }
- }
- // aten::randint.low_generator_out(int low, int high, SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_outf(int64_t low, int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, at::Tensor & out) {
- return at::_ops::randint_low_generator_out::call(low, high, c10::fromIntArrayRefSlow(size), generator, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, int64_t>::value>>
- at::Tensor & randint_outf(int64_t low, int64_t high, at::IntArrayRef size, c10::optional<at::Generator> generator, at::Tensor & out) {
- return at::_ops::randint_low_generator_out::call(low, high, c10::fromIntArrayRefSlow(size), generator, out);
- }
- }
- // aten::randint.low_generator_out(int low, int high, SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_symint_out(at::Tensor & out, int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator) {
- return at::_ops::randint_low_generator_out::call(low, high, size, generator, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor & randint_out(at::Tensor & out, int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator) {
- return at::_ops::randint_low_generator_out::call(low, high, size, generator, out);
- }
- }
- // aten::randint.low_generator_out(int low, int high, SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!)
- inline at::Tensor & randint_symint_outf(int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, at::Tensor & out) {
- return at::_ops::randint_low_generator_out::call(low, high, size, generator, out);
- }
- namespace symint {
- template <typename T, typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>>
- at::Tensor & randint_outf(int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, at::Tensor & out) {
- return at::_ops::randint_low_generator_out::call(low, high, size, generator, out);
- }
- }
- }
|