-
-
Save cpuhrsch/18662f814bf7ae9acc7644ca01dcfa09 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
diff --git a/build/aten/src/ATen/Declarations.yaml b/build/aten/src/ATen/Declarations.yaml | |
index 7ca5b4659..852e2cade 100644 | |
--- a/build/aten/src/ATen/Declarations.yaml | |
+++ b/build/aten/src/ATen/Declarations.yaml | |
@@ -18800,7 +18800,7 @@ | |
- name: _cudnn_init_dropout_state | |
matches_jit_signature: false | |
schema_string: aten::_cudnn_init_dropout_state(float dropout, bool train, int dropout_seed, | |
- TensorOptions options) -> Tensor | |
+ ScalarType dtype, Layout layout, Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -20373,8 +20373,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: arange | |
- matches_jit_signature: false | |
- schema_string: aten::arange(Scalar end, TensorOptions options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::arange(Scalar end, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -20386,6 +20387,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -20405,9 +20407,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: arange | |
- matches_jit_signature: false | |
- schema_string: aten::arange(Scalar start, Scalar end, TensorOptions options=[]) | |
- -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::arange(Scalar start, Scalar end, *, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -20424,6 +20426,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -20443,9 +20446,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: arange | |
- matches_jit_signature: false | |
- schema_string: aten::arange(Scalar start, Scalar end, Scalar step, TensorOptions | |
- options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::arange(Scalar start, Scalar end, Scalar step, *, ScalarType | |
+ dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -20467,6 +20470,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -21278,9 +21282,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: bartlett_window | |
- matches_jit_signature: false | |
- schema_string: aten::bartlett_window(int window_length, TensorOptions options=[]) | |
- -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::bartlett_window(int window_length, *, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -21292,6 +21296,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -21311,9 +21316,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: bartlett_window | |
- matches_jit_signature: false | |
- schema_string: aten::bartlett_window(int window_length, bool periodic, TensorOptions | |
- options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::bartlett_window(int window_length, bool periodic, *, ScalarType | |
+ dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -21330,6 +21335,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -21956,9 +21962,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: blackman_window | |
- matches_jit_signature: false | |
- schema_string: aten::blackman_window(int window_length, TensorOptions options=[]) | |
- -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::blackman_window(int window_length, *, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -21970,6 +21976,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -21989,9 +21996,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: blackman_window | |
- matches_jit_signature: false | |
- schema_string: aten::blackman_window(int window_length, bool periodic, TensorOptions | |
- options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::blackman_window(int window_length, bool periodic, *, ScalarType | |
+ dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -22008,6 +22015,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -26418,8 +26426,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: empty | |
- matches_jit_signature: false | |
- schema_string: aten::empty(int[] size, TensorOptions options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::empty(int[] size, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -26431,6 +26440,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -26542,7 +26552,8 @@ | |
deprecated: false | |
- name: empty_like | |
matches_jit_signature: false | |
- schema_string: aten::empty_like(Tensor self, *, TensorOptions options) -> Tensor | |
+ schema_string: aten::empty_like(Tensor self, *, ScalarType dtype, Layout layout, | |
+ Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -26573,9 +26584,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: empty_strided | |
- matches_jit_signature: false | |
- schema_string: aten::empty_strided(int[] size, int[] stride, *, TensorOptions options=[]) | |
- -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::empty_strided(int[] size, int[] stride, *, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27034,8 +27045,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: eye | |
- matches_jit_signature: false | |
- schema_string: aten::eye(int n, TensorOptions options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::eye(int n, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27047,6 +27059,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -27066,8 +27079,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: eye | |
- matches_jit_signature: false | |
- schema_string: aten::eye(int n, int m, TensorOptions options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::eye(int n, int m, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27084,6 +27098,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -27367,9 +27382,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: full | |
- matches_jit_signature: false | |
- schema_string: aten::full(int[] size, Scalar fill_value, TensorOptions options=[]) | |
- -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::full(int[] size, Scalar fill_value, *, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27386,6 +27401,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -27476,8 +27492,8 @@ | |
deprecated: false | |
- name: full_like | |
matches_jit_signature: false | |
- schema_string: aten::full_like(Tensor self, Scalar fill_value, *, TensorOptions | |
- options) -> Tensor | |
+ schema_string: aten::full_like(Tensor self, Scalar fill_value, *, ScalarType dtype, | |
+ Layout layout, Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27739,9 +27755,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: hann_window | |
- matches_jit_signature: false | |
- schema_string: aten::hann_window(int window_length, TensorOptions options=[]) -> | |
- Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::hann_window(int window_length, *, ScalarType dtype=float, Layout | |
+ layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27753,6 +27769,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -27772,9 +27789,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: hann_window | |
- matches_jit_signature: false | |
- schema_string: aten::hann_window(int window_length, bool periodic, TensorOptions | |
- options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::hann_window(int window_length, bool periodic, *, ScalarType | |
+ dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27791,6 +27808,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -27810,9 +27828,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: hamming_window | |
- matches_jit_signature: false | |
- schema_string: aten::hamming_window(int window_length, TensorOptions options=[]) | |
- -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::hamming_window(int window_length, *, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27824,6 +27842,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -27843,9 +27862,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: hamming_window | |
- matches_jit_signature: false | |
- schema_string: aten::hamming_window(int window_length, bool periodic, TensorOptions | |
- options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::hamming_window(int window_length, bool periodic, *, ScalarType | |
+ dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27862,6 +27881,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -27881,9 +27901,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: hamming_window | |
- matches_jit_signature: false | |
+ matches_jit_signature: true | |
schema_string: aten::hamming_window(int window_length, bool periodic, float alpha, | |
- TensorOptions options=[]) -> Tensor | |
+ *, ScalarType dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27905,6 +27925,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -27924,9 +27945,10 @@ | |
with_gil: false | |
deprecated: false | |
- name: hamming_window | |
- matches_jit_signature: false | |
+ matches_jit_signature: true | |
schema_string: aten::hamming_window(int window_length, bool periodic, float alpha, | |
- float beta, TensorOptions options=[]) -> Tensor | |
+ float beta, *, ScalarType dtype=float, Layout layout=strided, Device device=\"cpu\") | |
+ -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -27953,6 +27975,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -29640,9 +29663,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: linspace | |
- matches_jit_signature: false | |
- schema_string: aten::linspace(Scalar start, Scalar end, int steps=100, TensorOptions | |
- options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::linspace(Scalar start, Scalar end, int steps=100, *, ScalarType | |
+ dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -29665,6 +29688,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -30109,9 +30133,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: logspace | |
- matches_jit_signature: false | |
- schema_string: aten::logspace(Scalar start, Scalar end, int steps=100, TensorOptions | |
- options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::logspace(Scalar start, Scalar end, int steps=100, *, ScalarType | |
+ dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -30134,6 +30158,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -33602,8 +33627,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: ones | |
- matches_jit_signature: false | |
- schema_string: aten::ones(int[] size, TensorOptions options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::ones(int[] size, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -33615,6 +33641,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -33695,7 +33722,8 @@ | |
deprecated: false | |
- name: ones_like | |
matches_jit_signature: false | |
- schema_string: aten::ones_like(Tensor self, *, TensorOptions options) -> Tensor | |
+ schema_string: aten::ones_like(Tensor self, *, ScalarType dtype, Layout layout, | |
+ Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34085,8 +34113,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: scalar_tensor | |
- matches_jit_signature: false | |
- schema_string: aten::scalar_tensor(Scalar s, *, TensorOptions options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::scalar_tensor(Scalar s, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34118,8 +34147,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: rand | |
- matches_jit_signature: false | |
- schema_string: aten::rand(int[] size, *, TensorOptions options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::rand(int[] size, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34152,8 +34182,8 @@ | |
deprecated: false | |
- name: rand | |
matches_jit_signature: false | |
- schema_string: aten::rand(int[] size, *, Generator? generator, TensorOptions options=[]) | |
- -> Tensor | |
+ schema_string: aten::rand(int[] size, *, Generator? generator, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34293,7 +34323,8 @@ | |
deprecated: false | |
- name: rand_like | |
matches_jit_signature: false | |
- schema_string: aten::rand_like(Tensor self, *, TensorOptions options) -> Tensor | |
+ schema_string: aten::rand_like(Tensor self, *, ScalarType dtype, Layout layout, | |
+ Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34324,9 +34355,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: randint | |
- matches_jit_signature: false | |
- schema_string: aten::randint(int high, int[] size, *, TensorOptions options=[]) | |
- -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::randint(int high, int[] size, *, ScalarType dtype=float, Layout | |
+ layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34364,8 +34395,8 @@ | |
deprecated: false | |
- name: randint | |
matches_jit_signature: false | |
- schema_string: aten::randint(int high, int[] size, *, Generator? generator, TensorOptions | |
- options=[]) -> Tensor | |
+ schema_string: aten::randint(int high, int[] size, *, Generator? generator, ScalarType | |
+ dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34408,9 +34439,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: randint | |
- matches_jit_signature: false | |
- schema_string: aten::randint(int low, int high, int[] size, *, TensorOptions options=[]) | |
- -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::randint(int low, int high, int[] size, *, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34454,7 +34485,7 @@ | |
- name: randint | |
matches_jit_signature: false | |
schema_string: aten::randint(int low, int high, int[] size, *, Generator? generator, | |
- TensorOptions options=[]) -> Tensor | |
+ ScalarType dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34751,8 +34782,8 @@ | |
deprecated: false | |
- name: randint_like | |
matches_jit_signature: false | |
- schema_string: aten::randint_like(Tensor self, int high, *, TensorOptions options) | |
- -> Tensor | |
+ schema_string: aten::randint_like(Tensor self, int high, *, ScalarType dtype, Layout | |
+ layout, Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34789,8 +34820,8 @@ | |
deprecated: false | |
- name: randint_like | |
matches_jit_signature: false | |
- schema_string: aten::randint_like(Tensor self, int low, int high, *, TensorOptions | |
- options) -> Tensor | |
+ schema_string: aten::randint_like(Tensor self, int low, int high, *, ScalarType | |
+ dtype, Layout layout, Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34831,8 +34862,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: randn | |
- matches_jit_signature: false | |
- schema_string: aten::randn(int[] size, *, TensorOptions options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::randn(int[] size, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -34865,8 +34897,8 @@ | |
deprecated: false | |
- name: randn | |
matches_jit_signature: false | |
- schema_string: aten::randn(int[] size, *, Generator? generator, TensorOptions options=[]) | |
- -> Tensor | |
+ schema_string: aten::randn(int[] size, *, Generator? generator, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -35006,7 +35038,8 @@ | |
deprecated: false | |
- name: randn_like | |
matches_jit_signature: false | |
- schema_string: aten::randn_like(Tensor self, *, TensorOptions options) -> Tensor | |
+ schema_string: aten::randn_like(Tensor self, *, ScalarType dtype, Layout layout, | |
+ Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -35037,8 +35070,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: randperm | |
- matches_jit_signature: false | |
- schema_string: aten::randperm(int n, *, TensorOptions options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::randperm(int n, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -35071,8 +35105,8 @@ | |
deprecated: false | |
- name: randperm | |
matches_jit_signature: false | |
- schema_string: aten::randperm(int n, *, Generator? generator, TensorOptions options=[]) | |
- -> Tensor | |
+ schema_string: aten::randperm(int n, *, Generator? generator, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -35185,9 +35219,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: range | |
- matches_jit_signature: false | |
- schema_string: aten::range(Scalar start, Scalar end, Scalar step=1, TensorOptions | |
- options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::range(Scalar start, Scalar end, Scalar step=1, *, ScalarType | |
+ dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -35210,6 +35244,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -35229,9 +35264,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: range | |
- matches_jit_signature: false | |
- schema_string: aten::range(Scalar start, Scalar end, TensorOptions options=[]) -> | |
- Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::range(Scalar start, Scalar end, *, ScalarType dtype=float, | |
+ Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -35248,6 +35283,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -39842,8 +39878,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: zeros | |
- matches_jit_signature: false | |
- schema_string: aten::zeros(int[] size, TensorOptions options=[]) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::zeros(int[] size, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -39855,6 +39892,7 @@ | |
default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -39935,7 +39973,8 @@ | |
deprecated: false | |
- name: zeros_like | |
matches_jit_signature: false | |
- schema_string: aten::zeros_like(Tensor self, *, TensorOptions options) -> Tensor | |
+ schema_string: aten::zeros_like(Tensor self, *, ScalarType dtype, Layout layout, | |
+ Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -41681,8 +41720,8 @@ | |
deprecated: false | |
- name: sparse_coo_tensor | |
matches_jit_signature: false | |
- schema_string: aten::sparse_coo_tensor(int[] size, *, TensorOptions options) -> | |
- Tensor | |
+ schema_string: aten::sparse_coo_tensor(int[] size, *, ScalarType dtype, Layout layout, | |
+ Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -41714,8 +41753,8 @@ | |
deprecated: false | |
- name: sparse_coo_tensor | |
matches_jit_signature: false | |
- schema_string: aten::sparse_coo_tensor(IndexTensor indices, Tensor values, *, TensorOptions | |
- options=[]) -> Tensor | |
+ schema_string: aten::sparse_coo_tensor(IndexTensor indices, Tensor values, *, ScalarType | |
+ dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -41754,7 +41793,8 @@ | |
- name: sparse_coo_tensor | |
matches_jit_signature: false | |
schema_string: aten::sparse_coo_tensor(IndexTensor indices, Tensor values, int[] | |
- size, *, TensorOptions options=[]) -> Tensor | |
+ size, *, ScalarType dtype=float, Layout layout=strided, Device device=\"cpu\") | |
+ -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -41798,7 +41838,8 @@ | |
- name: _sparse_coo_tensor_unsafe | |
matches_jit_signature: false | |
schema_string: aten::_sparse_coo_tensor_unsafe(IndexTensor indices, Tensor values, | |
- int[] size, *, TensorOptions options=[]) -> Tensor | |
+ int[] size, *, ScalarType dtype=float, Layout layout=strided, Device device=\"cpu\") | |
+ -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -41842,7 +41883,7 @@ | |
- name: _sparse_coo_tensor_with_dims | |
matches_jit_signature: false | |
schema_string: aten::_sparse_coo_tensor_with_dims(int sparse_dim, int dense_dim, | |
- int[] size, *, TensorOptions options) -> Tensor | |
+ int[] size, *, ScalarType dtype, Layout layout, Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -41885,8 +41926,8 @@ | |
- name: _sparse_coo_tensor_with_dims_and_tensors | |
matches_jit_signature: false | |
schema_string: aten::_sparse_coo_tensor_with_dims_and_tensors(int sparse_dim, int | |
- dense_dim, int[] size, Tensor indices, Tensor values, *, TensorOptions options) | |
- -> Tensor | |
+ dense_dim, int[] size, Tensor indices, Tensor values, *, ScalarType dtype, Layout | |
+ layout, Device device) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -42621,8 +42662,8 @@ | |
deprecated: false | |
- name: to | |
matches_jit_signature: false | |
- schema_string: aten::to(Tensor self, TensorOptions options, bool non_blocking=False, | |
- bool copy=False) -> Tensor | |
+ schema_string: aten::to(Tensor self, *, ScalarType dtype=float, Layout layout=strided, | |
+ Device device=\"cpu\", bool non_blocking=False, bool copy=False) -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -42631,20 +42672,24 @@ | |
name: self | |
type: const Tensor & | |
- annotation: null | |
+ default: '{}' | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
- annotation: null | |
default: false | |
dynamic_type: bool | |
is_nullable: false | |
+ kwarg_only: true | |
name: non_blocking | |
type: bool | |
- annotation: null | |
default: false | |
dynamic_type: bool | |
is_nullable: false | |
+ kwarg_only: true | |
name: copy | |
type: bool | |
method_of: | |
@@ -48050,9 +48095,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: tril_indices | |
- matches_jit_signature: false | |
- schema_string: aten::tril_indices(int row, int col, int offset=0, TensorOptions | |
- options=at::kLong) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::tril_indices(int row, int col, int offset=0, *, ScalarType | |
+ dtype=long, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -48075,6 +48120,7 @@ | |
default: at::kLong | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
@@ -48094,9 +48140,9 @@ | |
with_gil: false | |
deprecated: false | |
- name: triu_indices | |
- matches_jit_signature: false | |
- schema_string: aten::triu_indices(int row, int col, int offset=0, TensorOptions | |
- options=at::kLong) -> Tensor | |
+ matches_jit_signature: true | |
+ schema_string: aten::triu_indices(int row, int col, int offset=0, *, ScalarType | |
+ dtype=long, Layout layout=strided, Device device=\"cpu\") -> Tensor | |
method_prefix_derived: '' | |
arguments: | |
- annotation: null | |
@@ -48119,6 +48165,7 @@ | |
default: at::kLong | |
dynamic_type: TensorOptions | |
is_nullable: false | |
+ kwarg_only: true | |
name: options | |
type: const TensorOptions & | |
method_of: | |
diff --git a/build/aten/src/ATen/NativeFunctions.h b/build/aten/src/ATen/NativeFunctions.h | |
index ac6b19a44..c2c9b0578 100644 | |
--- a/build/aten/src/ATen/NativeFunctions.h | |
+++ b/build/aten/src/ATen/NativeFunctions.h | |
@@ -799,7 +799,7 @@ CAFFE2_API int64_t numel(const Tensor & self); | |
CAFFE2_API std::vector<Tensor> unbind(const Tensor & self, int64_t dim=0); | |
CAFFE2_API Tensor dense_to_sparse(const Tensor & self, int64_t sparse_dim); | |
CAFFE2_API Tensor dense_to_sparse(const Tensor & self); | |
-CAFFE2_API Tensor to(const Tensor & self, const TensorOptions & options, bool non_blocking=false, bool copy=false); | |
+CAFFE2_API Tensor to(const Tensor & self, const TensorOptions & options={}, bool non_blocking=false, bool copy=false); | |
CAFFE2_API Tensor to(const Tensor & self, Device device, ScalarType dtype, bool non_blocking=false, bool copy=false); | |
CAFFE2_API Tensor to(const Tensor & self, ScalarType dtype, bool non_blocking=false, bool copy=false); | |
CAFFE2_API Tensor to(const Tensor & self, const Tensor & other, bool non_blocking=false, bool copy=false); | |
diff --git a/build/aten/src/ATen/core_tmp/Tensor.h b/build/aten/src/ATen/core_tmp/Tensor.h | |
index 699ea4674..2763f2046 100644 | |
--- a/build/aten/src/ATen/core_tmp/Tensor.h | |
+++ b/build/aten/src/ATen/core_tmp/Tensor.h | |
@@ -548,7 +548,7 @@ class CAFFE2_API Tensor { | |
std::vector<Tensor> unbind(int64_t dim=0) const; | |
Tensor to_sparse(int64_t sparse_dim) const; | |
Tensor to_sparse() const; | |
- Tensor to(const TensorOptions & options, bool non_blocking=false, bool copy=false) const; | |
+ Tensor to(const TensorOptions & options={}, bool non_blocking=false, bool copy=false) const; | |
Tensor to(Device device, ScalarType dtype, bool non_blocking=false, bool copy=false) const; | |
Tensor to(ScalarType dtype, bool non_blocking=false, bool copy=false) const; | |
Tensor to(const Tensor & other, bool non_blocking=false, bool copy=false) const; | |
diff --git a/torch/csrc/jit/generated/register_aten_ops_1.cpp b/torch/csrc/jit/generated/register_aten_ops_1.cpp | |
index 9e9b3cdf0..a7c825b38 100644 | |
--- a/torch/csrc/jit/generated/register_aten_ops_1.cpp | |
+++ b/torch/csrc/jit/generated/register_aten_ops_1.cpp | |
@@ -7846,7 +7846,7 @@ RegisterOperators reg({ | |
} | |
), | |
Operator( | |
- "aten::tril_indices(int row, int col, int offset=0, *, ScalarType dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor", | |
+ "aten::tril_indices(int row, int col, int offset=0, *, ScalarType dtype=long, Layout layout=strided, Device device=\"cpu\") -> Tensor", | |
[](Stack & stack) { | |
autograd::profiler::RecordFunction record("tril_indices"); | |
@@ -7877,7 +7877,7 @@ RegisterOperators reg({ | |
} | |
), | |
Operator( | |
- "aten::triu_indices(int row, int col, int offset=0, *, ScalarType dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor", | |
+ "aten::triu_indices(int row, int col, int offset=0, *, ScalarType dtype=long, Layout layout=strided, Device device=\"cpu\") -> Tensor", | |
[](Stack & stack) { | |
autograd::profiler::RecordFunction record("triu_indices"); | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment