From ba6567483430aa2762f2749106511b404e0e7f62 Mon Sep 17 00:00:00 2001 From: tlc Date: Sat, 14 Feb 2026 15:24:13 -0800 Subject: [PATCH 1/7] Update gotch to pytorch 2.10.0 --- device.go | 38 +- gen/gen.ml | 3 +- gen/pytorch/Declarations-v2.10.0.yaml | 215895 +++++++++++++++++++++++ libtch/c-generated.go | 704 +- libtch/lib.go | 8 +- libtch/tensor.go | 6 + libtch/torch_api.cpp | 7 +- libtch/torch_api.h | 1 + libtch/torch_api_generated.cpp.h | 900 +- libtch/torch_api_generated.h | 145 +- setup-gotch.sh | 4 +- setup-libtorch.sh | 4 +- ts/must-tensor-generated.go | 876 +- ts/tensor-generated.go | 2005 +- 14 files changed, 220102 insertions(+), 494 deletions(-) create mode 100644 gen/pytorch/Declarations-v2.10.0.yaml mode change 100644 => 100755 setup-libtorch.sh diff --git a/device.go b/device.go index 38df73e..908d982 100644 --- a/device.go +++ b/device.go @@ -12,9 +12,11 @@ type Device struct { } type Cuda Device +type Mps Device var ( CPU Device = Device{Name: "CPU", Value: -1} + MPS Mps = Mps{Name: "MPS", Value: -2} CUDA Cuda = Cuda{Name: "CUDA", Value: 0} ) @@ -34,6 +36,14 @@ func NewCuda() Device { return CudaBuilder(0) } +// MPS methods: +// ============ + +// IsAvailable returns true if MPS (Metal Performance Shaders) support is available +func (m Mps) IsAvailable() bool { + return lib.AtcMpsIsAvailable() +} + // Cuda methods: // ============= @@ -74,6 +84,8 @@ func (d Device) CInt() CInt { switch { case d.Name == "CPU": return -1 + case d.Name == "MPS": + return -2 case d.Name == "CUDA": // TODO: create a function to retrieve cuda_index var deviceIndex int = d.Value @@ -87,7 +99,9 @@ func (d Device) CInt() CInt { func (d Device) OfCInt(v CInt) Device { switch { case v == -1: - return Device{Name: "CPU", Value: 1} + return Device{Name: "CPU", Value: -1} + case v == -2: + return Device{Name: "MPS", Value: -2} case v >= 0: return CudaBuilder(uint(v)) default: @@ -124,3 +138,25 @@ func CudaIfAvailable() Device { return CPU } } + +// MPSIfAvailable returns an MPS device if available, else CPU. +func MPSIfAvailable() Device { + switch { + case MPS.IsAvailable(): + return Device{Name: "MPS", Value: -2} + default: + return CPU + } +} + +// BestAvailableDevice returns the best available device (CUDA > MPS > CPU). +func BestAvailableDevice() Device { + switch { + case CUDA.IsAvailable(): + return CudaBuilder(0) + case MPS.IsAvailable(): + return Device{Name: "MPS", Value: -2} + default: + return CPU + } +} diff --git a/gen/gen.ml b/gen/gen.ml index 16befd1..b39e8e8 100644 --- a/gen/gen.ml +++ b/gen/gen.ml @@ -73,6 +73,7 @@ let excluded_functions = ; "sym_size" ; "sym_stride" ; "sym_storage_offset" + ; "_sparse_semi_structured_addmm" ] let no_tensor_options = @@ -1441,7 +1442,7 @@ let run ~yaml_filename ~cpp_filename ~ffi_filename ~must_wrapper_filename let () = - run ~yaml_filename:"gen/pytorch/Declarations-v2.1.0.yaml" + run ~yaml_filename:"gen/pytorch/Declarations-v2.10.0.yaml" ~cpp_filename:"libtch/torch_api_generated" ~ffi_filename:"libtch/c-generated.go" ~must_wrapper_filename:"ts/must-tensor-generated.go" diff --git a/gen/pytorch/Declarations-v2.10.0.yaml b/gen/pytorch/Declarations-v2.10.0.yaml new file mode 100644 index 0000000..3bf94a5 --- /dev/null +++ b/gen/pytorch/Declarations-v2.10.0.yaml @@ -0,0 +1,215895 @@ +- name: _cast_Byte + operator_name: _cast_Byte + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cast_Byte(Tensor self, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cast_Char + operator_name: _cast_Char + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cast_Char(Tensor self, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cast_Double + operator_name: _cast_Double + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cast_Double(Tensor self, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cast_Float + operator_name: _cast_Float + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cast_Float(Tensor self, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cast_Int + operator_name: _cast_Int + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cast_Int(Tensor self, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cast_Long + operator_name: _cast_Long + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cast_Long(Tensor self, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cast_Short + operator_name: _cast_Short + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cast_Short(Tensor self, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cast_Half + operator_name: _cast_Half + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cast_Half(Tensor self, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _backward + operator_name: _backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_backward(Tensor self, Tensor[] inputs, Tensor? gradient=None, bool? retain_graph=None, bool create_graph=False) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: inputs + type: at::TensorList + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: gradient + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: retain_graph + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: create_graph + type: bool + schema_order_cpp_signature: void (const at::Tensor &, at::TensorList, const ::std::optional &, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: inputs + type: at::TensorList + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: gradient + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: retain_graph + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: create_graph + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: set_data + operator_name: set_data + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::set_data(Tensor(a!) self, Tensor new_data) -> () + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: new_data + type: const at::Tensor & + schema_order_cpp_signature: void (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: new_data + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: data + operator_name: data + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::data(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: is_leaf + operator_name: is_leaf + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_leaf(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: output_nr + operator_name: output_nr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::output_nr(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _version + operator_name: _version + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_version(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: requires_grad_ + operator_name: requires_grad_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::requires_grad_(Tensor(a!) self, bool requires_grad=True) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: requires_grad + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: requires_grad + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: retain_grad + operator_name: retain_grad + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::retain_grad(Tensor(a!) self) -> () + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: void (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: retains_grad + operator_name: retains_grad + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::retains_grad(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _fw_primal + operator_name: _fw_primal + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fw_primal(Tensor(a) self, int level) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _make_dual + operator_name: _make_dual + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_make_dual(Tensor(a) primal, Tensor tangent, int level) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: primal + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tangent + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: primal + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tangent + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _unpack_dual + operator_name: _unpack_dual + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unpack_dual(Tensor(a) dual, int level) -> (Tensor(a) primal, Tensor tangent) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: dual + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: dual + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: primal + name: primal + type: at::Tensor + - dynamic_type: at::Tensor + field_name: tangent + name: tangent + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _new_zeros_with_same_feature_meta + operator_name: _new_zeros_with_same_feature_meta + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_new_zeros_with_same_feature_meta(Tensor self, Tensor other, *, int self_num_batch_dims=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: self_num_batch_dims + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: self_num_batch_dims + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _has_same_storage_numel + operator_name: _has_same_storage_numel + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_has_same_storage_numel(Tensor self, Tensor other) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rename_ + operator_name: rename_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rename_(Tensor(a!) self, Dimname[]? names) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + name: names + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + name: names + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rename + operator_name: rename + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rename(Tensor(a) self, Dimname[]? names) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + name: names + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + name: names + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: align_to + operator_name: align_to + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::align_to(Tensor(a) self, Dimname[] names) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: names + type: at::DimnameList + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: names + type: at::DimnameList + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: align_to + operator_name: align_to + overload_name: ellipsis_idx + manual_kernel_registration: false + category_override: '' + schema_string: aten::align_to.ellipsis_idx(Tensor(a) self, Dimname[] order, int ellipsis_idx) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: order + type: at::DimnameList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ellipsis_idx + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: order + type: at::DimnameList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ellipsis_idx + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: align_as + operator_name: align_as + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::align_as(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: align_tensors + operator_name: align_tensors + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::align_tensors(Tensor[] tensors) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _assert_async + operator_name: _assert_async + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_assert_async(Tensor self) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: void (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _assert_async + operator_name: _assert_async + overload_name: msg + manual_kernel_registration: false + category_override: '' + schema_string: aten::_assert_async.msg(Tensor self, str assert_msg) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: assert_msg + type: c10::string_view + schema_order_cpp_signature: void (const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: assert_msg + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _assert_scalar + operator_name: _assert_scalar + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_assert_scalar(Scalar self, str assert_msg) -> () + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: assert_msg + type: c10::string_view + schema_order_cpp_signature: void (const at::Scalar &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: assert_msg + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _functional_assert_scalar + operator_name: _functional_assert_scalar + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_functional_assert_scalar(Scalar self, str assert_msg, Tensor dep_token) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: assert_msg + type: c10::string_view + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dep_token + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, c10::string_view, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: assert_msg + type: c10::string_view + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dep_token + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _functional_assert_async + operator_name: _functional_assert_async + overload_name: msg + manual_kernel_registration: false + category_override: '' + schema_string: aten::_functional_assert_async.msg(Tensor self, str assert_msg, Tensor dep_token) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: assert_msg + type: c10::string_view + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dep_token + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, c10::string_view, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: assert_msg + type: c10::string_view + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dep_token + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _assert_tensor_metadata + operator_name: _assert_tensor_metadata + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_assert_tensor_metadata(Tensor a, SymInt[]? size=None, SymInt[]? stride=None, ScalarType? dtype=None, *, Device? device=None, Layout? layout=None) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: a + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: size + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: stride + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + schema_order_cpp_signature: void (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: a + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: size + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: stride + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _print + operator_name: _print + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_print(str s) -> () + arguments: + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: s + type: c10::string_view + schema_order_cpp_signature: void (c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: s + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sym_constrain_range + operator_name: sym_constrain_range + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sym_constrain_range(Scalar size, *, int? min=None, int? max=None) -> () + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: size + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: min + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: max + type: ::std::optional + schema_order_cpp_signature: void (const at::Scalar &, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: size + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: min + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: max + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sym_constrain_range_for_size + operator_name: sym_constrain_range_for_size + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sym_constrain_range_for_size(Scalar size, *, int? min=None, int? max=None) -> () + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: size + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: min + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: max + type: ::std::optional + schema_order_cpp_signature: void (const at::Scalar &, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: size + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: min + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: max + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _functional_sym_constrain_range + operator_name: _functional_sym_constrain_range + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_functional_sym_constrain_range(Scalar size, int? min, int? max, Tensor dep_token) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: size + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: min + type: ::std::optional + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: max + type: ::std::optional + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dep_token + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, ::std::optional, ::std::optional, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: size + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: min + type: ::std::optional + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: max + type: ::std::optional + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dep_token + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _functional_sym_constrain_range_for_size + operator_name: _functional_sym_constrain_range_for_size + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_functional_sym_constrain_range_for_size(Scalar size, int? min, int? max, Tensor dep_token) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: size + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: min + type: ::std::optional + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: max + type: ::std::optional + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dep_token + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, ::std::optional, ::std::optional, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: size + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: min + type: ::std::optional + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: max + type: ::std::optional + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dep_token + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _make_dep_token + operator_name: _make_dep_token + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_make_dep_token(*, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: refine_names + operator_name: refine_names + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::refine_names(Tensor(a) self, Dimname[] names) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: names + type: at::DimnameList + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: names + type: at::DimnameList + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _use_cudnn_ctc_loss + operator_name: _use_cudnn_ctc_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_use_cudnn_ctc_loss(Tensor log_probs, Tensor targets, int[] input_lengths, int[] target_lengths, int blank) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + schema_order_cpp_signature: bool (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _use_cudnn_ctc_loss + operator_name: _use_cudnn_ctc_loss + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_use_cudnn_ctc_loss.Tensor(Tensor log_probs, Tensor targets, Tensor input_lengths, Tensor target_lengths, int blank) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + schema_order_cpp_signature: bool (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_ctc_loss + operator_name: _cudnn_ctc_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_ctc_loss(Tensor log_probs, Tensor targets, int[] input_lengths, int[] target_lengths, int blank, bool deterministic, bool zero_infinity) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_ctc_loss + operator_name: _cudnn_ctc_loss + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_ctc_loss.Tensor(Tensor log_probs, Tensor targets, Tensor input_lengths, Tensor target_lengths, int blank, bool deterministic, bool zero_infinity) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _use_cudnn_rnn_flatten_weight + operator_name: _use_cudnn_rnn_flatten_weight + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_use_cudnn_rnn_flatten_weight() -> bool + arguments: [] + schema_order_cpp_signature: bool () + schema_order_arguments: [] + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cudnn_rnn_flatten_weight + operator_name: _cudnn_rnn_flatten_weight + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_rnn_flatten_weight(Tensor[] weight_arr, int weight_stride0, SymInt input_size, int mode, SymInt hidden_size, SymInt proj_size, int num_layers, bool batch_first, bool bidirectional) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight_arr + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: input_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + schema_order_cpp_signature: at::Tensor (at::TensorList, int64_t, int64_t, int64_t, int64_t, int64_t, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight_arr + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: input_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_rnn + operator_name: _cudnn_rnn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_rnn(Tensor input, Tensor[] weight, int weight_stride0, Tensor? weight_buf, Tensor hx, Tensor? cx, int mode, SymInt hidden_size, SymInt proj_size, int num_layers, bool batch_first, float dropout, bool train, bool bidirectional, SymInt[] batch_sizes, Tensor? dropout_state) -> (Tensor, Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight_buf + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::TensorList, int64_t, const ::std::optional &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, int64_t, int64_t, bool, double, bool, bool, at::IntArrayRef, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight_buf + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: at::Tensor + name: result4 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_rnn_backward + operator_name: _cudnn_rnn_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_rnn_backward(Tensor input, Tensor[] weight, int weight_stride0, Tensor weight_buf, Tensor hx, Tensor? cx, Tensor output, Tensor? grad_output, Tensor? grad_hy, Tensor? grad_cy, int mode, SymInt hidden_size, SymInt proj_size, int num_layers, bool batch_first, float dropout, bool train, bool bidirectional, SymInt[] batch_sizes, Tensor? dropout_state, Tensor reserve, bool[4] output_mask) -> (Tensor, Tensor, Tensor, Tensor[]) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_buf + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserve + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple> (const at::Tensor &, at::TensorList, int64_t, const at::Tensor &, const at::Tensor &, const ::std::optional &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, int64_t, int64_t, int64_t, int64_t, bool, double, bool, bool, at::IntArrayRef, const ::std::optional &, const at::Tensor &, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_buf + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserve + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::TensorList + name: result3 + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_init_dropout_state + operator_name: _cudnn_init_dropout_state + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_init_dropout_state(float dropout, bool train, int dropout_seed, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dropout_seed + type: int64_t + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (double, bool, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dropout_seed + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _debug_has_internal_overlap + operator_name: _debug_has_internal_overlap + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_debug_has_internal_overlap(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _fused_dropout + operator_name: _fused_dropout + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_dropout(Tensor self, float p, Generator? generator=None) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _masked_scale + operator_name: _masked_scale + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_masked_scale(Tensor self, Tensor mask, float scale) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_dropout + operator_name: native_dropout + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_dropout(Tensor input, float p, bool? train) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: true + name: train + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: true + name: train + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_dropout_backward + operator_name: native_dropout_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_dropout_backward(Tensor grad_output, Tensor mask, float scale) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sobol_engine_draw + operator_name: _sobol_engine_draw + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sobol_engine_draw(Tensor quasi, int n, Tensor sobolstate, int dimension, int num_generated, ScalarType? dtype) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: quasi + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sobolstate + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_generated + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, const at::Tensor &, int64_t, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: quasi + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sobolstate + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_generated + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sobol_engine_ff_ + operator_name: _sobol_engine_ff_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sobol_engine_ff_(Tensor(a!) self, int n, Tensor sobolstate, int dimension, int num_generated) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sobolstate + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_generated + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sobolstate + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_generated + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sobol_engine_scramble_ + operator_name: _sobol_engine_scramble_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sobol_engine_scramble_(Tensor(a!) self, Tensor ltm, int dimension) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ltm + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ltm + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sobol_engine_initialize_state_ + operator_name: _sobol_engine_initialize_state_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sobol_engine_initialize_state_(Tensor(a!) self, int dimension) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _reshape_from_tensor + operator_name: _reshape_from_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_reshape_from_tensor(Tensor self, Tensor shape) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: shape + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: shape + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _shape_as_tensor + operator_name: _shape_as_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_shape_as_tensor(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: dropout + operator_name: dropout + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::dropout(Tensor input, float p, bool train) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: dropout_ + operator_name: dropout_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::dropout_(Tensor(a!) self, float p, bool train) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: feature_dropout + operator_name: feature_dropout + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::feature_dropout(Tensor input, float p, bool train) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: feature_dropout_ + operator_name: feature_dropout_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::feature_dropout_(Tensor(a!) self, float p, bool train) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: alpha_dropout + operator_name: alpha_dropout + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::alpha_dropout(Tensor input, float p, bool train) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: alpha_dropout_ + operator_name: alpha_dropout_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::alpha_dropout_(Tensor(a!) self, float p, bool train) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: feature_alpha_dropout + operator_name: feature_alpha_dropout + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::feature_alpha_dropout(Tensor input, float p, bool train) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: feature_alpha_dropout_ + operator_name: feature_alpha_dropout_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::feature_alpha_dropout_(Tensor(a!) self, float p, bool train) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: abs + operator_name: abs + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::abs(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: abs_ + operator_name: abs_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::abs_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: abs_out + operator_name: abs + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::abs.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: absolute + operator_name: absolute + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::absolute(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: absolute_ + operator_name: absolute_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::absolute_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: absolute_out + operator_name: absolute + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::absolute.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: angle + operator_name: angle + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::angle(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: angle_out + operator_name: angle + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::angle.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: view_as_real + operator_name: view_as_real + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_as_real(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: view_as_complex + operator_name: view_as_complex + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_as_complex(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sgn + operator_name: sgn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sgn(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sgn_ + operator_name: sgn_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sgn_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sgn_out + operator_name: sgn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sgn.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: chalf + operator_name: chalf + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::chalf(Tensor self, *, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: real + operator_name: real + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::real(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: imag + operator_name: imag + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::imag(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _conj + operator_name: _conj + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_conj(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: conj + operator_name: conj + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::conj(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _conj_physical + operator_name: _conj_physical + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_conj_physical(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: conj_physical + operator_name: conj_physical + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::conj_physical(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conj_physical_out + operator_name: conj_physical + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::conj_physical.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: conj_physical_ + operator_name: conj_physical_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::conj_physical_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: resolve_conj + operator_name: resolve_conj + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::resolve_conj(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: resolve_neg + operator_name: resolve_neg + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::resolve_neg(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _neg_view + operator_name: _neg_view + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_neg_view(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: acos + operator_name: acos + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::acos(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: acos_ + operator_name: acos_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::acos_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: acos_out + operator_name: acos + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::acos.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arccos + operator_name: arccos + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arccos(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arccos_ + operator_name: arccos_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arccos_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arccos_out + operator_name: arccos + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::arccos.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: avg_pool1d + operator_name: avg_pool1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::avg_pool1d(Tensor self, int[1] kernel_size, int[1] stride=[], int[1] padding=0, bool ceil_mode=False, bool count_include_pad=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: adaptive_avg_pool1d + operator_name: adaptive_avg_pool1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_avg_pool1d(Tensor self, int[1] output_size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: adaptive_max_pool1d + operator_name: adaptive_max_pool1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_max_pool1d(Tensor self, int[1] output_size) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: add + operator_name: add + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::add.Tensor(Tensor self, Tensor other, *, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: add_ + operator_name: add_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::add_.Tensor(Tensor(a!) self, Tensor other, *, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: add_out + operator_name: add + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::add.out(Tensor self, Tensor other, *, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _add_relu + operator_name: _add_relu + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_add_relu.Tensor(Tensor self, Tensor other, *, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _add_relu_ + operator_name: _add_relu_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_add_relu_.Tensor(Tensor(a!) self, Tensor other, *, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _add_relu_out + operator_name: _add_relu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_add_relu.out(Tensor self, Tensor other, *, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _add_relu + operator_name: _add_relu + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_add_relu.Scalar(Tensor self, Scalar other, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _add_relu_ + operator_name: _add_relu_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_add_relu_.Scalar(Tensor(a!) self, Scalar other, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: add + operator_name: add + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::add.Scalar(Tensor self, Scalar other, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: add_ + operator_name: add_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::add_.Scalar(Tensor(a!) self, Scalar other, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addmv + operator_name: addmv + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addmv(Tensor self, Tensor mat, Tensor vec, *, Scalar beta=1, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addmv_ + operator_name: addmv_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addmv_(Tensor(a!) self, Tensor mat, Tensor vec, *, Scalar beta=1, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addmv_out + operator_name: addmv + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::addmv.out(Tensor self, Tensor mat, Tensor vec, *, Scalar beta=1, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addr + operator_name: addr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addr(Tensor self, Tensor vec1, Tensor vec2, *, Scalar beta=1, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addr_ + operator_name: addr_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addr_(Tensor(a!) self, Tensor vec1, Tensor vec2, *, Scalar beta=1, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addr_out + operator_name: addr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::addr.out(Tensor self, Tensor vec1, Tensor vec2, *, Scalar beta=1, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: affine_grid_generator + operator_name: affine_grid_generator + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::affine_grid_generator(Tensor theta, SymInt[] size, bool align_corners) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: theta + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: theta + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: affine_grid_generator_backward + operator_name: affine_grid_generator_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::affine_grid_generator_backward(Tensor grad, SymInt[] size, bool align_corners) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _is_all_true + operator_name: _is_all_true + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_is_all_true(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _is_any_true + operator_name: _is_any_true + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_is_any_true(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_check_tensor + operator_name: _test_check_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_check_tensor(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _test_functorch_fallback + operator_name: _test_functorch_fallback + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_functorch_fallback(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: all + operator_name: all + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::all.dim(Tensor self, int dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: all + operator_name: all + overload_name: dims + manual_kernel_registration: false + category_override: '' + schema_string: aten::all.dims(Tensor self, int[]? dim=None, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: all_out + operator_name: all + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::all.out(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: all_out + operator_name: all + overload_name: dims_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::all.dims_out(Tensor self, int[]? dim=None, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: all + operator_name: all + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::all.dimname(Tensor self, Dimname dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: all_out + operator_name: all + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::all.dimname_out(Tensor self, Dimname dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Dimname, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: allclose + operator_name: allclose + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::allclose(Tensor self, Tensor other, float rtol=1e-05, float atol=1e-08, bool equal_nan=False) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1.0e-05 + dynamic_type: double + is_nullable: false + name: rtol + type: double + - annotation: null + default: 1.0e-08 + dynamic_type: double + is_nullable: false + name: atol + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: equal_nan + type: bool + schema_order_cpp_signature: bool (const at::Tensor &, const at::Tensor &, double, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1.0e-05 + dynamic_type: double + is_nullable: false + name: rtol + type: double + - annotation: null + default: 1.0e-08 + dynamic_type: double + is_nullable: false + name: atol + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: equal_nan + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: any + operator_name: any + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::any.dim(Tensor self, int dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: any + operator_name: any + overload_name: dims + manual_kernel_registration: false + category_override: '' + schema_string: aten::any.dims(Tensor self, int[]? dim=None, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: any_out + operator_name: any + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::any.out(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: any_out + operator_name: any + overload_name: dims_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::any.dims_out(Tensor self, int[]? dim=None, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: any + operator_name: any + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::any.dimname(Tensor self, Dimname dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: any_out + operator_name: any + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::any.dimname_out(Tensor self, Dimname dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Dimname, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arange + operator_name: arange + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arange(Scalar end, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Scalar &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arange + operator_name: arange + overload_name: start + manual_kernel_registration: false + category_override: '' + schema_string: aten::arange.start(Scalar start, Scalar end, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Scalar &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arange + operator_name: arange + overload_name: start_step + manual_kernel_registration: false + category_override: '' + schema_string: aten::arange.start_step(Scalar start, Scalar end, Scalar step=1, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: step + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Scalar &, const at::Scalar &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: step + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arange_out + operator_name: arange + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::arange.out(Scalar end, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arange_out + operator_name: arange + overload_name: start_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::arange.start_out(Scalar start, Scalar end, Scalar step=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: step + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: step + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _dim_arange + operator_name: _dim_arange + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_dim_arange(Tensor like, int dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: like + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: like + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: argmax + operator_name: argmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::argmax(Tensor self, int? dim=None, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: argmax_out + operator_name: argmax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::argmax.out(Tensor self, int? dim=None, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: argmin + operator_name: argmin + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::argmin(Tensor self, int? dim=None, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: argmin_out + operator_name: argmin + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::argmin.out(Tensor self, int? dim=None, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: acosh + operator_name: acosh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::acosh(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: acosh_ + operator_name: acosh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::acosh_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: acosh_out + operator_name: acosh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::acosh.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arccosh + operator_name: arccosh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arccosh(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arccosh_ + operator_name: arccosh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arccosh_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arccosh_out + operator_name: arccosh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::arccosh.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: asinh + operator_name: asinh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::asinh(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: asinh_ + operator_name: asinh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::asinh_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: asinh_out + operator_name: asinh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::asinh.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arcsinh + operator_name: arcsinh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arcsinh(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arcsinh_ + operator_name: arcsinh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arcsinh_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arcsinh_out + operator_name: arcsinh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::arcsinh.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: atanh + operator_name: atanh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::atanh(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: atanh_ + operator_name: atanh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::atanh_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: atanh_out + operator_name: atanh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::atanh.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arctanh + operator_name: arctanh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arctanh(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arctanh_ + operator_name: arctanh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arctanh_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arctanh_out + operator_name: arctanh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::arctanh.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: as_strided + operator_name: as_strided + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::as_strided(Tensor(a) self, SymInt[] size, SymInt[] stride, SymInt? storage_offset=None) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: as_strided_ + operator_name: as_strided_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::as_strided_(Tensor(a!) self, SymInt[] size, SymInt[] stride, SymInt? storage_offset=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: const at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: asin + operator_name: asin + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::asin(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: asin_ + operator_name: asin_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::asin_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: asin_out + operator_name: asin + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::asin.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arcsin + operator_name: arcsin + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arcsin(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arcsin_ + operator_name: arcsin_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arcsin_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arcsin_out + operator_name: arcsin + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::arcsin.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: atan + operator_name: atan + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::atan(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: atan_ + operator_name: atan_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::atan_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: atan_out + operator_name: atan + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::atan.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arctan + operator_name: arctan + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arctan(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arctan_ + operator_name: arctan_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arctan_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arctan_out + operator_name: arctan + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::arctan.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: atleast_1d + operator_name: atleast_1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::atleast_1d(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: atleast_1d + operator_name: atleast_1d + overload_name: Sequence + manual_kernel_registration: false + category_override: '' + schema_string: aten::atleast_1d.Sequence(Tensor[] tensors) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: atleast_2d + operator_name: atleast_2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::atleast_2d(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: atleast_2d + operator_name: atleast_2d + overload_name: Sequence + manual_kernel_registration: false + category_override: '' + schema_string: aten::atleast_2d.Sequence(Tensor[] tensors) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: atleast_3d + operator_name: atleast_3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::atleast_3d(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: atleast_3d + operator_name: atleast_3d + overload_name: Sequence + manual_kernel_registration: false + category_override: '' + schema_string: aten::atleast_3d.Sequence(Tensor[] tensors) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: baddbmm + operator_name: baddbmm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::baddbmm(Tensor self, Tensor batch1, Tensor batch2, *, Scalar beta=1, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: baddbmm_ + operator_name: baddbmm_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::baddbmm_(Tensor(a!) self, Tensor batch1, Tensor batch2, *, Scalar beta=1, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: baddbmm_out + operator_name: baddbmm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::baddbmm.out(Tensor self, Tensor batch1, Tensor batch2, *, Scalar beta=1, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: baddbmm + operator_name: baddbmm + overload_name: dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::baddbmm.dtype(Tensor self, Tensor batch1, Tensor batch2, ScalarType out_dtype, *, Scalar beta=1, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::ScalarType, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: baddbmm_out + operator_name: baddbmm + overload_name: dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::baddbmm.dtype_out(Tensor self, Tensor batch1, Tensor batch2, ScalarType out_dtype, *, Scalar beta=1, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::ScalarType, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bartlett_window + operator_name: bartlett_window + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::bartlett_window(int window_length, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bartlett_window + operator_name: bartlett_window + overload_name: periodic + manual_kernel_registration: false + category_override: '' + schema_string: aten::bartlett_window.periodic(int window_length, bool periodic, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, bool, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm + operator_name: batch_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm(Tensor input, Tensor? weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, bool training, float momentum, float eps, bool cudnn_enabled) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: quantized_batch_norm + operator_name: quantized_batch_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_batch_norm(Tensor input, Tensor? weight, Tensor? bias, Tensor mean, Tensor var, float eps, float output_scale, int output_zero_point) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: var + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: output_scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: output_zero_point + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, double, double, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: var + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: output_scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: output_zero_point + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _batch_norm_impl_index + operator_name: _batch_norm_impl_index + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_batch_norm_impl_index(Tensor input, Tensor? weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, bool training, float momentum, float eps, bool cudnn_enabled) -> (Tensor, Tensor, Tensor, Tensor, int) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: int64_t + name: result4 + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _batch_norm_impl_index_backward + operator_name: _batch_norm_impl_index_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_batch_norm_impl_index_backward(int impl_index, Tensor input, Tensor grad_output, Tensor? weight, Tensor? running_mean, Tensor? running_var, Tensor? save_mean, Tensor? save_var_transform, bool train, float eps, bool[3] output_mask, Tensor reservedSpace) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: impl_index + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var_transform + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reservedSpace + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (int64_t, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, ::std::array, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: impl_index + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var_transform + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reservedSpace + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: bernoulli + operator_name: bernoulli + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::bernoulli(Tensor self, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bernoulli_out + operator_name: bernoulli + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bernoulli.out(Tensor self, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bernoulli_ + operator_name: bernoulli_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bernoulli_.Tensor(Tensor(a!) self, Tensor p, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: p + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: p + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bernoulli_ + operator_name: bernoulli_ + overload_name: float + manual_kernel_registration: false + category_override: '' + schema_string: aten::bernoulli_.float(Tensor(a!) self, float p=0.5, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bernoulli + operator_name: bernoulli + overload_name: p + manual_kernel_registration: false + category_override: '' + schema_string: aten::bernoulli.p(Tensor self, float p, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bilinear + operator_name: bilinear + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::bilinear(Tensor input1, Tensor input2, Tensor weight, Tensor? bias=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: binary_cross_entropy + operator_name: binary_cross_entropy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::binary_cross_entropy(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: binary_cross_entropy_out + operator_name: binary_cross_entropy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::binary_cross_entropy.out(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: binary_cross_entropy_backward + operator_name: binary_cross_entropy_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::binary_cross_entropy_backward(Tensor grad_output, Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: binary_cross_entropy_backward_out + operator_name: binary_cross_entropy_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::binary_cross_entropy_backward.grad_input(Tensor grad_output, Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: binary_cross_entropy_with_logits + operator_name: binary_cross_entropy_with_logits + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::binary_cross_entropy_with_logits(Tensor self, Tensor target, Tensor? weight=None, Tensor? pos_weight=None, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: pos_weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: pos_weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bincount + operator_name: bincount + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::bincount(Tensor self, Tensor? weights=None, SymInt minlength=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weights + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: minlength + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weights + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: minlength + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_not + operator_name: bitwise_not + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_not(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_not_ + operator_name: bitwise_not_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_not_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_not_out + operator_name: bitwise_not + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_not.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: copysign_out + operator_name: copysign + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::copysign.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: copysign + operator_name: copysign + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::copysign.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: copysign_ + operator_name: copysign_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::copysign_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: copysign + operator_name: copysign + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::copysign.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: copysign_ + operator_name: copysign_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::copysign_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: copysign_out + operator_name: copysign + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::copysign.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _lazy_clone + operator_name: _lazy_clone + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_lazy_clone(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_not + operator_name: logical_not + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_not(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_not_ + operator_name: logical_not_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_not_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_not_out + operator_name: logical_not + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_not.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_xor + operator_name: logical_xor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_xor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_xor_ + operator_name: logical_xor_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_xor_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_xor_out + operator_name: logical_xor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_xor.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_and + operator_name: logical_and + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_and(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_and_ + operator_name: logical_and_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_and_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_and_out + operator_name: logical_and + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_and.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_or + operator_name: logical_or + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_or(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_or_ + operator_name: logical_or_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_or_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logical_or_out + operator_name: logical_or + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logical_or.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: blackman_window + operator_name: blackman_window + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::blackman_window(int window_length, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: blackman_window + operator_name: blackman_window + overload_name: periodic + manual_kernel_registration: false + category_override: '' + schema_string: aten::blackman_window.periodic(int window_length, bool periodic, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, bool, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bmm + operator_name: bmm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::bmm(Tensor self, Tensor mat2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bmm_out + operator_name: bmm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bmm.out(Tensor self, Tensor mat2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bmm + operator_name: bmm + overload_name: dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::bmm.dtype(Tensor self, Tensor mat2, ScalarType out_dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bmm_out + operator_name: bmm + overload_name: dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bmm.dtype_out(Tensor self, Tensor mat2, ScalarType out_dtype, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: broadcast_tensors + operator_name: broadcast_tensors + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::broadcast_tensors(Tensor[] tensors) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: broadcast_to + operator_name: broadcast_to + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::broadcast_to(Tensor(a) self, SymInt[] size) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_broadcast_to + operator_name: _sparse_broadcast_to + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_broadcast_to(Tensor(a) self, int[] size) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cat + operator_name: cat + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cat(Tensor[] tensors, int dim=0) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::ITensorListRef & + is_nullable: false + name: tensors + type: const at::ITensorListRef & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::ITensorListRef &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: const at::ITensorListRef & + is_nullable: false + name: tensors + type: const at::ITensorListRef & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cat_out + operator_name: cat + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cat.out(Tensor[] tensors, int dim=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::ITensorListRef & + is_nullable: false + name: tensors + type: const at::ITensorListRef & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::ITensorListRef &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::ITensorListRef & + is_nullable: false + name: tensors + type: const at::ITensorListRef & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cat + operator_name: cat + overload_name: names + manual_kernel_registration: false + category_override: '' + schema_string: aten::cat.names(Tensor[] tensors, Dimname dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor (at::TensorList, at::Dimname) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cat_out + operator_name: cat + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cat.names_out(Tensor[] tensors, Dimname dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Dimname, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: concat + operator_name: concat + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::concat(Tensor[] tensors, int dim=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (at::TensorList, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: concat_out + operator_name: concat + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::concat.out(Tensor[] tensors, int dim=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::TensorList, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: concat + operator_name: concat + overload_name: names + manual_kernel_registration: false + category_override: '' + schema_string: aten::concat.names(Tensor[] tensors, Dimname dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor (at::TensorList, at::Dimname) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: concat_out + operator_name: concat + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::concat.names_out(Tensor[] tensors, Dimname dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Dimname, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: concatenate + operator_name: concatenate + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::concatenate(Tensor[] tensors, int dim=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (at::TensorList, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: concatenate_out + operator_name: concatenate + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::concatenate.out(Tensor[] tensors, int dim=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::TensorList, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: concatenate + operator_name: concatenate + overload_name: names + manual_kernel_registration: false + category_override: '' + schema_string: aten::concatenate.names(Tensor[] tensors, Dimname dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor (at::TensorList, at::Dimname) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: concatenate_out + operator_name: concatenate + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::concatenate.names_out(Tensor[] tensors, Dimname dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Dimname, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: block_diag + operator_name: block_diag + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::block_diag(Tensor[] tensors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ceil + operator_name: ceil + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::ceil(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ceil_ + operator_name: ceil_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::ceil_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ceil_out + operator_name: ceil + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ceil.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: chain_matmul + operator_name: chain_matmul + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::chain_matmul(Tensor[] matrices) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: matrices + type: at::TensorList + schema_order_cpp_signature: at::Tensor (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: matrices + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: chain_matmul_out + operator_name: chain_matmul + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::chain_matmul.out(Tensor[] matrices, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: matrices + type: at::TensorList + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: matrices + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: unsafe_chunk + operator_name: unsafe_chunk + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unsafe_chunk(Tensor self, int chunks, int dim=0) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: chunks + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: chunks + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: chunk + operator_name: chunk + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::chunk(Tensor(a -> *) self, int chunks, int dim=0) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: chunks + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: chunks + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: tensor_split + operator_name: tensor_split + overload_name: sections + manual_kernel_registration: false + category_override: '' + schema_string: aten::tensor_split.sections(Tensor(a -> *) self, SymInt sections, int dim=0) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sections + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sections + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: tensor_split + operator_name: tensor_split + overload_name: indices + manual_kernel_registration: false + category_override: '' + schema_string: aten::tensor_split.indices(Tensor(a -> *) self, SymInt[] indices, int dim=0) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: indices + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: indices + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: tensor_split + operator_name: tensor_split + overload_name: tensor_indices_or_sections + manual_kernel_registration: false + category_override: '' + schema_string: aten::tensor_split.tensor_indices_or_sections(Tensor(a -> *) self, Tensor tensor_indices_or_sections, int dim=0) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor_indices_or_sections + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor_indices_or_sections + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: clamp + operator_name: clamp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp(Tensor self, Scalar? min=None, Scalar? max=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp + operator_name: clamp + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp.Tensor(Tensor self, Tensor? min=None, Tensor? max=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_ + operator_name: clamp_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_(Tensor(a!) self, Scalar? min=None, Scalar? max=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_ + operator_name: clamp_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_.Tensor(Tensor(a!) self, Tensor? min=None, Tensor? max=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_out + operator_name: clamp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp.out(Tensor self, Scalar? min=None, Scalar? max=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_out + operator_name: clamp + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp.Tensor_out(Tensor self, Tensor? min=None, Tensor? max=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_max + operator_name: clamp_max + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_max(Tensor self, Scalar max) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: max + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: max + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_max + operator_name: clamp_max + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_max.Tensor(Tensor self, Tensor max) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: max + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: max + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_max_ + operator_name: clamp_max_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_max_(Tensor(a!) self, Scalar max) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: max + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: max + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_max_ + operator_name: clamp_max_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_max_.Tensor(Tensor(a!) self, Tensor max) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: max + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: max + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_max_out + operator_name: clamp_max + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_max.out(Tensor self, Scalar max, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: max + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: max + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_max_out + operator_name: clamp_max + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_max.Tensor_out(Tensor self, Tensor max, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: max + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: max + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_min + operator_name: clamp_min + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_min(Tensor self, Scalar min) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: min + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: min + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_min + operator_name: clamp_min + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_min.Tensor(Tensor self, Tensor min) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: min + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: min + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_min_ + operator_name: clamp_min_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_min_(Tensor(a!) self, Scalar min) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: min + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: min + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_min_ + operator_name: clamp_min_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_min_.Tensor(Tensor(a!) self, Tensor min) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: min + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: min + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_min_out + operator_name: clamp_min + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_min.out(Tensor self, Scalar min, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: min + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: min + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clamp_min_out + operator_name: clamp_min + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::clamp_min.Tensor_out(Tensor self, Tensor min, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: min + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: min + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: clip + operator_name: clip + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::clip(Tensor self, Scalar? min=None, Scalar? max=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: clip + operator_name: clip + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::clip.Tensor(Tensor self, Tensor? min=None, Tensor? max=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: clip_ + operator_name: clip_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::clip_(Tensor(a!) self, Scalar? min=None, Scalar? max=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: clip_ + operator_name: clip_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::clip_.Tensor(Tensor(a!) self, Tensor? min=None, Tensor? max=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: clip_out + operator_name: clip + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::clip.out(Tensor self, Scalar? min=None, Scalar? max=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: max + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: clip_out + operator_name: clip + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::clip.Tensor_out(Tensor self, Tensor? min=None, Tensor? max=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cudnn_is_acceptable + operator_name: cudnn_is_acceptable + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_is_acceptable(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: complex + operator_name: complex + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::complex(Tensor real, Tensor imag) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: real + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: imag + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: real + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: imag + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: complex_out + operator_name: complex + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::complex.out(Tensor real, Tensor imag, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: real + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: imag + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: real + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: imag + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: polar + operator_name: polar + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::polar(Tensor abs, Tensor angle) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: abs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: angle + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: abs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: angle + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: polar_out + operator_name: polar + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::polar.out(Tensor abs, Tensor angle, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: abs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: angle + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: abs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: angle + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: constant_pad_nd + operator_name: constant_pad_nd + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::constant_pad_nd(Tensor self, SymInt[] pad, Scalar value=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: pad + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: pad + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: contiguous + operator_name: contiguous + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::contiguous(Tensor(a) self, *, MemoryFormat memory_format=contiguous_format) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: false + kwarg_only: true + name: memory_format + type: at::MemoryFormat + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::MemoryFormat) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: false + kwarg_only: true + name: memory_format + type: at::MemoryFormat + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: convolution + operator_name: convolution + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::convolution(Tensor input, Tensor weight, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: convolution_backward + operator_name: convolution_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::convolution_backward(Tensor grad_output, Tensor input, Tensor weight, SymInt[]? bias_sizes, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups, bool[3] output_mask) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: bias_sizes + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: bias_sizes + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: convolution_overrideable + operator_name: convolution_overrideable + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::convolution_overrideable(Tensor input, Tensor weight, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: convolution_backward_overrideable + operator_name: convolution_backward_overrideable + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::convolution_backward_overrideable(Tensor grad_output, Tensor input, Tensor weight, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups, bool[3] output_mask) -> (Tensor grad_input, Tensor grad_weight, Tensor grad_bias) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: grad_input + name: grad_input + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_weight + name: grad_weight + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_bias + name: grad_bias + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _convolution + operator_name: _convolution + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convolution(Tensor input, Tensor weight, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups, bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t, bool, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _convolution + operator_name: _convolution + overload_name: deprecated + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convolution.deprecated(Tensor input, Tensor weight, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, int[] output_padding, SymInt groups, bool benchmark, bool deterministic, bool cudnn_enabled) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _convolution_mode + operator_name: _convolution_mode + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convolution_mode(Tensor input, Tensor weight, Tensor? bias, SymInt[] stride, str padding, SymInt[] dilation, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: padding + type: c10::string_view + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, c10::string_view, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: padding + type: c10::string_view + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _convolution_double_backward + operator_name: _convolution_double_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convolution_double_backward(Tensor? ggI, Tensor? ggW, Tensor? ggb, Tensor gO, Tensor weight, Tensor self, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups, bool[3] output_mask) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: ggI + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: ggW + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: ggb + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: gO + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const ::std::optional &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: ggI + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: ggW + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: ggb + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: gO + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conv1d + operator_name: conv1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv1d(Tensor input, Tensor weight, Tensor? bias=None, SymInt[1] stride=1, SymInt[1] padding=0, SymInt[1] dilation=1, SymInt groups=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conv2d + operator_name: conv2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv2d(Tensor input, Tensor weight, Tensor? bias=None, SymInt[2] stride=1, SymInt[2] padding=0, SymInt[2] dilation=1, SymInt groups=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conv3d + operator_name: conv3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv3d(Tensor input, Tensor weight, Tensor? bias=None, SymInt[3] stride=1, SymInt[3] padding=0, SymInt[3] dilation=1, SymInt groups=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conv1d + operator_name: conv1d + overload_name: padding + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv1d.padding(Tensor input, Tensor weight, Tensor? bias=None, SymInt[1] stride=1, str padding="valid", SymInt[1] dilation=1, SymInt groups=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: '"valid"' + dynamic_type: c10::string_view + is_nullable: false + name: padding + type: c10::string_view + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, c10::string_view, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: '"valid"' + dynamic_type: c10::string_view + is_nullable: false + name: padding + type: c10::string_view + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conv2d + operator_name: conv2d + overload_name: padding + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv2d.padding(Tensor input, Tensor weight, Tensor? bias=None, SymInt[2] stride=1, str padding="valid", SymInt[2] dilation=1, SymInt groups=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: '"valid"' + dynamic_type: c10::string_view + is_nullable: false + name: padding + type: c10::string_view + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, c10::string_view, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: '"valid"' + dynamic_type: c10::string_view + is_nullable: false + name: padding + type: c10::string_view + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conv3d + operator_name: conv3d + overload_name: padding + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv3d.padding(Tensor input, Tensor weight, Tensor? bias=None, SymInt[3] stride=1, str padding="valid", SymInt[3] dilation=1, SymInt groups=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: '"valid"' + dynamic_type: c10::string_view + is_nullable: false + name: padding + type: c10::string_view + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, c10::string_view, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: '"valid"' + dynamic_type: c10::string_view + is_nullable: false + name: padding + type: c10::string_view + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conv_tbc + operator_name: conv_tbc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv_tbc(Tensor self, Tensor weight, Tensor bias, int pad=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: pad + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: pad + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: conv_tbc_backward + operator_name: conv_tbc_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv_tbc_backward(Tensor self, Tensor input, Tensor weight, Tensor bias, int pad) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: pad + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: pad + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conv_transpose1d + operator_name: conv_transpose1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv_transpose1d(Tensor input, Tensor weight, Tensor? bias=None, SymInt[1] stride=1, SymInt[1] padding=0, SymInt[1] output_padding=0, SymInt groups=1, SymInt[1] dilation=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conv_transpose2d + operator_name: conv_transpose2d + overload_name: input + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv_transpose2d.input(Tensor input, Tensor weight, Tensor? bias=None, SymInt[2] stride=1, SymInt[2] padding=0, SymInt[2] output_padding=0, SymInt groups=1, SymInt[2] dilation=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: conv_transpose3d + operator_name: conv_transpose3d + overload_name: input + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv_transpose3d.input(Tensor input, Tensor weight, Tensor? bias=None, SymInt[3] stride=1, SymInt[3] padding=0, SymInt[3] output_padding=0, SymInt groups=1, SymInt[3] dilation=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: copy + operator_name: copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::copy(Tensor self, Tensor src, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: copy_ + operator_name: copy_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::copy_(Tensor(a!) self, Tensor src, bool non_blocking=False) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _copy_from + operator_name: _copy_from + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_copy_from(Tensor self, Tensor dst, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dst + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dst + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _copy_from_and_resize + operator_name: _copy_from_and_resize + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_copy_from_and_resize(Tensor self, Tensor dst) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dst + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dst + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cos + operator_name: cos + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cos(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cos_ + operator_name: cos_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cos_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cos_out + operator_name: cos + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cos.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cosh + operator_name: cosh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cosh(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cosh_ + operator_name: cosh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cosh_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cosh_out + operator_name: cosh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cosh.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cosine_embedding_loss + operator_name: cosine_embedding_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cosine_embedding_loss(Tensor input1, Tensor input2, Tensor target, float margin=0.0, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: margin + type: double + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, double, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: margin + type: double + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: count_nonzero + operator_name: count_nonzero + overload_name: dim_IntList + manual_kernel_registration: false + category_override: '' + schema_string: aten::count_nonzero.dim_IntList(Tensor self, int[] dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: count_nonzero + operator_name: count_nonzero + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::count_nonzero(Tensor self, int? dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cov + operator_name: cov + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cov(Tensor self, *, int correction=1, Tensor? fweights=None, Tensor? aweights=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: correction + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: fweights + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: aweights + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: correction + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: fweights + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: aweights + type: const ::std::optional & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: corrcoef + operator_name: corrcoef + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::corrcoef(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cudnn_affine_grid_generator + operator_name: cudnn_affine_grid_generator + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_affine_grid_generator(Tensor theta, int N, int C, int H, int W) -> Tensor grid + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: theta + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: H + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: W + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: theta + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: H + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: W + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: grid + name: grid + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_affine_grid_generator_backward + operator_name: cudnn_affine_grid_generator_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_affine_grid_generator_backward(Tensor grad, int N, int C, int H, int W) -> Tensor grad_theta + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: H + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: W + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: H + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: W + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: grad_theta + name: grad_theta + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_batch_norm + operator_name: cudnn_batch_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_batch_norm(Tensor input, Tensor weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, bool training, float exponential_average_factor, float epsilon) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: exponential_average_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: exponential_average_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_batch_norm_out + operator_name: cudnn_batch_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_batch_norm.out(Tensor input, Tensor weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, bool training, float exponential_average_factor, float epsilon, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: exponential_average_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, double, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: exponential_average_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_batch_norm_backward + operator_name: cudnn_batch_norm_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_batch_norm_backward(Tensor input, Tensor grad_output, Tensor weight, Tensor? running_mean, Tensor? running_var, Tensor? save_mean, Tensor? save_var, float epsilon, Tensor reserveSpace) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserveSpace + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, double, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserveSpace + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_convolution + operator_name: cudnn_convolution + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_convolution(Tensor self, Tensor weight, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool benchmark, bool deterministic, bool allow_tf32) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_convolution_out + operator_name: cudnn_convolution + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_convolution.out(Tensor self, Tensor weight, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool benchmark, bool deterministic, bool allow_tf32, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_convolution_transpose + operator_name: cudnn_convolution_transpose + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_convolution_transpose(Tensor self, Tensor weight, SymInt[] padding, SymInt[] output_padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool benchmark, bool deterministic, bool allow_tf32) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _mps_convolution_transpose + operator_name: _mps_convolution_transpose + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_mps_convolution_transpose(Tensor self, Tensor weight, SymInt[] padding, SymInt[] output_padding, SymInt[] stride, SymInt[] dilation, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mps_convolution_transpose_backward + operator_name: mps_convolution_transpose_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mps_convolution_transpose_backward(Tensor self, Tensor grad_output, Tensor weight, SymInt[] padding, SymInt[] output_padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool[2] output_mask) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_convolution_relu + operator_name: cudnn_convolution_relu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_convolution_relu(Tensor self, Tensor weight, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_convolution_add_relu + operator_name: cudnn_convolution_add_relu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_convolution_add_relu(Tensor self, Tensor weight, Tensor z, Scalar? alpha, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: z + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: alpha + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: z + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: alpha + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_grid_sampler + operator_name: cudnn_grid_sampler + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_grid_sampler(Tensor self, Tensor grid) -> Tensor output + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_grid_sampler_backward + operator_name: cudnn_grid_sampler_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_grid_sampler_backward(Tensor self, Tensor grid, Tensor grad_output) -> (Tensor grad_self, Tensor grad_grid) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: grad_self + name: grad_self + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_grid + name: grad_grid + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cummax + operator_name: cummax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cummax(Tensor self, int dim) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cummax_out + operator_name: cummax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cummax.out(Tensor self, int dim, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cummax + operator_name: cummax + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::cummax.dimname(Tensor self, Dimname dim) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cummax_out + operator_name: cummax + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cummax.dimname_out(Tensor self, Dimname dim, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cummax_helper + operator_name: _cummax_helper + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cummax_helper(Tensor self, Tensor(a!) values, Tensor(b!) indices, int dim) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: values + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: void (const at::Tensor &, at::Tensor &, at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: values + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cummin + operator_name: cummin + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cummin(Tensor self, int dim) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cummin_out + operator_name: cummin + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cummin.out(Tensor self, int dim, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cummin + operator_name: cummin + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::cummin.dimname(Tensor self, Dimname dim) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cummin_out + operator_name: cummin + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cummin.dimname_out(Tensor self, Dimname dim, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cummin_helper + operator_name: _cummin_helper + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cummin_helper(Tensor self, Tensor(a!) values, Tensor(b!) indices, int dim) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: values + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: void (const at::Tensor &, at::Tensor &, at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: values + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cummaxmin_backward + operator_name: cummaxmin_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cummaxmin_backward(Tensor grad, Tensor input, Tensor indices, int dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: cumprod + operator_name: cumprod + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumprod(Tensor self, int dim, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cumprod_ + operator_name: cumprod_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumprod_(Tensor(a!) self, int dim, *, ScalarType? dtype=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cumprod_out + operator_name: cumprod + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumprod.out(Tensor self, int dim, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cumprod + operator_name: cumprod + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumprod.dimname(Tensor self, Dimname dim, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cumprod_ + operator_name: cumprod_ + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumprod_.dimname(Tensor(a!) self, Dimname dim, *, ScalarType? dtype=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::Dimname, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cumprod_out + operator_name: cumprod + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumprod.dimname_out(Tensor self, Dimname dim, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Dimname, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cumprod_backward + operator_name: cumprod_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumprod_backward(Tensor grad, Tensor input, int dim, Tensor output) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: cumsum + operator_name: cumsum + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumsum(Tensor self, int dim, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cumsum_ + operator_name: cumsum_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumsum_(Tensor(a!) self, int dim, *, ScalarType? dtype=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cumsum_out + operator_name: cumsum + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumsum.out(Tensor self, int dim, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cumsum + operator_name: cumsum + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumsum.dimname(Tensor self, Dimname dim, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cumsum_ + operator_name: cumsum_ + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumsum_.dimname(Tensor(a!) self, Dimname dim, *, ScalarType? dtype=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::Dimname, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cumsum_out + operator_name: cumsum + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumsum.dimname_out(Tensor self, Dimname dim, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Dimname, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cumulative_trapezoid + operator_name: cumulative_trapezoid + overload_name: x + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumulative_trapezoid.x(Tensor y, Tensor x, *, int dim=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cumulative_trapezoid + operator_name: cumulative_trapezoid + overload_name: dx + manual_kernel_registration: false + category_override: '' + schema_string: aten::cumulative_trapezoid.dx(Tensor y, *, Scalar dx=1, int dim=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: dx + type: const at::Scalar & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: dx + type: const at::Scalar & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: ctc_loss + operator_name: ctc_loss + overload_name: IntList + manual_kernel_registration: false + category_override: '' + schema_string: aten::ctc_loss.IntList(Tensor log_probs, Tensor targets, int[] input_lengths, int[] target_lengths, int blank=0, int reduction=Mean, bool zero_infinity=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, int64_t, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: ctc_loss + operator_name: ctc_loss + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::ctc_loss.Tensor(Tensor log_probs, Tensor targets, Tensor input_lengths, Tensor target_lengths, int blank=0, int reduction=Mean, bool zero_infinity=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _ctc_loss + operator_name: _ctc_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_ctc_loss(Tensor log_probs, Tensor targets, int[] input_lengths, int[] target_lengths, int blank=0, bool zero_infinity=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _ctc_loss + operator_name: _ctc_loss + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_ctc_loss.Tensor(Tensor log_probs, Tensor targets, Tensor input_lengths, Tensor target_lengths, int blank=0, bool zero_infinity=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _ctc_loss_backward + operator_name: _ctc_loss_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_ctc_loss_backward(Tensor grad, Tensor log_probs, Tensor targets, int[] input_lengths, int[] target_lengths, Tensor neg_log_likelihood, Tensor log_alpha, int blank, bool zero_infinity=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: neg_log_likelihood + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_alpha + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, const at::Tensor &, const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: neg_log_likelihood + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_alpha + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _ctc_loss_backward + operator_name: _ctc_loss_backward + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_ctc_loss_backward.Tensor(Tensor grad, Tensor log_probs, Tensor targets, Tensor input_lengths, Tensor target_lengths, Tensor neg_log_likelihood, Tensor log_alpha, int blank, bool zero_infinity=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: neg_log_likelihood + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_alpha + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: neg_log_likelihood + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_alpha + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: diag_embed + operator_name: diag_embed + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::diag_embed(Tensor self, int offset=0, int dim1=-2, int dim2=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: -2 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: -2 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: diagflat + operator_name: diagflat + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::diagflat(Tensor self, int offset=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: diagonal + operator_name: diagonal + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::diagonal(Tensor(a) self, int offset=0, int dim1=0, int dim2=1) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_diagonal + operator_name: linalg_diagonal + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_diagonal(Tensor(a) A, *, int offset=0, int dim1=-2, int dim2=-1) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: offset + type: int64_t + - annotation: null + default: -2 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim1 + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim2 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: offset + type: int64_t + - annotation: null + default: -2 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim1 + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim2 + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: diagonal + operator_name: diagonal + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::diagonal.Dimname(Tensor(a) self, *, Dimname outdim, Dimname dim1, Dimname dim2, int offset=0) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + kwarg_only: true + name: outdim + type: at::Dimname + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + kwarg_only: true + name: dim1 + type: at::Dimname + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + kwarg_only: true + name: dim2 + type: at::Dimname + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: offset + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, at::Dimname, at::Dimname, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + kwarg_only: true + name: outdim + type: at::Dimname + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + kwarg_only: true + name: dim1 + type: at::Dimname + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + kwarg_only: true + name: dim2 + type: at::Dimname + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: offset + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: diagonal_backward + operator_name: diagonal_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::diagonal_backward(Tensor grad_output, SymInt[] input_sizes, int offset, int dim1, int dim2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: fill_diagonal_ + operator_name: fill_diagonal_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fill_diagonal_(Tensor(a!) self, Scalar fill_value, bool wrap=False) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: wrap + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: wrap + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: diff + operator_name: diff + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::diff(Tensor self, int n=1, int dim=-1, Tensor? prepend=None, Tensor? append=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: prepend + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: append + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: prepend + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: append + type: const ::std::optional & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: diff_out + operator_name: diff + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::diff.out(Tensor self, int n=1, int dim=-1, Tensor? prepend=None, Tensor? append=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: prepend + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: append + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, const ::std::optional &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: prepend + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: append + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gradient + operator_name: gradient + overload_name: scalarint + manual_kernel_registration: false + category_override: '' + schema_string: aten::gradient.scalarint(Tensor self, *, Scalar? spacing=None, int? dim=None, int edge_order=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: spacing + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dim + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, const ::std::optional &, ::std::optional, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: spacing + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dim + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gradient + operator_name: gradient + overload_name: scalararray + manual_kernel_registration: false + category_override: '' + schema_string: aten::gradient.scalararray(Tensor self, *, Scalar spacing, int[] dim, int edge_order=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: spacing + type: const at::Scalar & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + kwarg_only: true + name: dim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, const at::Scalar &, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: spacing + type: const at::Scalar & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + kwarg_only: true + name: dim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gradient + operator_name: gradient + overload_name: array + manual_kernel_registration: false + category_override: '' + schema_string: aten::gradient.array(Tensor self, *, int[] dim, int edge_order=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + kwarg_only: true + name: dim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + kwarg_only: true + name: dim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gradient + operator_name: gradient + overload_name: scalarrayint + manual_kernel_registration: false + category_override: '' + schema_string: aten::gradient.scalarrayint(Tensor self, *, Scalar[] spacing, int? dim=None, int edge_order=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + kwarg_only: true + name: spacing + type: at::ArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dim + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::ArrayRef, ::std::optional, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + kwarg_only: true + name: spacing + type: at::ArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dim + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gradient + operator_name: gradient + overload_name: scalarrayarray + manual_kernel_registration: false + category_override: '' + schema_string: aten::gradient.scalarrayarray(Tensor self, *, Scalar[] spacing, int[] dim, int edge_order=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + kwarg_only: true + name: spacing + type: at::ArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + kwarg_only: true + name: dim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::ArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + kwarg_only: true + name: spacing + type: at::ArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + kwarg_only: true + name: dim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gradient + operator_name: gradient + overload_name: tensorarrayint + manual_kernel_registration: false + category_override: '' + schema_string: aten::gradient.tensorarrayint(Tensor self, *, Tensor[] spacing, int? dim=None, int edge_order=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + kwarg_only: true + name: spacing + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dim + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::TensorList, ::std::optional, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + kwarg_only: true + name: spacing + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dim + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gradient + operator_name: gradient + overload_name: tensorarray + manual_kernel_registration: false + category_override: '' + schema_string: aten::gradient.tensorarray(Tensor self, *, Tensor[] spacing, int[] dim, int edge_order=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + kwarg_only: true + name: spacing + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + kwarg_only: true + name: dim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::TensorList, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + kwarg_only: true + name: spacing + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + kwarg_only: true + name: dim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: edge_order + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: div + operator_name: div + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::div.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: div_ + operator_name: div_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::div_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: div_out + operator_name: div + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::div.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: div + operator_name: div + overload_name: Tensor_mode + manual_kernel_registration: false + category_override: '' + schema_string: aten::div.Tensor_mode(Tensor self, Tensor other, *, str? rounding_mode) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: div_ + operator_name: div_ + overload_name: Tensor_mode + manual_kernel_registration: false + category_override: '' + schema_string: aten::div_.Tensor_mode(Tensor(a!) self, Tensor other, *, str? rounding_mode) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: div_out + operator_name: div + overload_name: out_mode + manual_kernel_registration: false + category_override: '' + schema_string: aten::div.out_mode(Tensor self, Tensor other, *, str? rounding_mode, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: div + operator_name: div + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::div.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: div_ + operator_name: div_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::div_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: div + operator_name: div + overload_name: Scalar_mode + manual_kernel_registration: false + category_override: '' + schema_string: aten::div.Scalar_mode(Tensor self, Scalar other, *, str? rounding_mode) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: div_ + operator_name: div_ + overload_name: Scalar_mode + manual_kernel_registration: false + category_override: '' + schema_string: aten::div_.Scalar_mode(Tensor(a!) self, Scalar other, *, str? rounding_mode) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: divide + operator_name: divide + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::divide.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: divide_ + operator_name: divide_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::divide_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: divide_out + operator_name: divide + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::divide.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: divide + operator_name: divide + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::divide.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: divide_ + operator_name: divide_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::divide_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: divide + operator_name: divide + overload_name: Tensor_mode + manual_kernel_registration: false + category_override: '' + schema_string: aten::divide.Tensor_mode(Tensor self, Tensor other, *, str? rounding_mode) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: divide_ + operator_name: divide_ + overload_name: Tensor_mode + manual_kernel_registration: false + category_override: '' + schema_string: aten::divide_.Tensor_mode(Tensor(a!) self, Tensor other, *, str? rounding_mode) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: divide_out + operator_name: divide + overload_name: out_mode + manual_kernel_registration: false + category_override: '' + schema_string: aten::divide.out_mode(Tensor self, Tensor other, *, str? rounding_mode, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: divide + operator_name: divide + overload_name: Scalar_mode + manual_kernel_registration: false + category_override: '' + schema_string: aten::divide.Scalar_mode(Tensor self, Scalar other, *, str? rounding_mode) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: divide_ + operator_name: divide_ + overload_name: Scalar_mode + manual_kernel_registration: false + category_override: '' + schema_string: aten::divide_.Scalar_mode(Tensor(a!) self, Scalar other, *, str? rounding_mode) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: true_divide + operator_name: true_divide + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::true_divide.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: true_divide_ + operator_name: true_divide_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::true_divide_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: true_divide_out + operator_name: true_divide + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::true_divide.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: true_divide + operator_name: true_divide + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::true_divide.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: true_divide_ + operator_name: true_divide_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::true_divide_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: dot + operator_name: dot + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::dot(Tensor self, Tensor tensor) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: dot_out + operator_name: dot + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::dot.out(Tensor self, Tensor tensor, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: vdot + operator_name: vdot + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::vdot(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: vdot_out + operator_name: vdot + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::vdot.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: einsum + operator_name: einsum + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::einsum(str equation, Tensor[] tensors, *, int[]? path=None) -> Tensor + arguments: + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: equation + type: c10::string_view + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + kwarg_only: true + name: path + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor (c10::string_view, at::TensorList, at::OptionalIntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: equation + type: c10::string_view + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + kwarg_only: true + name: path + type: at::OptionalIntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: embedding + operator_name: embedding + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding(Tensor weight, Tensor indices, SymInt padding_idx=-1, bool scale_grad_by_freq=False, bool sparse=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: embedding_backward + operator_name: embedding_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding_backward(Tensor grad, Tensor indices, SymInt num_weights, SymInt padding_idx, bool scale_grad_by_freq, bool sparse) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: embedding_dense_backward + operator_name: embedding_dense_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding_dense_backward(Tensor grad_output, Tensor indices, SymInt num_weights, SymInt padding_idx, bool scale_grad_by_freq) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: embedding_renorm_ + operator_name: embedding_renorm_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding_renorm_(Tensor(a!) self, Tensor indices, float max_norm, float norm_type) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: max_norm + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: norm_type + type: double + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, double, double) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: max_norm + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: norm_type + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: embedding_sparse_backward + operator_name: embedding_sparse_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding_sparse_backward(Tensor grad, Tensor indices, int num_weights, int padding_idx, bool scale_grad_by_freq) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _embedding_bag_forward_only + operator_name: _embedding_bag_forward_only + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_embedding_bag_forward_only(Tensor weight, Tensor indices, Tensor offsets, bool scale_grad_by_freq=False, int mode=0, bool sparse=False, Tensor? per_sample_weights=None, bool include_last_offset=False, int padding_idx=-1) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, int64_t, bool, const ::std::optional &, bool, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _rowwise_prune + operator_name: _rowwise_prune + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_rowwise_prune(Tensor weight, Tensor mask, ScalarType compressed_indices_dtype) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: compressed_indices_dtype + type: at::ScalarType + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: compressed_indices_dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: row_stack + operator_name: row_stack + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::row_stack(Tensor[] tensors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: row_stack_out + operator_name: row_stack + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::row_stack.out(Tensor[] tensors, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: embedding_bag + operator_name: embedding_bag + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding_bag(Tensor weight, Tensor indices, Tensor offsets, bool scale_grad_by_freq=False, int mode=0, bool sparse=False, Tensor? per_sample_weights=None, bool include_last_offset=False) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, int64_t, bool, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: embedding_bag + operator_name: embedding_bag + overload_name: padding_idx + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding_bag.padding_idx(Tensor weight, Tensor indices, Tensor offsets, bool scale_grad_by_freq, int mode, bool sparse, Tensor? per_sample_weights, bool include_last_offset, int? padding_idx) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: padding_idx + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, int64_t, bool, const ::std::optional &, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: padding_idx + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _embedding_bag + operator_name: _embedding_bag + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_embedding_bag(Tensor weight, Tensor indices, Tensor offsets, bool scale_grad_by_freq=False, int mode=0, bool sparse=False, Tensor? per_sample_weights=None, bool include_last_offset=False, int padding_idx=-1) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, int64_t, bool, const ::std::optional &, bool, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _embedding_bag_backward + operator_name: _embedding_bag_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_embedding_bag_backward(Tensor grad, Tensor indices, Tensor offsets, Tensor offset2bag, Tensor bag_size, Tensor maximum_indices, SymInt num_weights, bool scale_grad_by_freq, int mode, bool sparse, Tensor? per_sample_weights, int padding_idx=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bag_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: maximum_indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, bool, int64_t, bool, const ::std::optional &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bag_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: maximum_indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _embedding_bag_sparse_backward + operator_name: _embedding_bag_sparse_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_embedding_bag_sparse_backward(Tensor grad, Tensor indices, Tensor offsets, Tensor offset2bag, Tensor bag_size, SymInt num_weights, bool scale_grad_by_freq, int mode, Tensor? per_sample_weights, int padding_idx=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bag_size + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, bool, int64_t, const ::std::optional &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bag_size + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _embedding_bag_dense_backward + operator_name: _embedding_bag_dense_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_embedding_bag_dense_backward(Tensor grad, Tensor indices, Tensor offset2bag, Tensor bag_size, Tensor maximum_indices, SymInt num_weights, bool scale_grad_by_freq, int mode, Tensor? per_sample_weights, int padding_idx=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bag_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: maximum_indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, bool, int64_t, const ::std::optional &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bag_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: maximum_indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _embedding_bag_per_sample_weights_backward + operator_name: _embedding_bag_per_sample_weights_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_embedding_bag_per_sample_weights_backward(Tensor grad, Tensor weight, Tensor indices, Tensor offsets, Tensor offset2bag, int mode, int padding_idx=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty + operator_name: empty + overload_name: names + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty.names(int[] size, *, Dimname[]? names, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty + operator_name: empty + overload_name: memory_format + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty.memory_format(SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty_permuted + operator_name: empty_permuted + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty_permuted(SymInt[] size, int[] physical_layout, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: physical_layout + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: physical_layout + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: new_empty + operator_name: new_empty + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::new_empty(Tensor self, SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: new_empty_strided + operator_name: new_empty_strided + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::new_empty_strided(Tensor self, SymInt[] size, SymInt[] stride, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: new_full + operator_name: new_full + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::new_full(Tensor self, SymInt[] size, Scalar fill_value, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, const at::Scalar &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: new_zeros + operator_name: new_zeros + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::new_zeros(Tensor self, SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: new_ones + operator_name: new_ones + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::new_ones(Tensor self, SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _empty_affine_quantized + operator_name: _empty_affine_quantized + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_empty_affine_quantized(SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, float scale=1, int zero_point=0, MemoryFormat? memory_format=contiguous_format) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + kwarg_only: true + name: scale + type: double + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: zero_point + type: int64_t + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, double, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + kwarg_only: true + name: scale + type: double + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: zero_point + type: int64_t + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _empty_per_channel_affine_quantized + operator_name: _empty_per_channel_affine_quantized + overload_name: '' + manual_kernel_registration: false + category_override: factory + schema_string: aten::_empty_per_channel_affine_quantized(SymInt[] size, *, Tensor scales, Tensor zero_points, int axis, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=contiguous_format) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, const at::Tensor &, const at::Tensor &, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: resize_ + operator_name: resize_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::resize_(Tensor(a!) self, SymInt[] size, *, MemoryFormat? memory_format=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: const at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _resize_output_ + operator_name: _resize_output_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_resize_output_(Tensor(a!) self, SymInt[] size, Device device) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Device + is_nullable: false + name: device + type: at::Device + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Device) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Device + is_nullable: false + name: device + type: at::Device + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: const at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty_quantized + operator_name: empty_quantized + overload_name: '' + manual_kernel_registration: false + category_override: factory + schema_string: aten::empty_quantized(int[] size, Tensor qtensor, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qtensor + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qtensor + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty_out + operator_name: empty + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty.out(SymInt[] size, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: empty_like + operator_name: empty_like + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty_like(Tensor self, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty_strided + operator_name: empty_strided + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty_strided(SymInt[] size, SymInt[] stride, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: erf + operator_name: erf + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::erf(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: erf_ + operator_name: erf_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::erf_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: erf_out + operator_name: erf + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::erf.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: erfc + operator_name: erfc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::erfc(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: erfc_ + operator_name: erfc_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::erfc_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: erfc_out + operator_name: erfc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::erfc.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: exp + operator_name: exp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::exp(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: exp_ + operator_name: exp_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::exp_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: exp_out + operator_name: exp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::exp.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: exp2 + operator_name: exp2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::exp2(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: exp2_ + operator_name: exp2_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::exp2_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: exp2_out + operator_name: exp2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::exp2.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: expm1 + operator_name: expm1 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::expm1(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: expm1_ + operator_name: expm1_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::expm1_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: expm1_out + operator_name: expm1 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::expm1.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: expand + operator_name: expand + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::expand(Tensor(a) self, SymInt[] size, *, bool implicit=False) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: implicit + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: implicit + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: expand_as + operator_name: expand_as + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::expand_as(Tensor(a) self, Tensor other) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: eye + operator_name: eye + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::eye(SymInt n, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: eye + operator_name: eye + overload_name: m + manual_kernel_registration: false + category_override: '' + schema_string: aten::eye.m(SymInt n, SymInt m, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: m + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: m + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: eye_out + operator_name: eye + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::eye.out(SymInt n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + schema_order_cpp_signature: at::Tensor & (int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: eye_out + operator_name: eye + overload_name: m_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::eye.m_out(SymInt n, SymInt m, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: m + type: int64_t + schema_order_cpp_signature: at::Tensor & (int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: m + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: flatten + operator_name: flatten + overload_name: using_ints + manual_kernel_registration: false + category_override: '' + schema_string: aten::flatten.using_ints(Tensor(a) self, int start_dim=0, int end_dim=-1) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: start_dim + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: end_dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: start_dim + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: end_dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: flatten + operator_name: flatten + overload_name: named_out_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::flatten.named_out_dim(Tensor(a) self, int start_dim, int end_dim, Dimname out_dim) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: end_dim + type: int64_t + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: out_dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, at::Dimname) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: end_dim + type: int64_t + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: out_dim + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: flatten + operator_name: flatten + overload_name: using_names + manual_kernel_registration: false + category_override: '' + schema_string: aten::flatten.using_names(Tensor(a) self, Dimname start_dim, Dimname end_dim, Dimname out_dim) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: start_dim + type: at::Dimname + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: end_dim + type: at::Dimname + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: out_dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, at::Dimname, at::Dimname) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: start_dim + type: at::Dimname + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: end_dim + type: at::Dimname + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: out_dim + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: flatten + operator_name: flatten + overload_name: DimnameList + manual_kernel_registration: false + category_override: '' + schema_string: aten::flatten.DimnameList(Tensor(a) self, Dimname[] dims, Dimname out_dim) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dims + type: at::DimnameList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: out_dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList, at::Dimname) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dims + type: at::DimnameList + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: out_dim + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: unflatten + operator_name: unflatten + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::unflatten.int(Tensor(a) self, int dim, SymInt[] sizes) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sizes + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, at::IntArrayRef) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sizes + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: unflatten + operator_name: unflatten + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::unflatten.Dimname(Tensor(a) self, Dimname dim, SymInt[] sizes, Dimname[] names) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: names + type: at::DimnameList + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, at::IntArrayRef, at::DimnameList) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: names + type: at::DimnameList + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fill + operator_name: fill + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::fill.Scalar(Tensor self, Scalar value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fill + operator_name: fill + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::fill.Tensor(Tensor self, Tensor value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fill_ + operator_name: fill_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::fill_.Scalar(Tensor(a!) self, Scalar value) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fill_ + operator_name: fill_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::fill_.Tensor(Tensor(a!) self, Tensor value) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: floor + operator_name: floor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::floor(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: floor_ + operator_name: floor_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::floor_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: floor_out + operator_name: floor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::floor.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: floor_divide + operator_name: floor_divide + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::floor_divide(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: floor_divide_ + operator_name: floor_divide_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::floor_divide_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: floor_divide_out + operator_name: floor_divide + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::floor_divide.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: floor_divide + operator_name: floor_divide + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::floor_divide.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: floor_divide_ + operator_name: floor_divide_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::floor_divide_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: frac + operator_name: frac + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::frac(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: frac_ + operator_name: frac_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::frac_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: frac_out + operator_name: frac + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::frac.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: full + operator_name: full + overload_name: names + manual_kernel_registration: false + category_override: '' + schema_string: aten::full.names(int[] size, Scalar fill_value, *, Dimname[]? names, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, const at::Scalar &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: full + operator_name: full + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::full(SymInt[] size, Scalar fill_value, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, const at::Scalar &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: full_out + operator_name: full + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::full.out(SymInt[] size, Scalar fill_value, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: full_like + operator_name: full_like + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::full_like(Tensor self, Scalar fill_value, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: from_file + operator_name: from_file + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::from_file(str filename, bool? shared=None, int? size=0, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: filename + type: c10::string_view + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: shared + type: ::std::optional + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: true + name: size + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (c10::string_view, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: filename + type: c10::string_view + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: shared + type: ::std::optional + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: true + name: size + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gcd_out + operator_name: gcd + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::gcd.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gcd + operator_name: gcd + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::gcd(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gcd_ + operator_name: gcd_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::gcd_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lcm_out + operator_name: lcm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lcm.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lcm + operator_name: lcm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lcm(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lcm_ + operator_name: lcm_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lcm_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: grid_sampler + operator_name: grid_sampler + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::grid_sampler(Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: grid_sampler_2d + operator_name: grid_sampler_2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::grid_sampler_2d(Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: grid_sampler_2d_backward + operator_name: grid_sampler_2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::grid_sampler_2d_backward(Tensor grad_output, Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners, bool[2] output_mask) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _grid_sampler_2d_cpu_fallback + operator_name: _grid_sampler_2d_cpu_fallback + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_grid_sampler_2d_cpu_fallback(Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _grid_sampler_2d_cpu_fallback_backward + operator_name: _grid_sampler_2d_cpu_fallback_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_grid_sampler_2d_cpu_fallback_backward(Tensor grad_output, Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: grid_sampler_3d + operator_name: grid_sampler_3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::grid_sampler_3d(Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: grid_sampler_3d_backward + operator_name: grid_sampler_3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::grid_sampler_3d_backward(Tensor grad_output, Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners, bool[2] output_mask) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hann_window + operator_name: hann_window + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hann_window(int window_length, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hann_window + operator_name: hann_window + overload_name: periodic + manual_kernel_registration: false + category_override: '' + schema_string: aten::hann_window.periodic(int window_length, bool periodic, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, bool, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hamming_window + operator_name: hamming_window + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hamming_window(int window_length, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hamming_window + operator_name: hamming_window + overload_name: periodic + manual_kernel_registration: false + category_override: '' + schema_string: aten::hamming_window.periodic(int window_length, bool periodic, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, bool, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hamming_window + operator_name: hamming_window + overload_name: periodic_alpha + manual_kernel_registration: false + category_override: '' + schema_string: aten::hamming_window.periodic_alpha(int window_length, bool periodic, float alpha, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: alpha + type: double + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, bool, double, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: alpha + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hamming_window + operator_name: hamming_window + overload_name: periodic_alpha_beta + manual_kernel_registration: false + category_override: '' + schema_string: aten::hamming_window.periodic_alpha_beta(int window_length, bool periodic, float alpha, float beta, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: alpha + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, bool, double, double, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: alpha + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: kaiser_window + operator_name: kaiser_window + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::kaiser_window(int window_length, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: kaiser_window + operator_name: kaiser_window + overload_name: periodic + manual_kernel_registration: false + category_override: '' + schema_string: aten::kaiser_window.periodic(int window_length, bool periodic, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, bool, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: kaiser_window + operator_name: kaiser_window + overload_name: beta + manual_kernel_registration: false + category_override: '' + schema_string: aten::kaiser_window.beta(int window_length, bool periodic, float beta, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, bool, double, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hinge_embedding_loss + operator_name: hinge_embedding_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hinge_embedding_loss(Tensor self, Tensor target, float margin=1.0, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: margin + type: double + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, double, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: margin + type: double + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: group_norm + operator_name: group_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::group_norm(Tensor input, int num_groups, Tensor? weight=None, Tensor? bias=None, float eps=1e-05, bool cudnn_enabled=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_groups + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1.0e-05 + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const ::std::optional &, const ::std::optional &, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_groups + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1.0e-05 + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: native_group_norm + operator_name: native_group_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_group_norm(Tensor input, Tensor? weight, Tensor? bias, SymInt N, SymInt C, SymInt HxW, int group, float eps) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: HxW + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: group + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, int64_t, int64_t, int64_t, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: HxW + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: group + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_group_norm_backward + operator_name: native_group_norm_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_group_norm_backward(Tensor grad_out, Tensor input, Tensor mean, Tensor rstd, Tensor? weight, SymInt N, SymInt C, SymInt HxW, int group, bool[3] output_mask) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: HxW + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: group + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, int64_t, int64_t, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: HxW + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: group + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fft_r2c + operator_name: _fft_r2c + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fft_r2c(Tensor self, int[] dim, int normalization, bool onesided) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: onesided + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: onesided + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fft_r2c_out + operator_name: _fft_r2c + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fft_r2c.out(Tensor self, int[] dim, int normalization, bool onesided, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: onesided + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: onesided + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fft_c2r + operator_name: _fft_c2r + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fft_c2r(Tensor self, int[] dim, int normalization, SymInt last_dim_size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: last_dim_size + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: last_dim_size + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fft_c2r_out + operator_name: _fft_c2r + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fft_c2r.out(Tensor self, int[] dim, int normalization, SymInt last_dim_size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: last_dim_size + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: last_dim_size + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fft_c2c + operator_name: _fft_c2c + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fft_c2c(Tensor self, SymInt[] dim, int normalization, bool forward) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: forward + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: forward + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fft_c2c_out + operator_name: _fft_c2c + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fft_c2c.out(Tensor self, SymInt[] dim, int normalization, bool forward, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: forward + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: normalization + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: forward + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _validate_compressed_sparse_indices + operator_name: _validate_compressed_sparse_indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_validate_compressed_sparse_indices(bool is_crow, Tensor compressed_idx, Tensor plain_idx, int cdim, int dim, int nnz) -> () + arguments: + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_crow + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_idx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: plain_idx + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: cdim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: nnz + type: int64_t + schema_order_cpp_signature: void (bool, const at::Tensor &, const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_crow + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_idx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: plain_idx + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: cdim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: nnz + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cufft_get_plan_cache_size + operator_name: _cufft_get_plan_cache_size + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cufft_get_plan_cache_size(DeviceIndex device_index) -> int + arguments: + - annotation: null + dynamic_type: at::DeviceIndex + is_nullable: false + name: device_index + type: at::DeviceIndex + schema_order_cpp_signature: int64_t (at::DeviceIndex) + schema_order_arguments: + - annotation: null + dynamic_type: at::DeviceIndex + is_nullable: false + name: device_index + type: at::DeviceIndex + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cufft_get_plan_cache_max_size + operator_name: _cufft_get_plan_cache_max_size + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cufft_get_plan_cache_max_size(DeviceIndex device_index) -> int + arguments: + - annotation: null + dynamic_type: at::DeviceIndex + is_nullable: false + name: device_index + type: at::DeviceIndex + schema_order_cpp_signature: int64_t (at::DeviceIndex) + schema_order_arguments: + - annotation: null + dynamic_type: at::DeviceIndex + is_nullable: false + name: device_index + type: at::DeviceIndex + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cufft_set_plan_cache_max_size + operator_name: _cufft_set_plan_cache_max_size + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cufft_set_plan_cache_max_size(DeviceIndex device_index, int max_size) -> () + arguments: + - annotation: null + dynamic_type: at::DeviceIndex + is_nullable: false + name: device_index + type: at::DeviceIndex + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_size + type: int64_t + schema_order_cpp_signature: void (at::DeviceIndex, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::DeviceIndex + is_nullable: false + name: device_index + type: at::DeviceIndex + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_size + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _cufft_clear_plan_cache + operator_name: _cufft_clear_plan_cache + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cufft_clear_plan_cache(DeviceIndex device_index) -> () + arguments: + - annotation: null + dynamic_type: at::DeviceIndex + is_nullable: false + name: device_index + type: at::DeviceIndex + schema_order_cpp_signature: void (at::DeviceIndex) + schema_order_arguments: + - annotation: null + dynamic_type: at::DeviceIndex + is_nullable: false + name: device_index + type: at::DeviceIndex + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: index + operator_name: index + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::index.Tensor(Tensor self, Tensor?[] indices) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const c10::List<::std::optional> &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_out + operator_name: index + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::index.Tensor_out(Tensor self, Tensor?[] indices, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const c10::List<::std::optional> &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _unsafe_index + operator_name: _unsafe_index + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unsafe_index.Tensor(Tensor self, Tensor?[] indices) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const c10::List<::std::optional> &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _unsafe_masked_index + operator_name: _unsafe_masked_index + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unsafe_masked_index(Tensor self, Tensor mask, Tensor?[] indices, Scalar fill) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const c10::List<::std::optional> &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _unsafe_masked_index_put_accumulate + operator_name: _unsafe_masked_index_put_accumulate + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unsafe_masked_index_put_accumulate(Tensor self, Tensor mask, Tensor?[] indices, Tensor values) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const c10::List<::std::optional> &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_copy_out + operator_name: index_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_copy.out(Tensor self, int dim, Tensor index, Tensor source, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_copy_ + operator_name: index_copy_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_copy_(Tensor(a!) self, int dim, Tensor index, Tensor source) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_copy + operator_name: index_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_copy(Tensor self, int dim, Tensor index, Tensor source) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_copy_ + operator_name: index_copy_ + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_copy_.dimname(Tensor(a!) self, Dimname dim, Tensor index, Tensor source) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::Dimname, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: index_copy + operator_name: index_copy + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_copy.dimname(Tensor self, Dimname dim, Tensor index, Tensor source) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: index_put_ + operator_name: index_put_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_put_(Tensor(a!) self, Tensor?[] indices, Tensor values, bool accumulate=False) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const c10::List<::std::optional> &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_put + operator_name: index_put + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_put(Tensor self, Tensor?[] indices, Tensor values, bool accumulate=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const c10::List<::std::optional> &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _unsafe_index_put + operator_name: _unsafe_index_put + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unsafe_index_put(Tensor self, Tensor?[] indices, Tensor values, bool accumulate=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const c10::List<::std::optional> &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _index_put_impl_ + operator_name: _index_put_impl_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_index_put_impl_(Tensor(a!) self, Tensor?[] indices, Tensor values, bool accumulate=False, bool unsafe=False) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: unsafe + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const c10::List<::std::optional> &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: unsafe + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: instance_norm + operator_name: instance_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::instance_norm(Tensor input, Tensor? weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, bool use_input_stats, float momentum, float eps, bool cudnn_enabled) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: use_input_stats + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: use_input_stats + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: isclose + operator_name: isclose + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::isclose(Tensor self, Tensor other, float rtol=1e-05, float atol=1e-08, bool equal_nan=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1.0e-05 + dynamic_type: double + is_nullable: false + name: rtol + type: double + - annotation: null + default: 1.0e-08 + dynamic_type: double + is_nullable: false + name: atol + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: equal_nan + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, double, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1.0e-05 + dynamic_type: double + is_nullable: false + name: rtol + type: double + - annotation: null + default: 1.0e-08 + dynamic_type: double + is_nullable: false + name: atol + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: equal_nan + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: isin_out + operator_name: isin + overload_name: Tensor_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::isin.Tensor_Tensor_out(Tensor elements, Tensor test_elements, *, bool assume_unique=False, bool invert=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: elements + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: test_elements + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: elements + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: test_elements + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isin + operator_name: isin + overload_name: Tensor_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::isin.Tensor_Tensor(Tensor elements, Tensor test_elements, *, bool assume_unique=False, bool invert=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: elements + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: test_elements + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: elements + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: test_elements + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isin_out + operator_name: isin + overload_name: Tensor_Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::isin.Tensor_Scalar_out(Tensor elements, Scalar test_element, *, bool assume_unique=False, bool invert=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: elements + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: test_element + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: elements + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: test_element + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isin + operator_name: isin + overload_name: Tensor_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::isin.Tensor_Scalar(Tensor elements, Scalar test_element, *, bool assume_unique=False, bool invert=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: elements + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: test_element + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: elements + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: test_element + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isin_out + operator_name: isin + overload_name: Scalar_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::isin.Scalar_Tensor_out(Scalar element, Tensor test_elements, *, bool assume_unique=False, bool invert=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: element + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: test_elements + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: element + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: test_elements + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isin + operator_name: isin + overload_name: Scalar_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::isin.Scalar_Tensor(Scalar element, Tensor test_elements, *, bool assume_unique=False, bool invert=False) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: element + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: test_elements + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: element + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: test_elements + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: assume_unique + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: invert + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isnan + operator_name: isnan + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::isnan(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: is_distributed + operator_name: is_distributed + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_distributed(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: is_floating_point + operator_name: is_floating_point + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_floating_point(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: is_complex + operator_name: is_complex + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_complex(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: is_conj + operator_name: is_conj + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_conj(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: _is_zerotensor + operator_name: _is_zerotensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_is_zerotensor(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: is_neg + operator_name: is_neg + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_neg(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: isreal + operator_name: isreal + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::isreal(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: is_nonzero + operator_name: is_nonzero + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_nonzero(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: is_same_size + operator_name: is_same_size + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_same_size(Tensor self, Tensor other) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: is_signed + operator_name: is_signed + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_signed(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: is_inference + operator_name: is_inference + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_inference(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: kl_div + operator_name: kl_div + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::kl_div(Tensor self, Tensor target, int reduction=Mean, *, bool log_target=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: log_target + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: log_target + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: kron + operator_name: kron + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::kron(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: kron_out + operator_name: kron + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::kron.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: kthvalue + operator_name: kthvalue + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::kthvalue(Tensor self, SymInt k, int dim=-1, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: kthvalue_out + operator_name: kthvalue + overload_name: values + manual_kernel_registration: false + category_override: '' + schema_string: aten::kthvalue.values(Tensor self, SymInt k, int dim=-1, bool keepdim=False, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: kthvalue + operator_name: kthvalue + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::kthvalue.dimname(Tensor self, SymInt k, Dimname dim, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: kthvalue_out + operator_name: kthvalue + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::kthvalue.dimname_out(Tensor self, SymInt k, Dimname dim, bool keepdim=False, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, at::Dimname, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: layer_norm + operator_name: layer_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::layer_norm(Tensor input, SymInt[] normalized_shape, Tensor? weight=None, Tensor? bias=None, float eps=1e-05, bool cudnn_enable=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1.0e-05 + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: cudnn_enable + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, const ::std::optional &, const ::std::optional &, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1.0e-05 + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: cudnn_enable + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: native_layer_norm + operator_name: native_layer_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_layer_norm(Tensor input, SymInt[] normalized_shape, Tensor? weight, Tensor? bias, float eps) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, const ::std::optional &, const ::std::optional &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_layer_norm_backward + operator_name: native_layer_norm_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_layer_norm_backward(Tensor grad_out, Tensor input, SymInt[] normalized_shape, Tensor mean, Tensor rstd, Tensor? weight, Tensor? bias, bool[3] output_mask) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rms_norm + operator_name: rms_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rms_norm(Tensor input, SymInt[] normalized_shape, Tensor? weight=None, float? eps=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, const ::std::optional &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _fused_rms_norm + operator_name: _fused_rms_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_rms_norm(Tensor input, int[] normalized_shape, Tensor? weight, float? eps) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, const ::std::optional &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _fused_rms_norm_backward + operator_name: _fused_rms_norm_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_rms_norm_backward(Tensor grad_out, Tensor input, int[] normalized_shape, Tensor rstd, Tensor? weight, bool[2] output_mask) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const at::Tensor &, const ::std::optional &, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nan_to_num + operator_name: nan_to_num + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nan_to_num(Tensor self, float? nan=None, float? posinf=None, float? neginf=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: nan + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: posinf + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: neginf + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: nan + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: posinf + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: neginf + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nan_to_num_ + operator_name: nan_to_num_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nan_to_num_(Tensor(a!) self, float? nan=None, float? posinf=None, float? neginf=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: nan + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: posinf + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: neginf + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: nan + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: posinf + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: neginf + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nan_to_num_out + operator_name: nan_to_num + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nan_to_num.out(Tensor self, float? nan=None, float? posinf=None, float? neginf=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: nan + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: posinf + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: neginf + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: nan + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: posinf + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: neginf + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linear + operator_name: linear + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linear(Tensor input, Tensor weight, Tensor? bias=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linear_backward + operator_name: linear_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linear_backward(Tensor self, Tensor grad_output, Tensor weight, bool[3] output_mask) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linear_out + operator_name: linear + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linear.out(Tensor input, Tensor weight, Tensor? bias=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_linear + operator_name: mkldnn_linear + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_linear(Tensor self, Tensor weight, Tensor? bias=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_linear_backward_input + operator_name: mkldnn_linear_backward_input + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_linear_backward_input(int[] input_size, Tensor grad_output, Tensor weight) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_linear_backward_weights + operator_name: mkldnn_linear_backward_weights + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_linear_backward_weights(Tensor grad_output, Tensor input, Tensor weight, bool bias_defined) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: bias_defined + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: bias_defined + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_linear_backward + operator_name: mkldnn_linear_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_linear_backward(Tensor self, Tensor grad_output, Tensor weight, bool[3] output_mask) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cslt_compress + operator_name: _cslt_compress + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cslt_compress(Tensor input) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cslt_sparse_mm + operator_name: _cslt_sparse_mm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cslt_sparse_mm(Tensor compressed_A, Tensor dense_B, Tensor? bias=None, Tensor? alpha=None, ScalarType? out_dtype=None, bool transpose_result=False, int alg_id=0, int split_k=1, int split_k_mode=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense_B + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: alpha + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose_result + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: alg_id + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: split_k + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: split_k_mode + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, ::std::optional, bool, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense_B + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: alpha + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose_result + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: alg_id + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: split_k + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: split_k_mode + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cslt_sparse_mm_search + operator_name: _cslt_sparse_mm_search + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cslt_sparse_mm_search(Tensor compressed_A, Tensor dense_B, Tensor? bias=None, Tensor? alpha=None, ScalarType? out_dtype=None, bool transpose_result=False) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense_B + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: alpha + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose_result + type: bool + schema_order_cpp_signature: int64_t (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense_B + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: alpha + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose_result + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_semi_structured_tile + operator_name: _sparse_semi_structured_tile + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_semi_structured_tile(Tensor input, str algorithm="", bool use_cutlass=True) -> (Tensor, Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + default: '""' + dynamic_type: c10::string_view + is_nullable: false + name: algorithm + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: use_cutlass + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, c10::string_view, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + default: '""' + dynamic_type: c10::string_view + is_nullable: false + name: algorithm + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: use_cutlass + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: at::Tensor + name: result4 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_semi_structured_apply + operator_name: _sparse_semi_structured_apply + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_semi_structured_apply(Tensor input, Tensor thread_masks) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: thread_masks + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: thread_masks + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_semi_structured_apply_dense + operator_name: _sparse_semi_structured_apply_dense + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_semi_structured_apply_dense(Tensor input, Tensor thread_masks) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: thread_masks + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: thread_masks + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_semi_structured_linear + operator_name: _sparse_semi_structured_linear + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_semi_structured_linear(Tensor input, Tensor weight, Tensor meta, *, Tensor? bias=None, str? activation=None, ScalarType? out_dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: meta + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: bias + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: activation + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: out_dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: meta + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: bias + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: activation + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: out_dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_semi_structured_mm + operator_name: _sparse_semi_structured_mm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_semi_structured_mm(Tensor mat1, Tensor mat1_meta, Tensor mat2, *, ScalarType? out_dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1_meta + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: out_dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1_meta + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: out_dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_semi_structured_addmm + operator_name: _sparse_semi_structured_addmm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_semi_structured_addmm(Tensor input, Tensor mat1, Tensor mat1_meta, Tensor mat2, *, Scalar alpha=1, Scalar beta=1, ScalarType? out_dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1_meta + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: out_dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1_meta + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: out_dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _mixed_dtypes_linear + operator_name: _mixed_dtypes_linear + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_mixed_dtypes_linear(Tensor input, Tensor weight, Tensor scale, *, Tensor? bias=None, str? activation=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: bias + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: activation + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: bias + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: activation + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fbgemm_linear_int8_weight_fp32_activation + operator_name: fbgemm_linear_int8_weight_fp32_activation + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fbgemm_linear_int8_weight_fp32_activation(Tensor input, Tensor weight, Tensor packed, Tensor col_offsets, Scalar weight_scale, Scalar weight_zero_point, Tensor bias) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight_scale + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight_zero_point + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight_scale + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight_zero_point + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fbgemm_linear_int8_weight + operator_name: fbgemm_linear_int8_weight + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fbgemm_linear_int8_weight(Tensor input, Tensor weight, Tensor packed, Tensor col_offsets, Scalar weight_scale, Scalar weight_zero_point, Tensor bias) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight_scale + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight_zero_point + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight_scale + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight_zero_point + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fbgemm_linear_quantize_weight + operator_name: fbgemm_linear_quantize_weight + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fbgemm_linear_quantize_weight(Tensor input) -> (Tensor, Tensor, float, int) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: double + name: result2 + type: double + - dynamic_type: int64_t + name: result3 + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fbgemm_pack_gemm_matrix_fp16 + operator_name: fbgemm_pack_gemm_matrix_fp16 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fbgemm_pack_gemm_matrix_fp16(Tensor input) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _wrapped_linear_prepack + operator_name: _wrapped_linear_prepack + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_wrapped_linear_prepack(Tensor weight, Tensor weight_scale, Tensor weight_zero_point, Tensor bias) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _wrapped_quantized_linear_prepacked + operator_name: _wrapped_quantized_linear_prepacked + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_wrapped_quantized_linear_prepacked(Tensor input, Tensor input_scale, Tensor input_zero_point, Tensor packed_weight, Tensor output_scale, Tensor output_zero_point, int out_channel) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output_scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output_zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: out_channel + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output_scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output_zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: out_channel + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fbgemm_linear_fp16_weight_fp32_activation + operator_name: fbgemm_linear_fp16_weight_fp32_activation + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fbgemm_linear_fp16_weight_fp32_activation(Tensor input, Tensor packed_weight, Tensor? bias) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fbgemm_linear_fp16_weight_fp32_activation + operator_name: fbgemm_linear_fp16_weight_fp32_activation + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fbgemm_linear_fp16_weight_fp32_activation.out(Tensor input, Tensor packed_weight, Tensor? bias, Tensor(a!) output) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + type: at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fbgemm_linear_fp16_weight + operator_name: fbgemm_linear_fp16_weight + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fbgemm_linear_fp16_weight(Tensor input, Tensor packed_weight, Tensor bias) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fbgemm_linear_fp16_weight + operator_name: fbgemm_linear_fp16_weight + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fbgemm_linear_fp16_weight.out(Tensor input, Tensor packed_weight, Tensor bias, Tensor(a!) output) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + type: at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fbgemm_pack_quantized_matrix + operator_name: fbgemm_pack_quantized_matrix + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fbgemm_pack_quantized_matrix(Tensor input) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fbgemm_pack_quantized_matrix + operator_name: fbgemm_pack_quantized_matrix + overload_name: KN + manual_kernel_registration: false + category_override: '' + schema_string: aten::fbgemm_pack_quantized_matrix.KN(Tensor input, int K, int N) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: K + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: K + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: ldexp + operator_name: ldexp + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::ldexp.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: ldexp_ + operator_name: ldexp_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::ldexp_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: ldexp_out + operator_name: ldexp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ldexp.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linspace + operator_name: linspace + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linspace(Scalar start, Scalar end, int steps, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Scalar &, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linspace + operator_name: linspace + overload_name: Tensor_Tensor + manual_kernel_registration: false + category_override: factory + schema_string: aten::linspace.Tensor_Tensor(Tensor start, Tensor end, int steps, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linspace + operator_name: linspace + overload_name: Tensor_Scalar + manual_kernel_registration: false + category_override: factory + schema_string: aten::linspace.Tensor_Scalar(Tensor start, Scalar end, int steps, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linspace + operator_name: linspace + overload_name: Scalar_Tensor + manual_kernel_registration: false + category_override: factory + schema_string: aten::linspace.Scalar_Tensor(Scalar start, Tensor end, int steps, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linspace_out + operator_name: linspace + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linspace.out(Scalar start, Scalar end, int steps, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Scalar &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linspace_out + operator_name: linspace + overload_name: Tensor_Tensor_out + manual_kernel_registration: false + category_override: factory + schema_string: aten::linspace.Tensor_Tensor_out(Tensor start, Tensor end, int steps, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linspace_out + operator_name: linspace + overload_name: Tensor_Scalar_out + manual_kernel_registration: false + category_override: factory + schema_string: aten::linspace.Tensor_Scalar_out(Tensor start, Scalar end, int steps, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linspace_out + operator_name: linspace + overload_name: Scalar_Tensor_out + manual_kernel_registration: false + category_override: factory + schema_string: aten::linspace.Scalar_Tensor_out(Scalar start, Tensor end, int steps, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log + operator_name: log + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_ + operator_name: log_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_out + operator_name: log + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::log.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log10 + operator_name: log10 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log10(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log10_ + operator_name: log10_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log10_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log10_out + operator_name: log10 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::log10.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log1p + operator_name: log1p + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log1p(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log1p_ + operator_name: log1p_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log1p_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log1p_out + operator_name: log1p + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::log1p.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log2 + operator_name: log2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log2(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log2_ + operator_name: log2_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log2_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log2_out + operator_name: log2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::log2.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logaddexp_out + operator_name: logaddexp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logaddexp.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logaddexp + operator_name: logaddexp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logaddexp(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logaddexp2_out + operator_name: logaddexp2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logaddexp2.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logaddexp2 + operator_name: logaddexp2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logaddexp2(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: xlogy + operator_name: xlogy + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::xlogy.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: xlogy + operator_name: xlogy + overload_name: Scalar_Self + manual_kernel_registration: false + category_override: '' + schema_string: aten::xlogy.Scalar_Self(Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: xlogy + operator_name: xlogy + overload_name: Scalar_Other + manual_kernel_registration: false + category_override: '' + schema_string: aten::xlogy.Scalar_Other(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: xlogy_ + operator_name: xlogy_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::xlogy_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: xlogy_ + operator_name: xlogy_ + overload_name: Scalar_Other + manual_kernel_registration: false + category_override: '' + schema_string: aten::xlogy_.Scalar_Other(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: xlogy_out + operator_name: xlogy + overload_name: OutTensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::xlogy.OutTensor(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: xlogy_out + operator_name: xlogy + overload_name: OutScalar_Self + manual_kernel_registration: false + category_override: '' + schema_string: aten::xlogy.OutScalar_Self(Scalar self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: xlogy_out + operator_name: xlogy + overload_name: OutScalar_Other + manual_kernel_registration: false + category_override: '' + schema_string: aten::xlogy.OutScalar_Other(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logspace + operator_name: logspace + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logspace(Scalar start, Scalar end, int steps, float base=10.0, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Scalar &, int64_t, double, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logspace + operator_name: logspace + overload_name: Tensor_Tensor + manual_kernel_registration: false + category_override: factory + schema_string: aten::logspace.Tensor_Tensor(Tensor start, Tensor end, int steps, float base=10.0, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, double, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logspace + operator_name: logspace + overload_name: Tensor_Scalar + manual_kernel_registration: false + category_override: factory + schema_string: aten::logspace.Tensor_Scalar(Tensor start, Scalar end, int steps, float base=10.0, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, int64_t, double, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logspace + operator_name: logspace + overload_name: Scalar_Tensor + manual_kernel_registration: false + category_override: factory + schema_string: aten::logspace.Scalar_Tensor(Scalar start, Tensor end, int steps, float base=10.0, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &, int64_t, double, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logspace_out + operator_name: logspace + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logspace.out(Scalar start, Scalar end, int steps, float base=10.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Scalar &, int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logspace_out + operator_name: logspace + overload_name: Tensor_Tensor_out + manual_kernel_registration: false + category_override: factory + schema_string: aten::logspace.Tensor_Tensor_out(Tensor start, Tensor end, int steps, float base=10.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logspace_out + operator_name: logspace + overload_name: Tensor_Scalar_out + manual_kernel_registration: false + category_override: factory + schema_string: aten::logspace.Tensor_Scalar_out(Tensor start, Scalar end, int steps, float base=10.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logspace_out + operator_name: logspace + overload_name: Scalar_Tensor_out + manual_kernel_registration: false + category_override: factory + schema_string: aten::logspace.Scalar_Tensor_out(Scalar start, Tensor end, int steps, float base=10.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: steps + type: int64_t + - annotation: null + default: 10.0 + dynamic_type: double + is_nullable: false + name: base + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_softmax + operator_name: log_softmax + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_softmax.int(Tensor self, int dim, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: log_softmax_out + operator_name: log_softmax + overload_name: int_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_softmax.int_out(Tensor self, int dim, ScalarType? dtype=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_softmax + operator_name: log_softmax + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_softmax.Dimname(Tensor self, Dimname dim, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _log_softmax + operator_name: _log_softmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_log_softmax(Tensor self, int dim, bool half_to_float) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _log_softmax_out + operator_name: _log_softmax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_log_softmax.out(Tensor self, int dim, bool half_to_float, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _log_softmax_backward_data + operator_name: _log_softmax_backward_data + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_log_softmax_backward_data(Tensor grad_output, Tensor output, int dim, ScalarType input_dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: input_dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: input_dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _log_softmax_backward_data_out + operator_name: _log_softmax_backward_data + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_log_softmax_backward_data.out(Tensor grad_output, Tensor output, int dim, ScalarType input_dtype, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: input_dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: input_dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _logcumsumexp + operator_name: _logcumsumexp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_logcumsumexp(Tensor self, int dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _logcumsumexp_out + operator_name: _logcumsumexp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_logcumsumexp.out(Tensor self, int dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logcumsumexp + operator_name: logcumsumexp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logcumsumexp(Tensor self, int dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logcumsumexp_out + operator_name: logcumsumexp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logcumsumexp.out(Tensor self, int dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logcumsumexp + operator_name: logcumsumexp + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::logcumsumexp.dimname(Tensor self, Dimname dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: logcumsumexp_out + operator_name: logcumsumexp + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logcumsumexp.dimname_out(Tensor self, Dimname dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Dimname, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: logsumexp + operator_name: logsumexp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logsumexp(Tensor self, int[1] dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logsumexp_out + operator_name: logsumexp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logsumexp.out(Tensor self, int[1] dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logsumexp + operator_name: logsumexp + overload_name: names + manual_kernel_registration: false + category_override: '' + schema_string: aten::logsumexp.names(Tensor self, Dimname[1] dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: logsumexp_out + operator_name: logsumexp + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logsumexp.names_out(Tensor self, Dimname[1] dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::DimnameList, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: margin_ranking_loss + operator_name: margin_ranking_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::margin_ranking_loss(Tensor input1, Tensor input2, Tensor target, float margin=0.0, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: margin + type: double + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, double, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: margin + type: double + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: matmul + operator_name: matmul + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::matmul(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: matmul_backward + operator_name: matmul_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::matmul_backward(Tensor grad, Tensor self, Tensor other, bool[2] mask) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: matmul_out + operator_name: matmul + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::matmul.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: matrix_power + operator_name: matrix_power + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::matrix_power(Tensor self, int n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: matrix_power_out + operator_name: matrix_power + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::matrix_power.out(Tensor self, int n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: matrix_exp + operator_name: matrix_exp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::matrix_exp(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: matrix_exp_backward + operator_name: matrix_exp_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::matrix_exp_backward(Tensor self, Tensor grad) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _aminmax + operator_name: _aminmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_aminmax(Tensor self) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _aminmax + operator_name: _aminmax + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::_aminmax.dim(Tensor self, int dim, bool keepdim=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: aminmax + operator_name: aminmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::aminmax(Tensor self, *, int? dim=None, bool keepdim=False) -> (Tensor min, Tensor max) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: min + name: min + type: at::Tensor + - dynamic_type: at::Tensor + field_name: max + name: max + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: aminmax_out + operator_name: aminmax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::aminmax.out(Tensor self, *, int? dim=None, bool keepdim=False, Tensor(a!) min, Tensor(b!) max) -> (Tensor(a!) min, Tensor(b!) max) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: min + is_nullable: false + name: min + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: max + is_nullable: false + name: max + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, ::std::optional, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: min + is_nullable: false + name: min + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: max + is_nullable: false + name: max + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: min + name: min + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: max + name: max + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _compute_linear_combination + operator_name: _compute_linear_combination + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_compute_linear_combination(Tensor input, Tensor coefficients) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: coefficients + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: coefficients + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _compute_linear_combination_out + operator_name: _compute_linear_combination + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_compute_linear_combination.out(Tensor input, Tensor coefficients, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: coefficients + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: coefficients + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max + operator_name: max + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::max.dim(Tensor self, int dim, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_out + operator_name: max + overload_name: dim_max + manual_kernel_registration: false + category_override: '' + schema_string: aten::max.dim_max(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) max, Tensor(b!) max_values) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: max + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: max_values + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: max + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: max_values + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: max + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: max_values + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max + operator_name: max + overload_name: names_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::max.names_dim(Tensor self, Dimname dim, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: max_out + operator_name: max + overload_name: names_dim_max + manual_kernel_registration: false + category_override: '' + schema_string: aten::max.names_dim_max(Tensor self, Dimname dim, bool keepdim=False, *, Tensor(a!) max, Tensor(b!) max_values) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: max + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: max_values + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: max + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: max_values + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: max + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: max_values + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: value_selecting_reduction_backward + operator_name: value_selecting_reduction_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::value_selecting_reduction_backward(Tensor grad, int dim, Tensor indices, SymInt[] sizes, bool keepdim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: amax + operator_name: amax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::amax(Tensor self, int[1] dim=[], bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: amax_out + operator_name: amax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::amax.out(Tensor self, int[1] dim=[], bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool1d_with_indices + operator_name: max_pool1d_with_indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool1d_with_indices(Tensor self, int[1] kernel_size, int[1] stride=[], int[1] padding=0, int[1] dilation=1, bool ceil_mode=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: max_pool1d + operator_name: max_pool1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool1d(Tensor self, int[1] kernel_size, int[1] stride=[], int[1] padding=0, int[1] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: max_pool2d + operator_name: max_pool2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool2d(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: max_pool2d_backward + operator_name: max_pool2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool2d_backward(Tensor grad_output, Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_max_pool2d + operator_name: mkldnn_max_pool2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_max_pool2d(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_max_pool2d_backward + operator_name: mkldnn_max_pool2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_max_pool2d_backward(Tensor grad_output, Tensor output, Tensor input, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_max_pool3d + operator_name: mkldnn_max_pool3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_max_pool3d(Tensor self, int[3] kernel_size, int[3] stride=[], int[3] padding=0, int[3] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_max_pool3d_backward + operator_name: mkldnn_max_pool3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_max_pool3d_backward(Tensor grad_output, Tensor output, Tensor input, int[3] kernel_size, int[3] stride=[], int[3] padding=0, int[3] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantized_max_pool1d + operator_name: quantized_max_pool1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_max_pool1d(Tensor self, int[1] kernel_size, int[1] stride=[], int[1] padding=0, int[1] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantized_max_pool2d + operator_name: quantized_max_pool2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_max_pool2d(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantized_max_pool3d + operator_name: quantized_max_pool3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_max_pool3d(Tensor self, int[3] kernel_size, int[3] stride=[], int[3] padding=0, int[3] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool3d + operator_name: max_pool3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool3d(Tensor self, int[3] kernel_size, int[3] stride=[], int[3] padding=0, int[3] dilation=1, bool ceil_mode=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: mean + operator_name: mean + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mean(Tensor self, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mean_out + operator_name: mean + overload_name: dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mean.dtype_out(Tensor self, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mean + operator_name: mean + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::mean.dim(Tensor self, int[1]? dim, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mean_out + operator_name: mean + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mean.out(Tensor self, int[1]? dim, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mean + operator_name: mean + overload_name: names_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::mean.names_dim(Tensor self, Dimname[1] dim, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: mean_out + operator_name: mean + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mean.names_out(Tensor self, Dimname[1] dim, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::DimnameList, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nanmean + operator_name: nanmean + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanmean(Tensor self, int[1]? dim=None, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nanmean_out + operator_name: nanmean + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanmean.out(Tensor self, int[1]? dim=None, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: median + operator_name: median + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::median(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: median + operator_name: median + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::median.dim(Tensor self, int dim, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: median_out + operator_name: median + overload_name: dim_values + manual_kernel_registration: false + category_override: '' + schema_string: aten::median.dim_values(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: median + operator_name: median + overload_name: names_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::median.names_dim(Tensor self, Dimname dim, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: median_out + operator_name: median + overload_name: names_dim_values + manual_kernel_registration: false + category_override: '' + schema_string: aten::median.names_dim_values(Tensor self, Dimname dim, bool keepdim=False, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nanmedian + operator_name: nanmedian + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanmedian(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nanmedian + operator_name: nanmedian + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanmedian.dim(Tensor self, int dim, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nanmedian_out + operator_name: nanmedian + overload_name: dim_values + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanmedian.dim_values(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nanmedian + operator_name: nanmedian + overload_name: names_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanmedian.names_dim(Tensor self, Dimname dim, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nanmedian_out + operator_name: nanmedian + overload_name: names_dim_values + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanmedian.names_dim_values(Tensor self, Dimname dim, bool keepdim=False, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: min + operator_name: min + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::min.dim(Tensor self, int dim, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: min_out + operator_name: min + overload_name: dim_min + manual_kernel_registration: false + category_override: '' + schema_string: aten::min.dim_min(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) min, Tensor(b!) min_indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: min + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: min_indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: min + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: min_indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: min + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: min_indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: min + operator_name: min + overload_name: names_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::min.names_dim(Tensor self, Dimname dim, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: min_out + operator_name: min + overload_name: names_dim_min + manual_kernel_registration: false + category_override: '' + schema_string: aten::min.names_dim_min(Tensor self, Dimname dim, bool keepdim=False, *, Tensor(a!) min, Tensor(b!) min_indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: min + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: min_indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: min + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: min_indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: min + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: min_indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: amin + operator_name: amin + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::amin(Tensor self, int[1] dim=[], bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: amin_out + operator_name: amin + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::amin.out(Tensor self, int[1] dim=[], bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _mps_convolution + operator_name: _mps_convolution + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_mps_convolution(Tensor self, Tensor weight, Tensor? bias, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mps_convolution_backward + operator_name: mps_convolution_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mps_convolution_backward(Tensor self, Tensor grad_output, Tensor weight, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool[3] output_mask) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_convolution + operator_name: mkldnn_convolution + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_convolution(Tensor self, Tensor weight, Tensor? bias, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_rnn_layer + operator_name: mkldnn_rnn_layer + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_rnn_layer(Tensor input, Tensor weight0, Tensor weight1, Tensor weight2, Tensor weight3, Tensor hx_, Tensor cx_, bool reverse, int[] batch_sizes, int mode, int hidden_size, int num_layers, bool has_biases, bool bidirectional, bool batch_first, bool train) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight0 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx_ + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: reverse + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, at::IntArrayRef, int64_t, int64_t, int64_t, bool, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight0 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx_ + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: reverse + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_rnn_layer_backward + operator_name: mkldnn_rnn_layer_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_rnn_layer_backward(Tensor input, Tensor weight1, Tensor weight2, Tensor weight3, Tensor weight4, Tensor hx_, Tensor cx_tmp, Tensor output, Tensor hy_, Tensor cy_, Tensor? grad_output, Tensor? grad_hy, Tensor? grad_cy, bool reverse, int mode, int hidden_size, int num_layers, bool has_biases, bool train, bool bidirectional, int[] batch_sizes, bool batch_first, Tensor workspace) -> (Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight4 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx_tmp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hy_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: reverse + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, int64_t, int64_t, int64_t, bool, bool, bool, at::IntArrayRef, bool, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight4 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx_tmp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hy_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: reverse + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: at::Tensor + name: result4 + type: at::Tensor + - dynamic_type: at::Tensor + name: result5 + type: at::Tensor + - dynamic_type: at::Tensor + name: result6 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_batch_norm + operator_name: miopen_batch_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_batch_norm(Tensor input, Tensor weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, bool training, float exponential_average_factor, float epsilon) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: exponential_average_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: exponential_average_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_batch_norm_backward + operator_name: miopen_batch_norm_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_batch_norm_backward(Tensor input, Tensor grad_output, Tensor weight, Tensor? running_mean, Tensor? running_var, Tensor? save_mean, Tensor? save_var, float epsilon) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_convolution + operator_name: miopen_convolution + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_convolution(Tensor self, Tensor weight, Tensor? bias, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool benchmark, bool deterministic) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_convolution_transpose + operator_name: miopen_convolution_transpose + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_convolution_transpose(Tensor self, Tensor weight, Tensor? bias, SymInt[] padding, SymInt[] output_padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool benchmark, bool deterministic) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_depthwise_convolution + operator_name: miopen_depthwise_convolution + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_depthwise_convolution(Tensor self, Tensor weight, Tensor? bias, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool benchmark, bool deterministic) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_convolution_relu + operator_name: miopen_convolution_relu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_convolution_relu(Tensor self, Tensor weight, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_convolution_add_relu + operator_name: miopen_convolution_add_relu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_convolution_add_relu(Tensor self, Tensor weight, Tensor z, Scalar? alpha, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: z + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: alpha + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: z + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: alpha + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_rnn + operator_name: miopen_rnn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_rnn(Tensor input, Tensor[] weight, int weight_stride0, Tensor hx, Tensor? cx, int mode, int hidden_size, int num_layers, bool batch_first, float dropout, bool train, bool bidirectional, int[] batch_sizes, Tensor? dropout_state) -> (Tensor, Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::TensorList, int64_t, const at::Tensor &, const ::std::optional &, int64_t, int64_t, int64_t, bool, double, bool, bool, at::IntArrayRef, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: at::Tensor + name: result4 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_rnn_backward + operator_name: miopen_rnn_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_rnn_backward(Tensor input, Tensor[] weight, int weight_stride0, Tensor weight_buf, Tensor hx, Tensor? cx, Tensor output, Tensor? grad_output, Tensor? grad_hy, Tensor? grad_cy, int mode, int hidden_size, int num_layers, bool batch_first, float dropout, bool train, bool bidirectional, int[] batch_sizes, Tensor? dropout_state, Tensor reserve, bool[4] output_mask) -> (Tensor, Tensor, Tensor, Tensor[]) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_buf + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserve + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple> (const at::Tensor &, at::TensorList, int64_t, const at::Tensor &, const at::Tensor &, const ::std::optional &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, int64_t, int64_t, int64_t, bool, double, bool, bool, at::IntArrayRef, const ::std::optional &, const at::Tensor &, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_buf + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserve + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::TensorList + name: result3 + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mm + operator_name: mm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mm(Tensor self, Tensor mat2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mm_out + operator_name: mm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mm.out(Tensor self, Tensor mat2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mm + operator_name: mm + overload_name: dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::mm.dtype(Tensor self, Tensor mat2, ScalarType out_dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mm_out + operator_name: mm + overload_name: dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mm.dtype_out(Tensor self, Tensor mat2, ScalarType out_dtype, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _int_mm + operator_name: _int_mm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_int_mm(Tensor self, Tensor mat2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _int_mm_out + operator_name: _int_mm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_int_mm.out(Tensor self, Tensor mat2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _convert_weight_to_int4pack + operator_name: _convert_weight_to_int4pack + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convert_weight_to_int4pack(Tensor self, int innerKTiles) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: innerKTiles + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: innerKTiles + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _weight_int4pack_mm + operator_name: _weight_int4pack_mm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_weight_int4pack_mm(Tensor self, Tensor mat2, int qGroupSize, Tensor qScaleAndZeros) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: qGroupSize + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qScaleAndZeros + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: qGroupSize + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qScaleAndZeros + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _weight_int4pack_mm_with_scales_and_zeros + operator_name: _weight_int4pack_mm_with_scales_and_zeros + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_weight_int4pack_mm_with_scales_and_zeros(Tensor self, Tensor mat2, int qGroupSize, Tensor qScale, Tensor qZeros) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: qGroupSize + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qScale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qZeros + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: qGroupSize + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qScale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qZeros + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _convert_weight_to_int4pack_for_cpu + operator_name: _convert_weight_to_int4pack_for_cpu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convert_weight_to_int4pack_for_cpu(Tensor self, int innerKTiles) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: innerKTiles + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: innerKTiles + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _weight_int4pack_mm_for_cpu + operator_name: _weight_int4pack_mm_for_cpu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_weight_int4pack_mm_for_cpu(Tensor self, Tensor mat2, int qGroupSize, Tensor qScaleAndZeros) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: qGroupSize + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qScaleAndZeros + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: qGroupSize + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qScaleAndZeros + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _dyn_quant_pack_4bit_weight + operator_name: _dyn_quant_pack_4bit_weight + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_dyn_quant_pack_4bit_weight(Tensor weights, Tensor scales_zeros, Tensor? bias, int block_size, int in_features, int out_features) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weights + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales_zeros + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: block_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: in_features + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: out_features + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weights + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales_zeros + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: block_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: in_features + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: out_features + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _dyn_quant_matmul_4bit + operator_name: _dyn_quant_matmul_4bit + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_dyn_quant_matmul_4bit(Tensor inp, Tensor packed_weights, int block_size, int in_features, int out_features) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: inp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weights + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: block_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: in_features + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: out_features + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: inp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_weights + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: block_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: in_features + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: out_features + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _weight_int8pack_mm + operator_name: _weight_int8pack_mm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_weight_int8pack_mm(Tensor self, Tensor mat2, Tensor scales) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_mm + operator_name: _sparse_mm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_mm(Tensor sparse, Tensor dense) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sparse + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sparse + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_mm + operator_name: _sparse_mm + overload_name: reduce + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_mm.reduce(Tensor sparse, Tensor dense, str reduce) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sparse + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sparse + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_sparse_matmul + operator_name: _sparse_sparse_matmul + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_sparse_matmul(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mode + operator_name: mode + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mode(Tensor self, int dim=-1, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mode_out + operator_name: mode + overload_name: values + manual_kernel_registration: false + category_override: '' + schema_string: aten::mode.values(Tensor self, int dim=-1, bool keepdim=False, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mode + operator_name: mode + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::mode.dimname(Tensor self, Dimname dim, bool keepdim=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: mode_out + operator_name: mode + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mode.dimname_out(Tensor self, Dimname dim, bool keepdim=False, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: mul + operator_name: mul + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::mul.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mul_ + operator_name: mul_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::mul_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mul_out + operator_name: mul + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mul.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mul + operator_name: mul + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::mul.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mul_ + operator_name: mul_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::mul_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: multiply + operator_name: multiply + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::multiply.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: multiply_ + operator_name: multiply_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::multiply_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: multiply_out + operator_name: multiply + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::multiply.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: multiply + operator_name: multiply + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::multiply.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: multiply_ + operator_name: multiply_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::multiply_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: mv + operator_name: mv + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mv(Tensor self, Tensor vec) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mv_out + operator_name: mv + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mv.out(Tensor self, Tensor vec, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mvlgamma_out + operator_name: mvlgamma + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mvlgamma.out(Tensor self, int p, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: p + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: p + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mvlgamma + operator_name: mvlgamma + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mvlgamma(Tensor self, int p) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: p + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: p + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mvlgamma_ + operator_name: mvlgamma_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mvlgamma_(Tensor(a!) self, int p) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: p + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: p + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: narrow_copy + operator_name: narrow_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::narrow_copy(Tensor self, int dim, SymInt start, SymInt length) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: length + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: length + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: narrow_copy_out + operator_name: narrow_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::narrow_copy.out(Tensor self, int dim, SymInt start, SymInt length, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: length + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: length + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: narrow + operator_name: narrow + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::narrow(Tensor(a) self, int dim, SymInt start, SymInt length) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: length + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: length + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: narrow + operator_name: narrow + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::narrow.Tensor(Tensor(a) self, int dim, Tensor start, SymInt length) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: length + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: start + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: length + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: native_batch_norm + operator_name: native_batch_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_batch_norm(Tensor input, Tensor? weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, bool training, float momentum, float eps) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_batch_norm_out + operator_name: native_batch_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_batch_norm.out(Tensor input, Tensor? weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, bool training, float momentum, float eps, *, Tensor(a!) out, Tensor(b!) save_mean, Tensor(c!) save_invstd) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: save_mean + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: save_invstd + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, double, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: save_mean + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: save_invstd + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + - dynamic_type: at::Tensor + name: save_mean + type: at::Tensor & + - dynamic_type: at::Tensor + name: save_invstd + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _native_batch_norm_legit + operator_name: _native_batch_norm_legit + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_native_batch_norm_legit(Tensor input, Tensor? weight, Tensor? bias, Tensor(a!) running_mean, Tensor(b!) running_var, bool training, float momentum, float eps) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, at::Tensor &, at::Tensor &, bool, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _native_batch_norm_legit_no_training + operator_name: _native_batch_norm_legit_no_training + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_native_batch_norm_legit_no_training(Tensor input, Tensor? weight, Tensor? bias, Tensor running_mean, Tensor running_var, float momentum, float eps) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _native_batch_norm_legit_out + operator_name: _native_batch_norm_legit + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_native_batch_norm_legit.out(Tensor input, Tensor? weight, Tensor? bias, Tensor(a!) running_mean, Tensor(b!) running_var, bool training, float momentum, float eps, *, Tensor(d!) out, Tensor(e!) save_mean, Tensor(f!) save_invstd) -> (Tensor(d!), Tensor(e!), Tensor(f!)) + arguments: + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: save_mean + output: true + type: at::Tensor & + - allocate: true + annotation: f! + dynamic_type: at::Tensor + is_nullable: false + name: save_invstd + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, at::Tensor &, at::Tensor &, bool, double, double, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: save_mean + output: true + type: at::Tensor & + - allocate: true + annotation: f! + dynamic_type: at::Tensor + is_nullable: false + name: save_invstd + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + - dynamic_type: at::Tensor + name: save_mean + type: at::Tensor & + - dynamic_type: at::Tensor + name: save_invstd + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _native_batch_norm_legit + operator_name: _native_batch_norm_legit + overload_name: no_stats + manual_kernel_registration: false + category_override: '' + schema_string: aten::_native_batch_norm_legit.no_stats(Tensor input, Tensor? weight, Tensor? bias, bool training, float momentum, float eps) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, bool, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _native_batch_norm_legit_out + operator_name: _native_batch_norm_legit + overload_name: no_stats_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_native_batch_norm_legit.no_stats_out(Tensor input, Tensor? weight, Tensor? bias, bool training, float momentum, float eps, *, Tensor(a!) out, Tensor(b!) save_mean, Tensor(c!) save_invstd) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: save_mean + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: save_invstd + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, bool, double, double, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: save_mean + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: save_invstd + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + - dynamic_type: at::Tensor + name: save_mean + type: at::Tensor & + - dynamic_type: at::Tensor + name: save_invstd + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_stats + operator_name: batch_norm_stats + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_stats(Tensor input, float eps) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_elemt + operator_name: batch_norm_elemt + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_elemt(Tensor input, Tensor? weight, Tensor? bias, Tensor mean, Tensor invstd, float eps) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_elemt_out + operator_name: batch_norm_elemt + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_elemt.out(Tensor input, Tensor? weight, Tensor? bias, Tensor mean, Tensor invstd, float eps, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_gather_stats + operator_name: batch_norm_gather_stats + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_gather_stats(Tensor input, Tensor mean, Tensor invstd, Tensor? running_mean, Tensor? running_var, float momentum, float eps, int count) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: count + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, double, double, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: count + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_gather_stats_with_counts + operator_name: batch_norm_gather_stats_with_counts + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_gather_stats_with_counts(Tensor input, Tensor mean, Tensor invstd, Tensor? running_mean, Tensor? running_var, float momentum, float eps, Tensor counts) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: counts + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, double, double, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: counts + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_batch_norm_backward + operator_name: native_batch_norm_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_batch_norm_backward(Tensor grad_out, Tensor input, Tensor? weight, Tensor? running_mean, Tensor? running_var, Tensor? save_mean, Tensor? save_invstd, bool train, float eps, bool[3] output_mask) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_invstd + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_invstd + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_backward_reduce + operator_name: batch_norm_backward_reduce + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_backward_reduce(Tensor grad_out, Tensor input, Tensor mean, Tensor invstd, Tensor? weight, bool input_g, bool weight_g, bool bias_g) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: input_g + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: weight_g + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bias_g + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: input_g + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: weight_g + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bias_g + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_backward_elemt + operator_name: batch_norm_backward_elemt + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_backward_elemt(Tensor grad_out, Tensor input, Tensor mean, Tensor invstd, Tensor? weight, Tensor sum_dy, Tensor sum_dy_xmu, Tensor count) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sum_dy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sum_dy_xmu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: count + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sum_dy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sum_dy_xmu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: count + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_update_stats + operator_name: batch_norm_update_stats + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_update_stats(Tensor input, Tensor? running_mean, Tensor? running_var, float momentum) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: is_vulkan_available + operator_name: is_vulkan_available + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_vulkan_available() -> bool + arguments: [] + schema_order_cpp_signature: bool () + schema_order_arguments: [] + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _nnpack_available + operator_name: _nnpack_available + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nnpack_available() -> bool + arguments: [] + schema_order_cpp_signature: bool () + schema_order_arguments: [] + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _nnpack_spatial_convolution + operator_name: _nnpack_spatial_convolution + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nnpack_spatial_convolution(Tensor input, Tensor weight, Tensor? bias, SymInt[2] padding, SymInt[2] stride=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ones + operator_name: ones + overload_name: names + manual_kernel_registration: false + category_override: '' + schema_string: aten::ones.names(int[] size, *, Dimname[]? names, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: ones + operator_name: ones + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::ones(SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ones_out + operator_name: ones + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ones.out(SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ones_like + operator_name: ones_like + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::ones_like(Tensor self, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pairwise_distance + operator_name: pairwise_distance + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::pairwise_distance(Tensor x1, Tensor x2, float p=2, float eps=1e-06, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: 1.0e-06 + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, double, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: 1.0e-06 + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cdist + operator_name: cdist + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cdist(Tensor x1, Tensor x2, float p=2, int? compute_mode=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: compute_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: compute_mode + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _euclidean_dist + operator_name: _euclidean_dist + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_euclidean_dist(Tensor x1, Tensor x2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cdist_forward + operator_name: _cdist_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cdist_forward(Tensor x1, Tensor x2, float p, int? compute_mode) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: compute_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: compute_mode + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cdist_backward + operator_name: _cdist_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cdist_backward(Tensor grad, Tensor x1, Tensor x2, float p, Tensor cdist) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cdist + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, double, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cdist + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pdist + operator_name: pdist + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::pdist(Tensor self, float p=2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _pdist_forward + operator_name: _pdist_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pdist_forward(Tensor self, float p=2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _pdist_backward + operator_name: _pdist_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pdist_backward(Tensor grad, Tensor self, float p, Tensor pdist) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pdist + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, double, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pdist + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cosine_similarity + operator_name: cosine_similarity + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cosine_similarity(Tensor x1, Tensor x2, int dim=1, float eps=1e-08) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: 1.0e-08 + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: 1.0e-08 + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: permute + operator_name: permute + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::permute(Tensor(a) self, int[] dims) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: movedim + operator_name: movedim + overload_name: intlist + manual_kernel_registration: false + category_override: '' + schema_string: aten::movedim.intlist(Tensor(a) self, int[] source, int[] destination) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: source + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: destination + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: source + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: destination + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: movedim + operator_name: movedim + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::movedim.int(Tensor(a) self, int source, int destination) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: source + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: destination + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: source + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: destination + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: moveaxis + operator_name: moveaxis + overload_name: intlist + manual_kernel_registration: false + category_override: '' + schema_string: aten::moveaxis.intlist(Tensor(a) self, int[] source, int[] destination) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: source + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: destination + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: source + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: destination + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: moveaxis + operator_name: moveaxis + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::moveaxis.int(Tensor(a) self, int source, int destination) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: source + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: destination + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: source + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: destination + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: numpy_T + operator_name: numpy_T + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::numpy_T(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: matrix_H + operator_name: matrix_H + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::matrix_H(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: mT + operator_name: mT + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mT(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: mH + operator_name: mH + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mH(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: adjoint + operator_name: adjoint + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::adjoint(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: pixel_shuffle + operator_name: pixel_shuffle + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::pixel_shuffle(Tensor self, int upscale_factor) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: upscale_factor + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: upscale_factor + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pixel_unshuffle + operator_name: pixel_unshuffle + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::pixel_unshuffle(Tensor self, int downscale_factor) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: downscale_factor + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: downscale_factor + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: channel_shuffle + operator_name: channel_shuffle + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::channel_shuffle(Tensor self, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_channel_shuffle + operator_name: native_channel_shuffle + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_channel_shuffle(Tensor self, SymInt groups) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: is_pinned + operator_name: is_pinned + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_pinned(Tensor self, Device? device=None) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + schema_order_cpp_signature: bool (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pin_memory + operator_name: pin_memory + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::pin_memory(Tensor(a) self, Device? device=None) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _pin_memory + operator_name: _pin_memory + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pin_memory(Tensor self, Device? device=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pinverse + operator_name: pinverse + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::pinverse(Tensor self, float rcond=1e-15) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1.0e-15 + dynamic_type: double + is_nullable: false + name: rcond + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1.0e-15 + dynamic_type: double + is_nullable: false + name: rcond + type: double + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: poisson_nll_loss + operator_name: poisson_nll_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::poisson_nll_loss(Tensor input, Tensor target, bool log_input, bool full, float eps, int reduction) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: log_input + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: full + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool, bool, double, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: log_input + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: full + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rad2deg + operator_name: rad2deg + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rad2deg(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rad2deg_ + operator_name: rad2deg_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rad2deg_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rad2deg_out + operator_name: rad2deg + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rad2deg.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: deg2rad + operator_name: deg2rad + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::deg2rad(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: deg2rad_ + operator_name: deg2rad_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::deg2rad_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: deg2rad_out + operator_name: deg2rad + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::deg2rad.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scalar_tensor + operator_name: scalar_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::scalar_tensor(Scalar s, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: s + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Scalar &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: s + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand + operator_name: rand + overload_name: names + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand.names(SymInt[] size, *, Dimname[]? names, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand + operator_name: rand + overload_name: generator_with_names + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand.generator_with_names(SymInt[] size, *, Generator? generator, Dimname[]? names, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand + operator_name: rand + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand(SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand + operator_name: rand + overload_name: generator + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand.generator(SymInt[] size, *, Generator? generator, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand_out + operator_name: rand + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand.out(SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand_out + operator_name: rand + overload_name: generator_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand.generator_out(SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rand_like + operator_name: rand_like + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand_like(Tensor self, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand_like + operator_name: rand_like + overload_name: generator + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand_like.generator(Tensor self, *, Generator? generator, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint + operator_name: randint + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint(SymInt high, SymInt[] size, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: at::kLong + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: at::kLong + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint + operator_name: randint + overload_name: generator + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint.generator(SymInt high, SymInt[] size, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: at::kLong + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: at::kLong + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint + operator_name: randint + overload_name: low + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint.low(SymInt low, SymInt high, SymInt[] size, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: at::kLong + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, int64_t, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: at::kLong + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint + operator_name: randint + overload_name: low_generator + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint.low_generator(SymInt low, SymInt high, SymInt[] size, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: at::kLong + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, int64_t, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: at::kLong + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_out + operator_name: randint + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint.out(SymInt high, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (int64_t, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_out + operator_name: randint + overload_name: generator_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint.generator_out(SymInt high, SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (int64_t, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_out + operator_name: randint + overload_name: low_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint.low_out(SymInt low, SymInt high, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (int64_t, int64_t, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_out + operator_name: randint + overload_name: low_generator_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint.low_generator_out(SymInt low, SymInt high, SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (int64_t, int64_t, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like + operator_name: randint_like + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like(Tensor self, SymInt high, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like + operator_name: randint_like + overload_name: generator + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.generator(Tensor self, SymInt high, *, Generator? generator, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like + operator_name: randint_like + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.Tensor(Tensor self, Tensor high, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: high + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: high + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like + operator_name: randint_like + overload_name: Tensor_generator + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.Tensor_generator(Tensor self, Tensor high, *, Generator? generator, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: high + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: high + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like + operator_name: randint_like + overload_name: low_dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.low_dtype(Tensor self, SymInt low, SymInt high, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like + operator_name: randint_like + overload_name: low_generator_dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.low_generator_dtype(Tensor self, SymInt low, SymInt high, *, Generator? generator, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randn + operator_name: randn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn(SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randn + operator_name: randn + overload_name: generator + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn.generator(SymInt[] size, *, Generator? generator, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randn + operator_name: randn + overload_name: names + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn.names(SymInt[] size, *, Dimname[]? names, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randn + operator_name: randn + overload_name: generator_with_names + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn.generator_with_names(SymInt[] size, *, Generator? generator, Dimname[]? names, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randn_out + operator_name: randn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn.out(SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: randn_out + operator_name: randn + overload_name: generator_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn.generator_out(SymInt[] size, *, Generator? generator, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: randn_like + operator_name: randn_like + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn_like(Tensor self, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randn_like + operator_name: randn_like + overload_name: generator + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn_like.generator(Tensor self, *, Generator? generator, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randperm + operator_name: randperm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::randperm(SymInt n, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: at::kLong + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: at::kLong + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randperm + operator_name: randperm + overload_name: generator + manual_kernel_registration: false + category_override: '' + schema_string: aten::randperm.generator(SymInt n, *, Generator? generator, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: at::kLong + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: at::kLong + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randperm_out + operator_name: randperm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randperm.out(SymInt n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + schema_order_cpp_signature: at::Tensor & (int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: randperm_out + operator_name: randperm + overload_name: generator_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randperm.generator_out(SymInt n, *, Generator? generator, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: range + operator_name: range + overload_name: step + manual_kernel_registration: false + category_override: '' + schema_string: aten::range.step(Scalar start, Scalar end, Scalar step=1, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: step + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Scalar &, const at::Scalar &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: step + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: range + operator_name: range + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::range(Scalar start, Scalar end, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Scalar &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: range_out + operator_name: range + overload_name: out_ + manual_kernel_registration: false + category_override: '' + schema_string: aten::range.out_(Scalar start, Scalar end, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: range_out + operator_name: range + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::range.out(Scalar start, Scalar end, Scalar step=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: step + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: start + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: end + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: step + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ravel + operator_name: ravel + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::ravel(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: reciprocal + operator_name: reciprocal + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::reciprocal(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reciprocal_ + operator_name: reciprocal_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::reciprocal_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reciprocal_out + operator_name: reciprocal + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::reciprocal.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: neg + operator_name: neg + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::neg(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: neg_ + operator_name: neg_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::neg_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: neg_out + operator_name: neg + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::neg.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: negative + operator_name: negative + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::negative(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: negative_ + operator_name: negative_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::negative_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: negative_out + operator_name: negative + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::negative.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: repeat + operator_name: repeat + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::repeat(Tensor self, SymInt[] repeats) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: repeats + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: repeats + type: at::IntArrayRef + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: repeat_interleave + operator_name: repeat_interleave + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::repeat_interleave.Tensor(Tensor repeats, *, SymInt? output_size=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: repeats + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: output_size + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: repeats + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: output_size + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: repeat_interleave + operator_name: repeat_interleave + overload_name: self_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::repeat_interleave.self_Tensor(Tensor self, Tensor repeats, int? dim=None, *, SymInt? output_size=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: repeats + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: output_size + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: repeats + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: output_size + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: repeat_interleave + operator_name: repeat_interleave + overload_name: self_int + manual_kernel_registration: false + category_override: '' + schema_string: aten::repeat_interleave.self_int(Tensor self, SymInt repeats, int? dim=None, *, SymInt? output_size=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: repeats + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: output_size + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: repeats + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: output_size + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: reshape + operator_name: reshape + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::reshape(Tensor(a) self, SymInt[] shape) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shape + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shape + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: _reshape_copy + operator_name: _reshape_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_reshape_copy(Tensor self, SymInt[] size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _reshape_alias + operator_name: _reshape_alias + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_reshape_alias(Tensor(a) self, SymInt[] size, SymInt[] stride) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _mkldnn_reshape + operator_name: _mkldnn_reshape + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_mkldnn_reshape(Tensor self, int[] shape) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shape + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shape + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: reshape_as + operator_name: reshape_as + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::reshape_as(Tensor(a) self, Tensor other) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: round + operator_name: round + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::round(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: round_ + operator_name: round_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::round_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: round_out + operator_name: round + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::round.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: round + operator_name: round + overload_name: decimals + manual_kernel_registration: false + category_override: '' + schema_string: aten::round.decimals(Tensor self, *, int decimals) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: decimals + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: decimals + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: round_ + operator_name: round_ + overload_name: decimals + manual_kernel_registration: false + category_override: '' + schema_string: aten::round_.decimals(Tensor(a!) self, *, int decimals) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: decimals + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: decimals + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: round_out + operator_name: round + overload_name: decimals_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::round.decimals_out(Tensor self, *, int decimals, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: decimals + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: decimals + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rrelu + operator_name: rrelu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rrelu(Tensor self, Scalar lower=0.125, Scalar upper=0.3333333333333333, bool training=False, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rrelu_ + operator_name: rrelu_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rrelu_(Tensor(a!) self, Scalar lower=0.125, Scalar upper=0.3333333333333333, bool training=False, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, const at::Scalar &, bool, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: relu + operator_name: relu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::relu(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: relu_ + operator_name: relu_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::relu_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: relu6 + operator_name: relu6 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::relu6(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: relu6_ + operator_name: relu6_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::relu6_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: prelu + operator_name: prelu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::prelu(Tensor self, Tensor weight) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _prelu_kernel + operator_name: _prelu_kernel + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_prelu_kernel(Tensor self, Tensor weight) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _prelu_kernel_backward + operator_name: _prelu_kernel_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_prelu_kernel_backward(Tensor grad_output, Tensor self, Tensor weight) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gelu_out + operator_name: gelu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::gelu.out(Tensor self, *, str approximate='none', Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"none"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: approximate + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, c10::string_view, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"none"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: approximate + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gelu_ + operator_name: gelu_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::gelu_(Tensor(a!) self, *, str approximate='none') -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: '"none"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: approximate + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: '"none"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: approximate + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gelu + operator_name: gelu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::gelu(Tensor self, *, str approximate='none') -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"none"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: approximate + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"none"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: approximate + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gelu_backward_out + operator_name: gelu_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::gelu_backward.grad_input(Tensor grad_output, Tensor self, *, str approximate='none', Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"none"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: approximate + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, c10::string_view, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"none"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: approximate + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gelu_backward + operator_name: gelu_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::gelu_backward(Tensor grad_output, Tensor self, *, str approximate='none') -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"none"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: approximate + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"none"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: approximate + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: infinitely_differentiable_gelu_backward + operator_name: infinitely_differentiable_gelu_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::infinitely_differentiable_gelu_backward(Tensor grad, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: hardshrink_out + operator_name: hardshrink + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardshrink.out(Tensor self, Scalar lambd=0.5, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardshrink + operator_name: hardshrink + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardshrink(Tensor self, Scalar lambd=0.5) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardshrink_backward_out + operator_name: hardshrink_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardshrink_backward.grad_input(Tensor grad_out, Tensor self, Scalar lambd, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardshrink_backward + operator_name: hardshrink_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardshrink_backward(Tensor grad_out, Tensor self, Scalar lambd) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rsqrt + operator_name: rsqrt + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rsqrt(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rsqrt_ + operator_name: rsqrt_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rsqrt_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rsqrt_out + operator_name: rsqrt + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rsqrt.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: select + operator_name: select + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::select.Dimname(Tensor(a) self, Dimname dim, int index) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: select + operator_name: select + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::select.int(Tensor(a) self, int dim, SymInt index) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: select_backward + operator_name: select_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::select_backward(Tensor grad_output, SymInt[] input_sizes, int dim, SymInt index) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_select_backward + operator_name: _nested_select_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_select_backward(Tensor grad_output, Tensor self, int dim, SymInt index) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: selu + operator_name: selu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::selu(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: selu_ + operator_name: selu_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::selu_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: celu + operator_name: celu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::celu(Tensor self, Scalar alpha=1.0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1.0 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1.0 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: celu_ + operator_name: celu_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::celu_(Tensor(a!) self, Scalar alpha=1.0) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 1.0 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 1.0 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: silu + operator_name: silu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::silu(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: silu_ + operator_name: silu_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::silu_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: silu_out + operator_name: silu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::silu.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: silu_backward_out + operator_name: silu_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::silu_backward.grad_input(Tensor grad_output, Tensor self, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: silu_backward + operator_name: silu_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::silu_backward(Tensor grad_output, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: mish + operator_name: mish + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mish(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mish_ + operator_name: mish_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mish_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mish_out + operator_name: mish + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mish.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mish_backward + operator_name: mish_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mish_backward(Tensor grad_output, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sigmoid + operator_name: sigmoid + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sigmoid(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sigmoid_ + operator_name: sigmoid_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sigmoid_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sigmoid_out + operator_name: sigmoid + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sigmoid.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logit + operator_name: logit + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logit(Tensor self, float? eps=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logit_ + operator_name: logit_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logit_(Tensor(a!) self, float? eps=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logit_out + operator_name: logit + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::logit.out(Tensor self, float? eps=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sin + operator_name: sin + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sin(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sin_ + operator_name: sin_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sin_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sin_out + operator_name: sin + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sin.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sinc + operator_name: sinc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sinc(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sinc_ + operator_name: sinc_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sinc_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sinc_out + operator_name: sinc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sinc.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sinh + operator_name: sinh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sinh(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sinh_ + operator_name: sinh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sinh_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sinh_out + operator_name: sinh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sinh.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: detach + operator_name: detach + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::detach(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: detach_ + operator_name: detach_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::detach_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: size + operator_name: size + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::size.int(Tensor self, int dim) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: int64_t (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: size + operator_name: size + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::size.Dimname(Tensor self, Dimname dim) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: int64_t (const at::Tensor &, at::Dimname) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: sym_size + operator_name: sym_size + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::sym_size.int(Tensor self, int dim) -> SymInt + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: c10::SymInt (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: c10::SymInt + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: sym_is_contiguous + operator_name: sym_is_contiguous + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sym_is_contiguous(Tensor self, MemoryFormat memory_format=contiguous_format) -> SymBool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: false + name: memory_format + type: at::MemoryFormat + schema_order_cpp_signature: c10::SymBool (const at::Tensor &, at::MemoryFormat) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: false + name: memory_format + type: at::MemoryFormat + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: c10::SymBool + name: result + type: c10::SymBool + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: sym_numel + operator_name: sym_numel + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sym_numel(Tensor self) -> SymInt + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: c10::SymInt (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: c10::SymInt + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: sym_storage_offset + operator_name: sym_storage_offset + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sym_storage_offset(Tensor self) -> SymInt + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: c10::SymInt (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: c10::SymInt + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: slice + operator_name: slice + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::slice.Tensor(Tensor(a) self, int dim=0, SymInt? start=None, SymInt? end=None, SymInt step=1) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional, ::std::optional, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: slice_backward + operator_name: slice_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::slice_backward(Tensor grad_output, SymInt[] input_sizes, int dim, SymInt start, SymInt end, SymInt step) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: end + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: end + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: slice_inverse + operator_name: slice_inverse + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::slice_inverse(Tensor(a) self, Tensor src, int dim=0, SymInt? start=None, SymInt? end=None, SymInt step=1) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, ::std::optional, ::std::optional, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: slice_scatter + operator_name: slice_scatter + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::slice_scatter(Tensor self, Tensor src, int dim=0, SymInt? start=None, SymInt? end=None, SymInt step=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, ::std::optional, ::std::optional, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: select_scatter + operator_name: select_scatter + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::select_scatter(Tensor self, Tensor src, int dim, SymInt index) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: diagonal_scatter + operator_name: diagonal_scatter + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::diagonal_scatter(Tensor self, Tensor src, int offset=0, int dim1=0, int dim2=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: as_strided_scatter + operator_name: as_strided_scatter + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::as_strided_scatter(Tensor self, Tensor src, SymInt[] size, SymInt[] stride, SymInt? storage_offset=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: smm + operator_name: smm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::smm(Tensor self, Tensor mat2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: softmax + operator_name: softmax + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::softmax.int(Tensor self, int dim, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: softmax_out + operator_name: softmax + overload_name: int_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::softmax.int_out(Tensor self, int dim, ScalarType? dtype=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: softmax + operator_name: softmax + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::softmax.Dimname(Tensor self, Dimname dim, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _softmax + operator_name: _softmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_softmax(Tensor self, int dim, bool half_to_float) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _softmax_out + operator_name: _softmax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_softmax.out(Tensor self, int dim, bool half_to_float, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _softmax_backward_data + operator_name: _softmax_backward_data + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_softmax_backward_data(Tensor grad_output, Tensor output, int dim, ScalarType input_dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: input_dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: input_dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _softmax_backward_data_out + operator_name: _softmax_backward_data + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_softmax_backward_data.out(Tensor grad_output, Tensor output, int dim, ScalarType input_dtype, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: input_dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: input_dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unsafe_split + operator_name: unsafe_split + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::unsafe_split.Tensor(Tensor self, SymInt split_size, int dim=0) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: split_size + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: split_size + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: split + operator_name: split + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::split.Tensor(Tensor(a -> *) self, SymInt split_size, int dim=0) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: split_size + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: split_size + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: split + operator_name: split + overload_name: sizes + manual_kernel_registration: false + category_override: '' + schema_string: aten::split.sizes(Tensor(a -> *) self, SymInt[] split_size, int dim=0) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_size + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_size + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: unsafe_split_with_sizes + operator_name: unsafe_split_with_sizes + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unsafe_split_with_sizes(Tensor self, SymInt[] split_sizes, int dim=0) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_sizes + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_sizes + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: split_with_sizes + operator_name: split_with_sizes + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::split_with_sizes(Tensor(a -> *) self, SymInt[] split_sizes, int dim=0) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_sizes + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_sizes + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: hsplit + operator_name: hsplit + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::hsplit.int(Tensor(a -> *) self, int sections) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sections + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sections + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: hsplit + operator_name: hsplit + overload_name: array + manual_kernel_registration: false + category_override: '' + schema_string: aten::hsplit.array(Tensor(a -> *) self, int[] indices) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: indices + type: at::IntArrayRef + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: indices + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: vsplit + operator_name: vsplit + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::vsplit.int(Tensor(a -> *) self, int sections) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sections + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sections + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: vsplit + operator_name: vsplit + overload_name: array + manual_kernel_registration: false + category_override: '' + schema_string: aten::vsplit.array(Tensor(a -> *) self, int[] indices) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: indices + type: at::IntArrayRef + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: indices + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: dsplit + operator_name: dsplit + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::dsplit.int(Tensor(a -> *) self, int sections) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sections + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sections + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: dsplit + operator_name: dsplit + overload_name: array + manual_kernel_registration: false + category_override: '' + schema_string: aten::dsplit.array(Tensor(a -> *) self, int[] indices) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: indices + type: at::IntArrayRef + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: indices + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: squeeze + operator_name: squeeze + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze + operator_name: squeeze + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze.dim(Tensor(a) self, int dim) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze + operator_name: squeeze + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze.dimname(Tensor(a) self, Dimname dim) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: squeeze + operator_name: squeeze + overload_name: dims + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze.dims(Tensor(a) self, int[] dim) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze_ + operator_name: squeeze_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze_ + operator_name: squeeze_ + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze_.dim(Tensor(a!) self, int dim) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze_ + operator_name: squeeze_ + overload_name: dims + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze_.dims(Tensor(a!) self, int[] dim) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze_ + operator_name: squeeze_ + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze_.dimname(Tensor(a!) self, Dimname dim) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::Dimname) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: sspaddmm + operator_name: sspaddmm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sspaddmm(Tensor self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sspaddmm_out + operator_name: sspaddmm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sspaddmm.out(Tensor self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _chunk_cat + operator_name: _chunk_cat + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_chunk_cat(Tensor[] tensors, int dim, int num_chunks) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_chunks + type: int64_t + schema_order_cpp_signature: at::Tensor (at::TensorList, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_chunks + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _chunk_cat_out + operator_name: _chunk_cat + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_chunk_cat.out(Tensor[] tensors, int dim, int num_chunks, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_chunks + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::TensorList, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_chunks + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: stack + operator_name: stack + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::stack(Tensor[] tensors, int dim=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (at::TensorList, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: stack_out + operator_name: stack + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::stack.out(Tensor[] tensors, int dim=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::TensorList, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _stack + operator_name: _stack + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_stack(Tensor[] tensors, int dim=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (at::TensorList, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _stack_out + operator_name: _stack + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_stack.out(Tensor[] tensors, int dim=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::TensorList, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hstack + operator_name: hstack + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hstack(Tensor[] tensors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: hstack_out + operator_name: hstack + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hstack.out(Tensor[] tensors, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: vstack + operator_name: vstack + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::vstack(Tensor[] tensors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: vstack_out + operator_name: vstack + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::vstack.out(Tensor[] tensors, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: dstack + operator_name: dstack + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::dstack(Tensor[] tensors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: dstack_out + operator_name: dstack + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::dstack.out(Tensor[] tensors, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: stft + operator_name: stft + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::stft(Tensor self, int n_fft, int? hop_length=None, int? win_length=None, Tensor? window=None, bool normalized=False, bool? onesided=None, bool? return_complex=None, bool? align_to_window=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n_fft + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: hop_length + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: win_length + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: window + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: normalized + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: onesided + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: return_complex + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: align_to_window + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional, ::std::optional, const ::std::optional &, bool, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n_fft + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: hop_length + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: win_length + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: window + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: normalized + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: onesided + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: return_complex + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: align_to_window + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: stft + operator_name: stft + overload_name: center + manual_kernel_registration: false + category_override: '' + schema_string: aten::stft.center(Tensor self, int n_fft, int? hop_length=None, int? win_length=None, Tensor? window=None, bool center=True, str pad_mode="reflect", bool normalized=False, bool? onesided=None, bool? return_complex=None, bool? align_to_window=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n_fft + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: hop_length + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: win_length + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: window + type: const ::std::optional & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: center + type: bool + - annotation: null + default: '"reflect"' + dynamic_type: c10::string_view + is_nullable: false + name: pad_mode + type: c10::string_view + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: normalized + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: onesided + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: return_complex + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: align_to_window + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional, ::std::optional, const ::std::optional &, bool, c10::string_view, bool, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n_fft + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: hop_length + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: win_length + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: window + type: const ::std::optional & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: center + type: bool + - annotation: null + default: '"reflect"' + dynamic_type: c10::string_view + is_nullable: false + name: pad_mode + type: c10::string_view + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: normalized + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: onesided + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: return_complex + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: align_to_window + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: istft + operator_name: istft + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::istft(Tensor self, int n_fft, int? hop_length=None, int? win_length=None, Tensor? window=None, bool center=True, bool normalized=False, bool? onesided=None, int? length=None, bool return_complex=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n_fft + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: hop_length + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: win_length + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: window + type: const ::std::optional & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: center + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: normalized + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: onesided + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: length + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_complex + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional, ::std::optional, const ::std::optional &, bool, bool, ::std::optional, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n_fft + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: hop_length + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: win_length + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: window + type: const ::std::optional & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: center + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: normalized + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: onesided + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: length + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_complex + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: stride + operator_name: stride + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::stride.int(Tensor self, int dim) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: int64_t (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: stride + operator_name: stride + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::stride.Dimname(Tensor self, Dimname dim) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: int64_t (const at::Tensor &, at::Dimname) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: sym_stride + operator_name: sym_stride + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::sym_stride.int(Tensor self, int dim) -> SymInt + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: c10::SymInt (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: c10::SymInt + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: sum + operator_name: sum + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sum(Tensor self, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sum + operator_name: sum + overload_name: dim_IntList + manual_kernel_registration: false + category_override: '' + schema_string: aten::sum.dim_IntList(Tensor self, int[1]? dim, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sum + operator_name: sum + overload_name: dim_DimnameList + manual_kernel_registration: false + category_override: '' + schema_string: aten::sum.dim_DimnameList(Tensor self, Dimname[1] dim, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sum_out + operator_name: sum + overload_name: IntList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sum.IntList_out(Tensor self, int[1]? dim, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sum_out + operator_name: sum + overload_name: DimnameList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sum.DimnameList_out(Tensor self, Dimname[1] dim, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::DimnameList, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _nested_sum_backward + operator_name: _nested_sum_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_sum_backward(Tensor grad, Tensor self, int[1]? dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::OptionalIntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nansum + operator_name: nansum + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nansum(Tensor self, int[1]? dim=None, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nansum_out + operator_name: nansum + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nansum.out(Tensor self, int[1]? dim=None, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hash_tensor + operator_name: hash_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hash_tensor(Tensor self, int[1] dim=[], *, bool keepdim=False, int mode=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: mode + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: mode + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hash_tensor_out + operator_name: hash_tensor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hash_tensor.out(Tensor self, int[1] dim=[], *, bool keepdim=False, int mode=0, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: mode + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: mode + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sum_to_size + operator_name: sum_to_size + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sum_to_size(Tensor self, SymInt[] size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: sqrt + operator_name: sqrt + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sqrt(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sqrt_ + operator_name: sqrt_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sqrt_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sqrt_out + operator_name: sqrt + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sqrt.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: square + operator_name: square + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::square(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: square_ + operator_name: square_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::square_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: square_out + operator_name: square + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::square.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std + operator_name: std + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::std(Tensor self, bool unbiased=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std + operator_name: std + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::std.dim(Tensor self, int[1]? dim, bool unbiased=True, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std + operator_name: std + overload_name: correction + manual_kernel_registration: false + category_override: '' + schema_string: aten::std.correction(Tensor self, int[1]? dim=None, *, Scalar? correction=None, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: std_mean + operator_name: std_mean + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::std_mean(Tensor self, bool unbiased=True) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std_mean + operator_name: std_mean + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::std_mean.dim(Tensor self, int[1]? dim, bool unbiased=True, bool keepdim=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::OptionalIntArrayRef, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std_mean + operator_name: std_mean + overload_name: correction + manual_kernel_registration: false + category_override: '' + schema_string: aten::std_mean.correction(Tensor self, int[1]? dim=None, *, Scalar? correction=None, bool keepdim=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::OptionalIntArrayRef, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: std_mean + operator_name: std_mean + overload_name: names_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::std_mean.names_dim(Tensor self, Dimname[1] dim, bool unbiased=True, bool keepdim=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::DimnameList, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std_mean + operator_name: std_mean + overload_name: correction_names + manual_kernel_registration: false + category_override: '' + schema_string: aten::std_mean.correction_names(Tensor self, Dimname[1] dim, *, Scalar? correction=None, bool keepdim=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::DimnameList, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std_out + operator_name: std + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::std.out(Tensor self, int[1]? dim, bool unbiased=True, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std_out + operator_name: std + overload_name: correction_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::std.correction_out(Tensor self, int[1]? dim=None, *, Scalar? correction=None, bool keepdim=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, const ::std::optional &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: std + operator_name: std + overload_name: names_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::std.names_dim(Tensor self, Dimname[1] dim, bool unbiased=True, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std_out + operator_name: std + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::std.names_out(Tensor self, Dimname[1] dim, bool unbiased=True, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::DimnameList, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std + operator_name: std + overload_name: correction_names + manual_kernel_registration: false + category_override: '' + schema_string: aten::std.correction_names(Tensor self, Dimname[1] dim, *, Scalar? correction=None, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: std_out + operator_name: std + overload_name: correction_names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::std.correction_names_out(Tensor self, Dimname[1] dim, *, Scalar? correction=None, bool keepdim=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::DimnameList, const ::std::optional &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: prod + operator_name: prod + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::prod(Tensor self, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: prod + operator_name: prod + overload_name: dim_int + manual_kernel_registration: false + category_override: '' + schema_string: aten::prod.dim_int(Tensor self, int dim, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: prod_out + operator_name: prod + overload_name: int_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::prod.int_out(Tensor self, int dim, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: prod + operator_name: prod + overload_name: dim_Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::prod.dim_Dimname(Tensor self, Dimname dim, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: prod_out + operator_name: prod + overload_name: Dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::prod.Dimname_out(Tensor self, Dimname dim, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Dimname, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: t + operator_name: t + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::t(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: t_ + operator_name: t_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::t_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: tan + operator_name: tan + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::tan(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tan_ + operator_name: tan_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::tan_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tan_out + operator_name: tan + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::tan.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tanh + operator_name: tanh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::tanh(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tanh_ + operator_name: tanh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::tanh_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tanh_out + operator_name: tanh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::tanh.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tensordot + operator_name: tensordot + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::tensordot(Tensor self, Tensor other, int[] dims_self, int[] dims_other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims_self + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims_other + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims_self + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims_other + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: tensordot_out + operator_name: tensordot + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::tensordot.out(Tensor self, Tensor other, int[] dims_self, int[] dims_other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims_self + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims_other + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims_self + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims_other + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: threshold + operator_name: threshold + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::threshold(Tensor self, Scalar threshold, Scalar value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: threshold_ + operator_name: threshold_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::threshold_(Tensor(a!) self, Scalar threshold, Scalar value) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: threshold_out + operator_name: threshold + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::threshold.out(Tensor self, Scalar threshold, Scalar value, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: threshold_backward_out + operator_name: threshold_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::threshold_backward.grad_input(Tensor grad_output, Tensor self, Scalar threshold, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: threshold_backward + operator_name: threshold_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::threshold_backward(Tensor grad_output, Tensor self, Scalar threshold) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tile + operator_name: tile + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::tile(Tensor self, SymInt[] dims) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: transpose + operator_name: transpose + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::transpose.int(Tensor(a) self, int dim0, int dim1) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: transpose + operator_name: transpose + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::transpose.Dimname(Tensor(a) self, Dimname dim0, Dimname dim1) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim0 + type: at::Dimname + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim1 + type: at::Dimname + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, at::Dimname) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim0 + type: at::Dimname + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim1 + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: _mkldnn_transpose + operator_name: _mkldnn_transpose + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_mkldnn_transpose(Tensor self, int dim0, int dim1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: transpose_ + operator_name: transpose_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::transpose_(Tensor(a!) self, int dim0, int dim1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _mkldnn_transpose_ + operator_name: _mkldnn_transpose_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_mkldnn_transpose_(Tensor(a!) self, int dim0, int dim1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: one_hot + operator_name: one_hot + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::one_hot(Tensor self, int num_classes=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: num_classes + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: num_classes + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: flip + operator_name: flip + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::flip(Tensor self, int[] dims) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fliplr + operator_name: fliplr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fliplr(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: flipud + operator_name: flipud + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::flipud(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: roll + operator_name: roll + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::roll(Tensor self, SymInt[1] shifts, int[1] dims=[]) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shifts + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + size: 1 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shifts + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + size: 1 + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rot90 + operator_name: rot90 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rot90(Tensor self, int k=1, int[] dims=[0,1]) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: '{0,1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: '{0,1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: trapezoid + operator_name: trapezoid + overload_name: x + manual_kernel_registration: false + category_override: '' + schema_string: aten::trapezoid.x(Tensor y, Tensor x, *, int dim=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: trapezoid + operator_name: trapezoid + overload_name: dx + manual_kernel_registration: false + category_override: '' + schema_string: aten::trapezoid.dx(Tensor y, *, Scalar dx=1, int dim=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: dx + type: const at::Scalar & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: dx + type: const at::Scalar & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: trapz + operator_name: trapz + overload_name: x + manual_kernel_registration: false + category_override: '' + schema_string: aten::trapz.x(Tensor y, Tensor x, *, int dim=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: trapz + operator_name: trapz + overload_name: dx + manual_kernel_registration: false + category_override: '' + schema_string: aten::trapz.dx(Tensor y, *, float dx=1, int dim=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dx + type: double + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dx + type: double + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _transform_bias_rescale_qkv + operator_name: _transform_bias_rescale_qkv + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_transform_bias_rescale_qkv(Tensor qkv, Tensor qkv_bias, int num_heads) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_heads + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_heads + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_from_mask + operator_name: _nested_tensor_from_mask + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_from_mask(Tensor t, Tensor mask, bool mask_check=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: t + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: mask_check + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: t + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: mask_check + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_from_mask_left_aligned + operator_name: _nested_tensor_from_mask_left_aligned + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_from_mask_left_aligned(Tensor t, Tensor mask) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: t + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: t + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_from_padded + operator_name: _nested_from_padded + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_from_padded(Tensor padded, Tensor cpu_nested_shape_example, bool fuse_transform_0213=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: padded + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cpu_nested_shape_example + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: fuse_transform_0213 + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: padded + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cpu_nested_shape_example + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: fuse_transform_0213 + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_size + operator_name: _nested_tensor_size + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_size(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_strides + operator_name: _nested_tensor_strides + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_strides(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_storage_offsets + operator_name: _nested_tensor_storage_offsets + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_storage_offsets(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_from_padded_and_nested_example + operator_name: _nested_from_padded_and_nested_example + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_from_padded_and_nested_example(Tensor padded, Tensor nt_example) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: padded + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nt_example + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: padded + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nt_example + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_view_from_buffer + operator_name: _nested_view_from_buffer + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_view_from_buffer(Tensor(a) self, Tensor nested_size, Tensor nested_strides, Tensor offsets) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_strides + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_strides + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_view_from_buffer_copy + operator_name: _nested_view_from_buffer_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_view_from_buffer_copy(Tensor self, Tensor nested_size, Tensor nested_strides, Tensor offsets) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_strides + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_strides + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_view_from_jagged + operator_name: _nested_view_from_jagged + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_view_from_jagged(Tensor(a) self, Tensor offsets, Tensor dummy, Tensor? lengths=None, int ragged_idx=1, Tensor? min_seqlen=None, Tensor? max_seqlen=None) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: lengths + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: ragged_idx + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min_seqlen + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max_seqlen + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: lengths + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: ragged_idx + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min_seqlen + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max_seqlen + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_view_from_jagged_copy + operator_name: _nested_view_from_jagged_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_view_from_jagged_copy(Tensor self, Tensor offsets, Tensor dummy, Tensor? lengths=None, int ragged_idx=1, Tensor? min_seqlen=None, Tensor? max_seqlen=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: lengths + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: ragged_idx + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min_seqlen + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max_seqlen + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: lengths + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: ragged_idx + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min_seqlen + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max_seqlen + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_get_values + operator_name: _nested_get_values + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_get_values(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_get_values_copy + operator_name: _nested_get_values_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_get_values_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_get_offsets + operator_name: _nested_get_offsets + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_get_offsets(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_get_lengths + operator_name: _nested_get_lengths + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_get_lengths(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_get_ragged_idx + operator_name: _nested_get_ragged_idx + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_get_ragged_idx(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_get_min_seqlen + operator_name: _nested_get_min_seqlen + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_get_min_seqlen(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_get_max_seqlen + operator_name: _nested_get_max_seqlen + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_get_max_seqlen(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_get_jagged_dummy + operator_name: _nested_get_jagged_dummy + overload_name: '' + manual_kernel_registration: false + category_override: dummy + schema_string: aten::_nested_get_jagged_dummy(Tensor any) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: any + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: any + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_compute_contiguous_strides_offsets + operator_name: _nested_compute_contiguous_strides_offsets + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_compute_contiguous_strides_offsets(Tensor nested_size) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_size + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_size + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _trilinear + operator_name: _trilinear + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_trilinear(Tensor i1, Tensor i2, Tensor i3, int[] expand1, int[] expand2, int[] expand3, int[] sumdim, int unroll_dim=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sumdim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: unroll_dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sumdim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: unroll_dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: triplet_margin_loss + operator_name: triplet_margin_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::triplet_margin_loss(Tensor anchor, Tensor positive, Tensor negative, float margin=1.0, float p=2, float eps=1e-06, bool swap=False, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: anchor + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: positive + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: negative + type: const at::Tensor & + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: margin + type: double + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: 1.0e-06 + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: swap + type: bool + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, double, double, double, bool, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: anchor + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: positive + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: negative + type: const at::Tensor & + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: margin + type: double + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: 1.0e-06 + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: swap + type: bool + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: trunc + operator_name: trunc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::trunc(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: trunc_ + operator_name: trunc_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::trunc_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: trunc_out + operator_name: trunc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::trunc.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fix + operator_name: fix + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fix(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fix_ + operator_name: fix_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fix_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fix_out + operator_name: fix + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fix.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: type_as + operator_name: type_as + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::type_as(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _has_compatible_shallow_copy_type + operator_name: _has_compatible_shallow_copy_type + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_has_compatible_shallow_copy_type(Tensor self, Tensor from) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: from + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: from + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _unique + operator_name: _unique + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unique(Tensor self, bool sorted=True, bool return_inverse=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unique_dim + operator_name: unique_dim + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unique_dim(Tensor self, int dim, bool sorted=True, bool return_inverse=False, bool return_counts=False) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unique_consecutive + operator_name: unique_consecutive + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unique_consecutive(Tensor self, bool return_inverse=False, bool return_counts=False, int? dim=None) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unique_dim_consecutive + operator_name: unique_dim_consecutive + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unique_dim_consecutive(Tensor self, int dim, bool return_inverse=False, bool return_counts=False) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _unique2 + operator_name: _unique2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unique2(Tensor self, bool sorted=True, bool return_inverse=False, bool return_counts=False) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _unsafe_view + operator_name: _unsafe_view + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unsafe_view(Tensor self, SymInt[] size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unsqueeze + operator_name: unsqueeze + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unsqueeze(Tensor(a) self, int dim) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: unsqueeze_ + operator_name: unsqueeze_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unsqueeze_(Tensor(a!) self, int dim) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: vander + operator_name: vander + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::vander(Tensor x, int? N=None, bool increasing=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: N + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: increasing + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: N + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: increasing + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var + operator_name: var + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::var(Tensor self, bool unbiased=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var + operator_name: var + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::var.dim(Tensor self, int[1]? dim, bool unbiased=True, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var + operator_name: var + overload_name: correction + manual_kernel_registration: false + category_override: '' + schema_string: aten::var.correction(Tensor self, int[1]? dim=None, *, Scalar? correction=None, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: var_out + operator_name: var + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::var.out(Tensor self, int[1]? dim, bool unbiased=True, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var_out + operator_name: var + overload_name: correction_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::var.correction_out(Tensor self, int[1]? dim=None, *, Scalar? correction=None, bool keepdim=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, const ::std::optional &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: var + operator_name: var + overload_name: names_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::var.names_dim(Tensor self, Dimname[1] dim, bool unbiased=True, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var_out + operator_name: var + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::var.names_out(Tensor self, Dimname[1] dim, bool unbiased=True, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::DimnameList, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var + operator_name: var + overload_name: correction_names + manual_kernel_registration: false + category_override: '' + schema_string: aten::var.correction_names(Tensor self, Dimname[1] dim, *, Scalar? correction=None, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::DimnameList, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var_out + operator_name: var + overload_name: correction_names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::var.correction_names_out(Tensor self, Dimname[1] dim, *, Scalar? correction=None, bool keepdim=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::DimnameList, const ::std::optional &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var_mean + operator_name: var_mean + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::var_mean(Tensor self, bool unbiased=True) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var_mean + operator_name: var_mean + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::var_mean.dim(Tensor self, int[1]? dim, bool unbiased=True, bool keepdim=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::OptionalIntArrayRef, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var_mean + operator_name: var_mean + overload_name: correction + manual_kernel_registration: false + category_override: '' + schema_string: aten::var_mean.correction(Tensor self, int[1]? dim=None, *, Scalar? correction=None, bool keepdim=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::OptionalIntArrayRef, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: var_mean + operator_name: var_mean + overload_name: names_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::var_mean.names_dim(Tensor self, Dimname[1] dim, bool unbiased=True, bool keepdim=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::DimnameList, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unbiased + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: var_mean + operator_name: var_mean + overload_name: correction_names + manual_kernel_registration: false + category_override: '' + schema_string: aten::var_mean.correction_names(Tensor self, Dimname[1] dim, *, Scalar? correction=None, bool keepdim=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::DimnameList, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: view_as + operator_name: view_as + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_as(Tensor(a) self, Tensor other) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: where + operator_name: where + overload_name: self + manual_kernel_registration: false + category_override: '' + schema_string: aten::where.self(Tensor condition, Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: where_out + operator_name: where + overload_name: self_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::where.self_out(Tensor condition, Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: where + operator_name: where + overload_name: ScalarSelf + manual_kernel_registration: false + category_override: '' + schema_string: aten::where.ScalarSelf(Tensor condition, Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: where + operator_name: where + overload_name: ScalarOther + manual_kernel_registration: false + category_override: '' + schema_string: aten::where.ScalarOther(Tensor condition, Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: where + operator_name: where + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::where.Scalar(Tensor condition, Scalar self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: where + operator_name: where + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::where(Tensor condition) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + schema_order_cpp_signature: ::std::vector (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: condition + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: norm_except_dim + operator_name: norm_except_dim + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm_except_dim(Tensor v, int pow=2, int dim=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: int64_t + is_nullable: false + name: pow + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: int64_t + is_nullable: false + name: pow + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _weight_norm + operator_name: _weight_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_weight_norm(Tensor v, Tensor g, int dim=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: g + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: g + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _weight_norm_interface + operator_name: _weight_norm_interface + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_weight_norm_interface(Tensor v, Tensor g, int dim=0) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: g + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: g + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _weight_norm_interface_backward + operator_name: _weight_norm_interface_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_weight_norm_interface_backward(Tensor grad_w, Tensor saved_v, Tensor saved_g, Tensor saved_norms, int dim) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_w + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_g + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_norms + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_w + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_g + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_norms + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _weight_norm_differentiable_backward + operator_name: _weight_norm_differentiable_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_weight_norm_differentiable_backward(Tensor grad_w, Tensor saved_v, Tensor saved_g, Tensor saved_norms, int dim) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_w + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_g + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_norms + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_w + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_g + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_norms + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: zeros + operator_name: zeros + overload_name: names + manual_kernel_registration: false + category_override: '' + schema_string: aten::zeros.names(int[] size, *, Dimname[]? names, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _efficientzerotensor + operator_name: _efficientzerotensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_efficientzerotensor(SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: zeros + operator_name: zeros + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::zeros(SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: zeros_out + operator_name: zeros + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::zeros.out(SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: zeros_like + operator_name: zeros_like + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::zeros_like(Tensor self, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _standard_gamma_grad + operator_name: _standard_gamma_grad + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_standard_gamma_grad(Tensor self, Tensor output) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _standard_gamma + operator_name: _standard_gamma + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_standard_gamma(Tensor self, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _dirichlet_grad + operator_name: _dirichlet_grad + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_dirichlet_grad(Tensor x, Tensor alpha, Tensor total) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: alpha + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: alpha + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sample_dirichlet + operator_name: _sample_dirichlet + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sample_dirichlet(Tensor self, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: poisson + operator_name: poisson + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::poisson(Tensor self, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: binomial + operator_name: binomial + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::binomial(Tensor count, Tensor prob, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: count + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: prob + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: count + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: prob + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_norm + operator_name: native_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_norm(Tensor self, Scalar p=2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_norm + operator_name: native_norm + overload_name: ScalarOpt_dim_dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_norm.ScalarOpt_dim_dtype(Tensor self, Scalar? p, int[1] dim, bool keepdim, ScalarType? dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, at::IntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _batch_norm_with_update + operator_name: _batch_norm_with_update + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_batch_norm_with_update(Tensor input, Tensor? weight, Tensor? bias, Tensor(a!) running_mean, Tensor(b!) running_var, float momentum, float eps) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, at::Tensor &, at::Tensor &, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _batch_norm_with_update_out + operator_name: _batch_norm_with_update + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_batch_norm_with_update.out(Tensor input, Tensor? weight, Tensor? bias, Tensor(a!) running_mean, Tensor(b!) running_var, float momentum, float eps, *, Tensor(d!) out, Tensor(e!) save_mean, Tensor(f!) save_invstd, Tensor(g!) reserve) -> (Tensor(d!), Tensor(e!), Tensor(f!), Tensor(g!)) + arguments: + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: save_mean + output: true + type: at::Tensor & + - allocate: true + annotation: f! + dynamic_type: at::Tensor + is_nullable: false + name: save_invstd + output: true + type: at::Tensor & + - allocate: true + annotation: g! + dynamic_type: at::Tensor + is_nullable: false + name: reserve + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, at::Tensor &, at::Tensor &, double, double, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: save_mean + output: true + type: at::Tensor & + - allocate: true + annotation: f! + dynamic_type: at::Tensor + is_nullable: false + name: save_invstd + output: true + type: at::Tensor & + - allocate: true + annotation: g! + dynamic_type: at::Tensor + is_nullable: false + name: reserve + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + - dynamic_type: at::Tensor + name: save_mean + type: at::Tensor & + - dynamic_type: at::Tensor + name: save_invstd + type: at::Tensor & + - dynamic_type: at::Tensor + name: reserve + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _batch_norm_no_update + operator_name: _batch_norm_no_update + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_batch_norm_no_update(Tensor input, Tensor? weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, float momentum, float eps) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_backward + operator_name: batch_norm_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_backward(Tensor grad_out, Tensor input, Tensor weight, Tensor? running_mean, Tensor? running_var, Tensor? save_mean, Tensor? save_var, bool update, float eps, bool[3] output_mask, Tensor reserve) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: update + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserve + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, ::std::array, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: update + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserve + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_sum + operator_name: _sparse_sum + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_sum(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_sum + operator_name: _sparse_sum + overload_name: dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_sum.dtype(Tensor self, *, ScalarType dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_sum + operator_name: _sparse_sum + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_sum.dim(Tensor self, int[1] dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_sum + operator_name: _sparse_sum + overload_name: dim_dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_sum.dim_dtype(Tensor self, int[1] dim, *, ScalarType dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_sum_backward + operator_name: _sparse_sum_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_sum_backward(Tensor grad, Tensor self, int[] dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_csr_sum + operator_name: _sparse_csr_sum + overload_name: dim_dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_csr_sum.dim_dtype(Tensor self, int[1] dim, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_csr_prod + operator_name: _sparse_csr_prod + overload_name: dim_dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_csr_prod.dim_dtype(Tensor self, int[1] dim, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_softmax + operator_name: _sparse_softmax + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_softmax.int(Tensor self, int dim, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_softmax + operator_name: _sparse_softmax + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_softmax.Dimname(Tensor self, Dimname dim, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_softmax + operator_name: _sparse_softmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_softmax(Tensor self, int dim, bool half_to_float) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_softmax_backward_data + operator_name: _sparse_softmax_backward_data + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_softmax_backward_data(Tensor grad_output, Tensor output, int dim, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_log_softmax + operator_name: _sparse_log_softmax + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_log_softmax.int(Tensor self, int dim, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_log_softmax + operator_name: _sparse_log_softmax + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_log_softmax.Dimname(Tensor self, Dimname dim, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_log_softmax + operator_name: _sparse_log_softmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_log_softmax(Tensor self, int dim, bool half_to_float) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_log_softmax_backward_data + operator_name: _sparse_log_softmax_backward_data + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_log_softmax_backward_data(Tensor grad_output, Tensor output, int dim, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _spdiags + operator_name: _spdiags + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_spdiags(Tensor diagonals, Tensor offsets, int[] shape, Layout? layout=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: diagonals + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shape + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + name: layout + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: diagonals + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shape + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + name: layout + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: norm + operator_name: norm + overload_name: ScalarOpt_dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.ScalarOpt_dtype(Tensor self, Scalar? p, *, ScalarType dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: norm + operator_name: norm + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.Scalar(Tensor self, Scalar p=2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: norm + operator_name: norm + overload_name: ScalarOpt_dim_dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.ScalarOpt_dim_dtype(Tensor self, Scalar? p, int[1] dim, bool keepdim, *, ScalarType dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, at::IntArrayRef, bool, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: norm + operator_name: norm + overload_name: ScalarOpt_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.ScalarOpt_dim(Tensor self, Scalar? p, int[1] dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: norm_out + operator_name: norm + overload_name: dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.dtype_out(Tensor self, Scalar? p, int[1] dim, bool keepdim, *, ScalarType dtype, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, at::IntArrayRef, bool, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: norm_out + operator_name: norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.out(Tensor self, Scalar? p, int[1] dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: norm + operator_name: norm + overload_name: names_ScalarOpt_dim_dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.names_ScalarOpt_dim_dtype(Tensor self, Scalar? p, Dimname[1] dim, bool keepdim, *, ScalarType dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, at::DimnameList, bool, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: norm + operator_name: norm + overload_name: names_ScalarOpt_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.names_ScalarOpt_dim(Tensor self, Scalar? p, Dimname[1] dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, at::DimnameList, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: norm_out + operator_name: norm + overload_name: names_dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.names_dtype_out(Tensor self, Scalar? p, Dimname[1] dim, bool keepdim, *, ScalarType dtype, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, at::DimnameList, bool, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: norm_out + operator_name: norm + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.names_out(Tensor self, Scalar? p, Dimname[1] dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, at::DimnameList, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: false + name: dim + size: 1 + type: at::DimnameList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: frexp + operator_name: frexp + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::frexp.Tensor(Tensor self) -> (Tensor mantissa, Tensor exponent) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: mantissa + name: mantissa + type: at::Tensor + - dynamic_type: at::Tensor + field_name: exponent + name: exponent + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: frexp_out + operator_name: frexp + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::frexp.Tensor_out(Tensor self, *, Tensor(a!) mantissa, Tensor(b!) exponent) -> (Tensor(a!) mantissa, Tensor(b!) exponent) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: mantissa + is_nullable: false + name: mantissa + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: exponent + is_nullable: false + name: exponent + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: mantissa + is_nullable: false + name: mantissa + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: exponent + is_nullable: false + name: exponent + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: mantissa + name: mantissa + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: exponent + name: exponent + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: frobenius_norm + operator_name: frobenius_norm + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::frobenius_norm.dim(Tensor self, int[1] dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: frobenius_norm_out + operator_name: frobenius_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::frobenius_norm.out(Tensor self, int[1] dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nuclear_norm + operator_name: nuclear_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nuclear_norm(Tensor self, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nuclear_norm_out + operator_name: nuclear_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nuclear_norm.out(Tensor self, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nuclear_norm + operator_name: nuclear_norm + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::nuclear_norm.dim(Tensor self, int[2] dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nuclear_norm_out + operator_name: nuclear_norm + overload_name: dim_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nuclear_norm.dim_out(Tensor self, int[2] dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: clone + operator_name: clone + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::clone(Tensor self, *, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: positive + operator_name: positive + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::positive(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: resize_as_ + operator_name: resize_as_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::resize_as_(Tensor(a!) self, Tensor the_template, *, MemoryFormat? memory_format=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: const at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: resize_as_sparse_ + operator_name: resize_as_sparse_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::resize_as_sparse_(Tensor(a!) self, Tensor the_template) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: const at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: zero_ + operator_name: zero_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::zero_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sub_out + operator_name: sub + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sub.out(Tensor self, Tensor other, *, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sub + operator_name: sub + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::sub.Tensor(Tensor self, Tensor other, *, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sub_ + operator_name: sub_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::sub_.Tensor(Tensor(a!) self, Tensor other, *, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sub + operator_name: sub + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::sub.Scalar(Tensor self, Scalar other, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sub_ + operator_name: sub_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::sub_.Scalar(Tensor(a!) self, Scalar other, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: subtract_out + operator_name: subtract + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::subtract.out(Tensor self, Tensor other, *, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: subtract + operator_name: subtract + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::subtract.Tensor(Tensor self, Tensor other, *, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: subtract_ + operator_name: subtract_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::subtract_.Tensor(Tensor(a!) self, Tensor other, *, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: subtract + operator_name: subtract + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::subtract.Scalar(Tensor self, Scalar other, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: subtract_ + operator_name: subtract_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::subtract_.Scalar(Tensor(a!) self, Scalar other, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rsub + operator_name: rsub + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::rsub.Tensor(Tensor self, Tensor other, *, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: heaviside_out + operator_name: heaviside + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::heaviside.out(Tensor self, Tensor values, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: heaviside + operator_name: heaviside + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::heaviside(Tensor self, Tensor values) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: heaviside_ + operator_name: heaviside_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::heaviside_(Tensor(a!) self, Tensor values) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rsub + operator_name: rsub + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::rsub.Scalar(Tensor self, Scalar other, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_addmm + operator_name: _sparse_addmm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_addmm(Tensor self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_sampled_addmm_out + operator_name: sparse_sampled_addmm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_sampled_addmm.out(Tensor self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_sampled_addmm + operator_name: sparse_sampled_addmm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_sampled_addmm(Tensor self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_mm_reduce_impl + operator_name: _sparse_mm_reduce_impl + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_mm_reduce_impl(Tensor self, Tensor other, str reduce) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_mm_reduce_impl_backward + operator_name: _sparse_mm_reduce_impl_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_mm_reduce_impl_backward(Tensor self, Tensor grad_out, Tensor weight, str reduce, Tensor arg_out, bool[2] output_mask) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: arg_out + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, c10::string_view, const at::Tensor &, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: arg_out + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addmm_out + operator_name: addmm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::addmm.out(Tensor self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addmm + operator_name: addmm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addmm(Tensor self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addmm + operator_name: addmm + overload_name: dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::addmm.dtype(Tensor self, Tensor mat1, Tensor mat2, ScalarType out_dtype, *, Scalar beta=1, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::ScalarType, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addmm_out + operator_name: addmm + overload_name: dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::addmm.dtype_out(Tensor self, Tensor mat1, Tensor mat2, ScalarType out_dtype, *, Scalar beta=1, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::ScalarType, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: out_dtype + type: at::ScalarType + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addmm_ + operator_name: addmm_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addmm_(Tensor(a!) self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _addmm_activation_out + operator_name: _addmm_activation + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_addmm_activation.out(Tensor self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1, bool use_gelu=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: use_gelu + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: use_gelu + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _addmm_activation + operator_name: _addmm_activation + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_addmm_activation(Tensor self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1, bool use_gelu=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: use_gelu + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: use_gelu + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_mm + operator_name: _scaled_mm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_mm(Tensor self, Tensor mat2, Tensor scale_a, Tensor scale_b, Tensor? bias=None, Tensor? scale_result=None, ScalarType? out_dtype=None, bool use_fast_accum=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_a + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_b + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: scale_result + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_a + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_b + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: scale_result + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_mm_out + operator_name: _scaled_mm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_mm.out(Tensor self, Tensor mat2, Tensor scale_a, Tensor scale_b, Tensor? bias=None, Tensor? scale_result=None, ScalarType? out_dtype=None, bool use_fast_accum=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_a + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_b + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: scale_result + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, ::std::optional, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_a + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_b + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: scale_result + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_mm_v2 + operator_name: _scaled_mm_v2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_mm_v2(Tensor self, Tensor mat2, Tensor[] scale_a, int[] recipe_a, int[] swizzle_a, Tensor[] scale_b, int[] recipe_b, int[] swizzle_b, Tensor? bias, ScalarType? out_dtype, int[] contraction_dim=[], bool use_fast_accum=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_a + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_b + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_b + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_b + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: contraction_dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::TensorList, at::IntArrayRef, at::IntArrayRef, at::TensorList, at::IntArrayRef, at::IntArrayRef, const ::std::optional &, ::std::optional, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_a + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_b + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_b + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_b + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: contraction_dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_mm_v2_out + operator_name: _scaled_mm_v2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_mm_v2.out(Tensor self, Tensor mat2, Tensor[] scale_a, int[] recipe_a, int[] swizzle_a, Tensor[] scale_b, int[] recipe_b, int[] swizzle_b, Tensor? bias, ScalarType? out_dtype, int[] contraction_dim=[], bool use_fast_accum=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_a + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_b + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_b + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_b + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: contraction_dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::TensorList, at::IntArrayRef, at::IntArrayRef, at::TensorList, at::IntArrayRef, at::IntArrayRef, const ::std::optional &, ::std::optional, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_a + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_b + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_b + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_b + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: contraction_dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_grouped_mm + operator_name: _scaled_grouped_mm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_grouped_mm(Tensor self, Tensor mat2, Tensor scale_a, Tensor scale_b, Tensor? offs=None, Tensor? bias=None, Tensor? scale_result=None, ScalarType? out_dtype=None, bool use_fast_accum=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_a + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_b + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: offs + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: scale_result + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_a + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale_b + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: offs + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: scale_result + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_grouped_mm_v2 + operator_name: _scaled_grouped_mm_v2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_grouped_mm_v2(Tensor self, Tensor mat2, Tensor[] scale_a, int[] recipe_a, int[] swizzle_a, Tensor[] scale_b, int[] recipe_b, int[] swizzle_b, Tensor? offs=None, Tensor? bias=None, ScalarType? out_dtype=None, int[] contraction_dim=[], bool use_fast_accum=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_a + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_b + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_b + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_b + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: offs + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: contraction_dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::TensorList, at::IntArrayRef, at::IntArrayRef, at::TensorList, at::IntArrayRef, at::IntArrayRef, const ::std::optional &, const ::std::optional &, ::std::optional, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_a + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_a + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: scale_b + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: recipe_b + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: swizzle_b + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: offs + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: contraction_dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: use_fast_accum + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _grouped_mm + operator_name: _grouped_mm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_grouped_mm(Tensor self, Tensor mat2, Tensor? offs=None, Tensor? bias=None, ScalarType? out_dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: offs + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: offs + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: out_dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_compressed_tensor_with_dims + operator_name: _sparse_compressed_tensor_with_dims + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_compressed_tensor_with_dims(int nnz, int dense_dim, int[] size, int[] blocksize, ScalarType index_dtype, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: nnz + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + type: at::IntArrayRef + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: index_dtype + type: at::ScalarType + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, int64_t, at::IntArrayRef, at::IntArrayRef, at::ScalarType, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: nnz + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + type: at::IntArrayRef + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: index_dtype + type: at::ScalarType + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_compressed_tensor + operator_name: sparse_compressed_tensor + overload_name: comp_plain_value_size + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_compressed_tensor.comp_plain_value_size(Tensor compressed_indices, Tensor plain_indices, Tensor values, SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: plain_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: plain_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_csr_tensor + operator_name: sparse_csr_tensor + overload_name: crow_col_value_size + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_csr_tensor.crow_col_value_size(Tensor crow_indices, Tensor col_indices, Tensor values, int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sparse_csc_tensor + operator_name: sparse_csc_tensor + overload_name: ccol_row_value_size + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_csc_tensor.ccol_row_value_size(Tensor ccol_indices, Tensor row_indices, Tensor values, int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sparse_bsr_tensor + operator_name: sparse_bsr_tensor + overload_name: crow_col_value_size + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_bsr_tensor.crow_col_value_size(Tensor crow_indices, Tensor col_indices, Tensor values, int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sparse_bsc_tensor + operator_name: sparse_bsc_tensor + overload_name: ccol_row_value_size + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_bsc_tensor.ccol_row_value_size(Tensor ccol_indices, Tensor row_indices, Tensor values, int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sparse_compressed_tensor + operator_name: sparse_compressed_tensor + overload_name: comp_plain_value + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_compressed_tensor.comp_plain_value(Tensor compressed_indices, Tensor plain_indices, Tensor values, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: plain_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: plain_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_csr_tensor + operator_name: sparse_csr_tensor + overload_name: crow_col_value + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_csr_tensor.crow_col_value(Tensor crow_indices, Tensor col_indices, Tensor values, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sparse_csc_tensor + operator_name: sparse_csc_tensor + overload_name: ccol_row_value + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_csc_tensor.ccol_row_value(Tensor ccol_indices, Tensor row_indices, Tensor values, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sparse_bsr_tensor + operator_name: sparse_bsr_tensor + overload_name: crow_col_value + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_bsr_tensor.crow_col_value(Tensor crow_indices, Tensor col_indices, Tensor values, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sparse_bsc_tensor + operator_name: sparse_bsc_tensor + overload_name: ccol_row_value + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_bsc_tensor.ccol_row_value(Tensor ccol_indices, Tensor row_indices, Tensor values, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_compressed_tensor_unsafe + operator_name: _sparse_compressed_tensor_unsafe + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_compressed_tensor_unsafe(Tensor compressed_indices, Tensor plain_indices, Tensor values, SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: plain_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: plain_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_csr_tensor_unsafe + operator_name: _sparse_csr_tensor_unsafe + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_csr_tensor_unsafe(Tensor crow_indices, Tensor col_indices, Tensor values, int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_csc_tensor_unsafe + operator_name: _sparse_csc_tensor_unsafe + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_csc_tensor_unsafe(Tensor ccol_indices, Tensor row_indices, Tensor values, int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_bsr_tensor_unsafe + operator_name: _sparse_bsr_tensor_unsafe + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_bsr_tensor_unsafe(Tensor crow_indices, Tensor col_indices, Tensor values, int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_bsc_tensor_unsafe + operator_name: _sparse_bsc_tensor_unsafe + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_bsc_tensor_unsafe(Tensor ccol_indices, Tensor row_indices, Tensor values, int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sparse_coo_tensor + operator_name: sparse_coo_tensor + overload_name: size + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_coo_tensor.size(int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_coo_tensor + operator_name: sparse_coo_tensor + overload_name: indices + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_coo_tensor.indices(Tensor indices, Tensor values, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, bool? is_coalesced=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: is_coalesced + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: is_coalesced + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sparse_coo_tensor + operator_name: sparse_coo_tensor + overload_name: indices_size + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_coo_tensor.indices_size(Tensor indices, Tensor values, int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, bool? is_coalesced=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: is_coalesced + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: is_coalesced + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_coo_tensor_unsafe + operator_name: _sparse_coo_tensor_unsafe + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_coo_tensor_unsafe(Tensor indices, Tensor values, SymInt[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, bool? is_coalesced=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: is_coalesced + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: is_coalesced + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _validate_sparse_coo_tensor_args + operator_name: _validate_sparse_coo_tensor_args + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_validate_sparse_coo_tensor_args(Tensor indices, Tensor values, int[] size, bool? is_coalesced=None, bool? check_pinning=None) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: is_coalesced + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + schema_order_cpp_signature: void (const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: is_coalesced + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _validate_sparse_compressed_tensor_args + operator_name: _validate_sparse_compressed_tensor_args + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_validate_sparse_compressed_tensor_args(Tensor compressed_indices, Tensor plain_indices, Tensor values, int[] size, Layout layout, bool? check_pinning=None) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: plain_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Layout + is_nullable: false + name: layout + type: at::Layout + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + schema_order_cpp_signature: void (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::Layout, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: compressed_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: plain_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Layout + is_nullable: false + name: layout + type: at::Layout + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _validate_sparse_csr_tensor_args + operator_name: _validate_sparse_csr_tensor_args + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_validate_sparse_csr_tensor_args(Tensor crow_indices, Tensor col_indices, Tensor values, int[] size, bool? check_pinning=None) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + schema_order_cpp_signature: void (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _validate_sparse_csc_tensor_args + operator_name: _validate_sparse_csc_tensor_args + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_validate_sparse_csc_tensor_args(Tensor ccol_indices, Tensor row_indices, Tensor values, int[] size, bool? check_pinning=None) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + schema_order_cpp_signature: void (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _validate_sparse_bsr_tensor_args + operator_name: _validate_sparse_bsr_tensor_args + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_validate_sparse_bsr_tensor_args(Tensor crow_indices, Tensor col_indices, Tensor values, int[] size, bool? check_pinning=None) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + schema_order_cpp_signature: void (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _validate_sparse_bsc_tensor_args + operator_name: _validate_sparse_bsc_tensor_args + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_validate_sparse_bsc_tensor_args(Tensor ccol_indices, Tensor row_indices, Tensor values, int[] size, bool? check_pinning=None) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + schema_order_cpp_signature: void (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ccol_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: row_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: check_pinning + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _sparse_coo_tensor_with_dims + operator_name: _sparse_coo_tensor_with_dims + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_coo_tensor_with_dims(int sparse_dim, int dense_dim, int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, int64_t, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_coo_tensor_with_dims_and_tensors + operator_name: _sparse_coo_tensor_with_dims_and_tensors + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_coo_tensor_with_dims_and_tensors(int sparse_dim, int dense_dim, SymInt[] size, Tensor indices, Tensor values, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=False, bool? is_coalesced=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: is_coalesced + type: ::std::optional + schema_order_cpp_signature: at::Tensor (int64_t, int64_t, at::IntArrayRef, const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: is_coalesced + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_resize_ + operator_name: sparse_resize_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_resize_(Tensor(a!) self, int[] size, int sparse_dim, int dense_dim) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: const at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_resize_and_clear_ + operator_name: sparse_resize_and_clear_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_resize_and_clear_(Tensor(a!) self, int[] size, int sparse_dim, int dense_dim) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: const at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_mask + operator_name: sparse_mask + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_mask(Tensor self, Tensor mask) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_mask_projection + operator_name: _sparse_mask_projection + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_mask_projection(Tensor self, Tensor mask, bool accumulate_matches=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate_matches + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate_matches + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _to_cpu + operator_name: _to_cpu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_cpu(Tensor[] tensors) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: to_dense + operator_name: to_dense + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_dense(Tensor self, ScalarType? dtype=None, *, bool? masked_grad=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: masked_grad + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: masked_grad + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _to_dense + operator_name: _to_dense + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_dense(Tensor self, ScalarType? dtype=None, bool? masked_grad=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: masked_grad + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: masked_grad + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_dense_backward + operator_name: to_dense_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_dense_backward(Tensor grad, Tensor input, bool? masked_grad=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: masked_grad + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: masked_grad + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sparse_dim + operator_name: sparse_dim + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_dim(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _dimI + operator_name: _dimI + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_dimI(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: dense_dim + operator_name: dense_dim + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::dense_dim(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _dimV + operator_name: _dimV + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_dimV(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nnz + operator_name: _nnz + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nnz(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: coalesce + operator_name: coalesce + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::coalesce(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _coalesce + operator_name: _coalesce + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_coalesce(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: is_coalesced + operator_name: is_coalesced + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_coalesced(Tensor self) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _indices + operator_name: _indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_indices(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _values + operator_name: _values + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_values(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _coalesced_ + operator_name: _coalesced_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_coalesced_(Tensor(a!) self, bool coalesced) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: coalesced + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: coalesced + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: indices + operator_name: indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::indices(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: values + operator_name: values + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::values(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: crow_indices + operator_name: crow_indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::crow_indices(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: col_indices + operator_name: col_indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::col_indices(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: ccol_indices + operator_name: ccol_indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::ccol_indices(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: row_indices + operator_name: row_indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::row_indices(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: hspmm_out + operator_name: hspmm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hspmm.out(Tensor mat1, Tensor mat2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hspmm + operator_name: hspmm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hspmm(Tensor mat1, Tensor mat2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: copy_sparse_to_sparse_ + operator_name: copy_sparse_to_sparse_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::copy_sparse_to_sparse_(Tensor(a!) self, Tensor src, bool non_blocking=False) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unbind + operator_name: unbind + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::unbind.int(Tensor(a -> *) self, int dim=0) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unbind + operator_name: unbind + overload_name: Dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::unbind.Dimname(Tensor(a -> *) self, Dimname dim) -> Tensor(a)[] + arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::Dimname) + schema_order_arguments: + - annotation: a -> * + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: to_sparse + operator_name: to_sparse + overload_name: sparse_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_sparse.sparse_dim(Tensor self, int sparse_dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _to_sparse + operator_name: _to_sparse + overload_name: sparse_dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse.sparse_dim(Tensor self, int sparse_dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_sparse + operator_name: to_sparse + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_sparse(Tensor self, *, Layout? layout=None, int[2]? blocksize=None, int? dense_dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + kwarg_only: true + name: blocksize + size: 2 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, at::OptionalIntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + kwarg_only: true + name: blocksize + size: 2 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dense_dim + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _to_sparse + operator_name: _to_sparse + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse(Tensor self, *, Layout? layout=None, int[2]? blocksize=None, int? dense_dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + kwarg_only: true + name: blocksize + size: 2 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, at::OptionalIntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + kwarg_only: true + name: blocksize + size: 2 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dense_dim + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_sparse_csr + operator_name: to_sparse_csr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_sparse_csr(Tensor self, int? dense_dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _to_sparse_csr + operator_name: _to_sparse_csr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse_csr(Tensor self, int? dense_dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_sparse_csc + operator_name: to_sparse_csc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_sparse_csc(Tensor self, int? dense_dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _to_sparse_csc + operator_name: _to_sparse_csc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse_csc(Tensor self, int? dense_dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_sparse_bsr + operator_name: to_sparse_bsr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_sparse_bsr(Tensor self, int[2] blocksize, int? dense_dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _to_sparse_bsr + operator_name: _to_sparse_bsr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse_bsr(Tensor self, int[2] blocksize, int? dense_dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_sparse_bsc + operator_name: to_sparse_bsc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_sparse_bsc(Tensor self, int[2] blocksize, int? dense_dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _to_sparse_bsc + operator_name: _to_sparse_bsc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse_bsc(Tensor self, int[2] blocksize, int? dense_dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _to_sparse_semi_structured + operator_name: _to_sparse_semi_structured + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse_semi_structured(Tensor dense) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_mkldnn + operator_name: to_mkldnn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_mkldnn(Tensor self, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_reorder_conv2d_weight + operator_name: mkldnn_reorder_conv2d_weight + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_reorder_conv2d_weight(Tensor self, SymInt[2] padding=0, SymInt[2] stride=1, SymInt[2] dilation=1, SymInt groups=1, SymInt[]? input_size=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: input_size + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::OptionalIntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: input_size + type: at::OptionalIntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_reorder_conv3d_weight + operator_name: mkldnn_reorder_conv3d_weight + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_reorder_conv3d_weight(Tensor self, SymInt[3] padding=0, SymInt[3] stride=1, SymInt[3] dilation=1, SymInt groups=1, SymInt[]? input_size=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: input_size + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::OptionalIntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: input_size + type: at::OptionalIntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_mkldnn_backward + operator_name: to_mkldnn_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_mkldnn_backward(Tensor grad, Tensor input) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: quantize_per_tensor_dynamic + operator_name: quantize_per_tensor_dynamic + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantize_per_tensor_dynamic(Tensor self, ScalarType dtype, bool reduce_range) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - annotation: null + dynamic_type: bool + is_nullable: false + name: reduce_range + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::ScalarType, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - annotation: null + dynamic_type: bool + is_nullable: false + name: reduce_range + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantize_per_tensor + operator_name: quantize_per_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantize_per_tensor(Tensor self, float scale, int zero_point, ScalarType dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, int64_t, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantize_per_tensor + operator_name: quantize_per_tensor + overload_name: tensor_qparams + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantize_per_tensor.tensor_qparams(Tensor self, Tensor scale, Tensor zero_point, ScalarType dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantize_per_tensor + operator_name: quantize_per_tensor + overload_name: tensors + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantize_per_tensor.tensors(Tensor[] tensors, Tensor scales, Tensor zero_points, ScalarType dtype) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Tensor &, const at::Tensor &, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantize_per_channel + operator_name: quantize_per_channel + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantize_per_channel(Tensor self, Tensor scales, Tensor zero_points, int axis, ScalarType dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: dequantize + operator_name: dequantize + overload_name: self + manual_kernel_registration: false + category_override: '' + schema_string: aten::dequantize.self(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: dequantize + operator_name: dequantize + overload_name: tensors + manual_kernel_registration: false + category_override: '' + schema_string: aten::dequantize.tensors(Tensor[] tensors) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: q_scale + operator_name: q_scale + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::q_scale(Tensor self) -> float + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: double (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: double + name: result + type: double + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: q_zero_point + operator_name: q_zero_point + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::q_zero_point(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: q_per_channel_scales + operator_name: q_per_channel_scales + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::q_per_channel_scales(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: q_per_channel_zero_points + operator_name: q_per_channel_zero_points + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::q_per_channel_zero_points(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: q_per_channel_axis + operator_name: q_per_channel_axis + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::q_per_channel_axis(Tensor self) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: int64_t (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: int_repr + operator_name: int_repr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::int_repr(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _make_per_tensor_quantized_tensor + operator_name: _make_per_tensor_quantized_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_make_per_tensor_quantized_tensor(Tensor self, float scale, int zero_point) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _make_per_channel_quantized_tensor + operator_name: _make_per_channel_quantized_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_make_per_channel_quantized_tensor(Tensor self, Tensor scale, Tensor zero_point, int axis) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: qscheme + operator_name: qscheme + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::qscheme(Tensor self) -> QScheme + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::QScheme (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::QScheme + name: result + type: at::QScheme + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fake_quantize_per_tensor_affine + operator_name: fake_quantize_per_tensor_affine + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fake_quantize_per_tensor_affine(Tensor self, float scale, int zero_point, int quant_min, int quant_max) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fake_quantize_per_tensor_affine + operator_name: fake_quantize_per_tensor_affine + overload_name: tensor_qparams + manual_kernel_registration: false + category_override: '' + schema_string: aten::fake_quantize_per_tensor_affine.tensor_qparams(Tensor self, Tensor scale, Tensor zero_point, int quant_min, int quant_max) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fake_quantize_per_tensor_affine_cachemask + operator_name: fake_quantize_per_tensor_affine_cachemask + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fake_quantize_per_tensor_affine_cachemask(Tensor self, float scale, int zero_point, int quant_min, int quant_max) -> (Tensor output, Tensor mask) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, double, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: mask + name: mask + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fake_quantize_per_tensor_affine_cachemask_tensor_qparams + operator_name: _fake_quantize_per_tensor_affine_cachemask_tensor_qparams + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fake_quantize_per_tensor_affine_cachemask_tensor_qparams(Tensor self, Tensor scale, Tensor zero_point, Tensor fake_quant_enabled, int quant_min, int quant_max) -> (Tensor output, Tensor mask) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_enabled + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_enabled + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: mask + name: mask + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fake_quantize_per_tensor_affine_cachemask_backward + operator_name: fake_quantize_per_tensor_affine_cachemask_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fake_quantize_per_tensor_affine_cachemask_backward(Tensor grad, Tensor mask) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _fake_quantize_learnable_per_tensor_affine + operator_name: _fake_quantize_learnable_per_tensor_affine + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fake_quantize_learnable_per_tensor_affine(Tensor self, Tensor scale, Tensor zero_point, int quant_min, int quant_max, float grad_factor=1.0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fake_quantize_learnable_per_tensor_affine_backward + operator_name: _fake_quantize_learnable_per_tensor_affine_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fake_quantize_learnable_per_tensor_affine_backward(Tensor grad, Tensor self, Tensor scale, Tensor zero_point, int quant_min, int quant_max, float grad_factor=1.0) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fake_quantize_per_channel_affine + operator_name: fake_quantize_per_channel_affine + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fake_quantize_per_channel_affine(Tensor self, Tensor scale, Tensor zero_point, int axis, int quant_min, int quant_max) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fake_quantize_per_channel_affine_cachemask + operator_name: fake_quantize_per_channel_affine_cachemask + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fake_quantize_per_channel_affine_cachemask(Tensor self, Tensor scale, Tensor zero_point, int axis, int quant_min, int quant_max) -> (Tensor output, Tensor mask) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: mask + name: mask + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fake_quantize_per_channel_affine_cachemask_backward + operator_name: fake_quantize_per_channel_affine_cachemask_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fake_quantize_per_channel_affine_cachemask_backward(Tensor grad, Tensor mask) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _fake_quantize_learnable_per_channel_affine + operator_name: _fake_quantize_learnable_per_channel_affine + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fake_quantize_learnable_per_channel_affine(Tensor self, Tensor scale, Tensor zero_point, int axis, int quant_min, int quant_max, float grad_factor=1.0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fake_quantize_learnable_per_channel_affine_backward + operator_name: _fake_quantize_learnable_per_channel_affine_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fake_quantize_learnable_per_channel_affine_backward(Tensor grad, Tensor self, Tensor scale, Tensor zero_point, int axis, int quant_min, int quant_max, float grad_factor=1.0) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fused_moving_avg_obs_fake_quant + operator_name: fused_moving_avg_obs_fake_quant + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fused_moving_avg_obs_fake_quant(Tensor self, Tensor observer_on, Tensor fake_quant_on, Tensor(a!) running_min, Tensor(b!) running_max, Tensor(c!) scale, Tensor(d!) zero_point, float averaging_const, int quant_min, int quant_max, int ch_axis, bool per_row_fake_quant=False, bool symmetric_quant=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: observer_on + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_on + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_min + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_max + type: at::Tensor & + - annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: at::Tensor & + - annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: averaging_const + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ch_axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: per_row_fake_quant + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: symmetric_quant + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, double, int64_t, int64_t, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: observer_on + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_on + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_min + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_max + type: at::Tensor & + - annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: at::Tensor & + - annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: averaging_const + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ch_axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: per_row_fake_quant + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: symmetric_quant + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _fused_moving_avg_obs_fq_helper + operator_name: _fused_moving_avg_obs_fq_helper + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_moving_avg_obs_fq_helper(Tensor self, Tensor observer_on, Tensor fake_quant_on, Tensor(a!) running_min, Tensor(b!) running_max, Tensor(c!) scale, Tensor(d!) zero_point, float averaging_const, int quant_min, int quant_max, int ch_axis, bool per_row_fake_quant=False, bool symmetric_quant=False) -> (Tensor output, Tensor mask) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: observer_on + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_on + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_min + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_max + type: at::Tensor & + - annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: at::Tensor & + - annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: averaging_const + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ch_axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: per_row_fake_quant + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: symmetric_quant + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, double, int64_t, int64_t, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: observer_on + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_on + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_min + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_max + type: at::Tensor & + - annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: at::Tensor & + - annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: averaging_const + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ch_axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: per_row_fake_quant + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: symmetric_quant + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: mask + name: mask + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _choose_qparams_per_tensor + operator_name: _choose_qparams_per_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_choose_qparams_per_tensor(Tensor self, bool reduce_range=False) -> (float, int) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: reduce_range + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: reduce_range + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: double + name: result0 + type: double + - dynamic_type: int64_t + name: result1 + type: int64_t + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _saturate_weight_to_fp16 + operator_name: _saturate_weight_to_fp16 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_saturate_weight_to_fp16(Tensor weight) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: choose_qparams_optimized + operator_name: choose_qparams_optimized + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::choose_qparams_optimized(Tensor input, int numel, int n_bins, float ratio, int bit_width) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: numel + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n_bins + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: ratio + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: bit_width + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, int64_t, double, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: numel + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n_bins + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: ratio + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: bit_width + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _autocast_to_reduced_precision + operator_name: _autocast_to_reduced_precision + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_autocast_to_reduced_precision(Tensor(a) self, bool cuda_enabled, bool cpu_enabled, ScalarType cuda_dtype, ScalarType cpu_dtype) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: cuda_enabled + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: cpu_enabled + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: cuda_dtype + type: at::ScalarType + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: cpu_dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool, bool, at::ScalarType, at::ScalarType) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: cuda_enabled + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: cpu_enabled + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: cuda_dtype + type: at::ScalarType + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: cpu_dtype + type: at::ScalarType + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: _autocast_to_full_precision + operator_name: _autocast_to_full_precision + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_autocast_to_full_precision(Tensor(a) self, bool cuda_enabled, bool cpu_enabled) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: cuda_enabled + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: cpu_enabled + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: cuda_enabled + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: cpu_enabled + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: _to_copy + operator_name: _to_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_copy(Tensor self, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, bool non_blocking=False, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: non_blocking + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: non_blocking + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: to + operator_name: to + overload_name: dtype_layout + manual_kernel_registration: false + category_override: '' + schema_string: aten::to.dtype_layout(Tensor(a) self, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, bool non_blocking=False, bool copy=False, MemoryFormat? memory_format=None) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::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 + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, ::std::optional, ::std::optional, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + - 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 + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: to + operator_name: to + overload_name: device + manual_kernel_registration: false + category_override: '' + schema_string: aten::to.device(Tensor(a) self, Device device, ScalarType dtype, bool non_blocking=False, bool copy=False, MemoryFormat? memory_format=None) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Device + is_nullable: false + name: device + type: at::Device + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: copy + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Device, at::ScalarType, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Device + is_nullable: false + name: device + type: at::Device + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: copy + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + name: memory_format + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: to + operator_name: to + overload_name: dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::to.dtype(Tensor(a) self, ScalarType dtype, bool non_blocking=False, bool copy=False, MemoryFormat? memory_format=None) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: copy + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::ScalarType, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: copy + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + name: memory_format + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: to + operator_name: to + overload_name: other + manual_kernel_registration: false + category_override: '' + schema_string: aten::to.other(Tensor(a) self, Tensor other, bool non_blocking=False, bool copy=False, MemoryFormat? memory_format=None) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: copy + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: copy + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + name: memory_format + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: meshgrid + operator_name: meshgrid + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::meshgrid(Tensor[] tensors) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: meshgrid + operator_name: meshgrid + overload_name: indexing + manual_kernel_registration: false + category_override: '' + schema_string: aten::meshgrid.indexing(Tensor[] tensors, *, str indexing) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: indexing + type: c10::string_view + schema_order_cpp_signature: ::std::vector (at::TensorList, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: indexing + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cartesian_prod + operator_name: cartesian_prod + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cartesian_prod(Tensor[] tensors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: combinations + operator_name: combinations + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::combinations(Tensor self, int r=2, bool with_replacement=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: int64_t + is_nullable: false + name: r + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: with_replacement + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: int64_t + is_nullable: false + name: r + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: with_replacement + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: item + operator_name: item + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::item(Tensor self) -> Scalar + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Scalar (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: const at::Scalar & + name: result + type: at::Scalar + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: result_type + operator_name: result_type + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::result_type.Tensor(Tensor tensor, Tensor other) -> ScalarType + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::ScalarType (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::ScalarType + name: result + type: at::ScalarType + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: result_type + operator_name: result_type + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::result_type.Scalar(Tensor tensor, Scalar other) -> ScalarType + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::ScalarType (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::ScalarType + name: result + type: at::ScalarType + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: result_type + operator_name: result_type + overload_name: Scalar_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::result_type.Scalar_Tensor(Scalar scalar, Tensor tensor) -> ScalarType + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + schema_order_cpp_signature: at::ScalarType (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::ScalarType + name: result + type: at::ScalarType + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: result_type + operator_name: result_type + overload_name: Scalar_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::result_type.Scalar_Scalar(Scalar scalar1, Scalar scalar2) -> ScalarType + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar1 + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar2 + type: const at::Scalar & + schema_order_cpp_signature: at::ScalarType (const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar1 + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar2 + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::ScalarType + name: result + type: at::ScalarType + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: can_cast + operator_name: can_cast + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::can_cast(ScalarType from_, ScalarType to) -> bool + arguments: + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: from_ + type: at::ScalarType + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: to + type: at::ScalarType + schema_order_cpp_signature: bool (at::ScalarType, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: from_ + type: at::ScalarType + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: to + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: promote_types + operator_name: promote_types + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::promote_types(ScalarType type1, ScalarType type2) -> ScalarType + arguments: + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: type1 + type: at::ScalarType + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: type2 + type: at::ScalarType + schema_order_cpp_signature: at::ScalarType (at::ScalarType, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: type1 + type: at::ScalarType + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: type2 + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::ScalarType + name: result + type: at::ScalarType + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _local_scalar_dense + operator_name: _local_scalar_dense + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_local_scalar_dense(Tensor self) -> Scalar + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Scalar (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: const at::Scalar & + name: result + type: at::Scalar + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _lstm_mps + operator_name: _lstm_mps + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_lstm_mps(Tensor input, Tensor[] hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional, bool batch_first) -> (Tensor, Tensor, Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::TensorList, at::TensorList, bool, int64_t, double, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: at::Tensor + name: result4 + type: at::Tensor + - dynamic_type: at::Tensor + name: result5 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lstm_mps_backward + operator_name: lstm_mps_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lstm_mps_backward(Tensor? grad_y, Tensor? grad_hy, Tensor? grad_cy, Tensor z_state, Tensor cell_state_fwd, Tensor input, Tensor layersOutputs, Tensor[] hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional, bool batch_first) -> (Tensor, Tensor[], Tensor[]) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_y + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: z_state + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cell_state_fwd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: layersOutputs + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: ::std::tuple,::std::vector> (const ::std::optional &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, at::TensorList, at::TensorList, bool, int64_t, double, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_y + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: z_state + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cell_state_fwd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: layersOutputs + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::TensorList + name: result1 + type: ::std::vector + - dynamic_type: at::TensorList + name: result2 + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _thnn_fused_lstm_cell + operator_name: _thnn_fused_lstm_cell + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_fused_lstm_cell(Tensor input_gates, Tensor hidden_gates, Tensor cx, Tensor? input_bias=None, Tensor? hidden_bias=None) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _thnn_fused_lstm_cell_backward_impl + operator_name: _thnn_fused_lstm_cell_backward_impl + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_fused_lstm_cell_backward_impl(Tensor? grad_hy, Tensor? grad_cy, Tensor cx, Tensor cy, Tensor workspace, bool has_bias) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_bias + type: bool + schema_order_cpp_signature: ::std::tuple (const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_bias + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _thnn_fused_lstm_cell_backward + operator_name: _thnn_fused_lstm_cell_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_fused_lstm_cell_backward(Tensor? grad_hy, Tensor? grad_cy, Tensor cx, Tensor cy, Tensor workspace, bool has_bias) -> (Tensor, Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_bias + type: bool + schema_order_cpp_signature: ::std::tuple (const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_bias + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: at::Tensor + name: result4 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _thnn_differentiable_lstm_cell_backward + operator_name: _thnn_differentiable_lstm_cell_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_differentiable_lstm_cell_backward(Tensor? grad_hy, Tensor? grad_cy, Tensor input_gates, Tensor hidden_gates, Tensor? input_bias, Tensor? hidden_bias, Tensor cx, Tensor cy) -> (Tensor, Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: at::Tensor + name: result4 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _thnn_fused_gru_cell + operator_name: _thnn_fused_gru_cell + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_fused_gru_cell(Tensor input_gates, Tensor hidden_gates, Tensor hx, Tensor? input_bias=None, Tensor? hidden_bias=None) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _thnn_fused_gru_cell_backward + operator_name: _thnn_fused_gru_cell_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_fused_gru_cell_backward(Tensor grad_hy, Tensor workspace, bool has_bias) -> (Tensor, Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_hy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_bias + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_hy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_bias + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: at::Tensor + name: result4 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _thnn_differentiable_gru_cell_backward + operator_name: _thnn_differentiable_gru_cell_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_differentiable_gru_cell_backward(Tensor grad_hy, Tensor input_gates, Tensor hidden_gates, Tensor hx, Tensor? input_bias, Tensor? hidden_bias) -> (Tensor, Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_hy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_hy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: at::Tensor + name: result4 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: lstm + operator_name: lstm + overload_name: input + manual_kernel_registration: false + category_override: '' + schema_string: aten::lstm.input(Tensor input, Tensor[] hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional, bool batch_first) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::TensorList, at::TensorList, bool, int64_t, double, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: lstm + operator_name: lstm + overload_name: data + manual_kernel_registration: false + category_override: '' + schema_string: aten::lstm.data(Tensor data, Tensor batch_sizes, Tensor[] hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::TensorList, at::TensorList, bool, int64_t, double, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gru + operator_name: gru + overload_name: input + manual_kernel_registration: false + category_override: '' + schema_string: aten::gru.input(Tensor input, Tensor hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional, bool batch_first) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::TensorList, bool, int64_t, double, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gru + operator_name: gru + overload_name: data + manual_kernel_registration: false + category_override: '' + schema_string: aten::gru.data(Tensor data, Tensor batch_sizes, Tensor hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::TensorList, bool, int64_t, double, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rnn_tanh + operator_name: rnn_tanh + overload_name: input + manual_kernel_registration: false + category_override: '' + schema_string: aten::rnn_tanh.input(Tensor input, Tensor hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional, bool batch_first) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::TensorList, bool, int64_t, double, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rnn_tanh + operator_name: rnn_tanh + overload_name: data + manual_kernel_registration: false + category_override: '' + schema_string: aten::rnn_tanh.data(Tensor data, Tensor batch_sizes, Tensor hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::TensorList, bool, int64_t, double, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rnn_relu + operator_name: rnn_relu + overload_name: input + manual_kernel_registration: false + category_override: '' + schema_string: aten::rnn_relu.input(Tensor input, Tensor hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional, bool batch_first) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::TensorList, bool, int64_t, double, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rnn_relu + operator_name: rnn_relu + overload_name: data + manual_kernel_registration: false + category_override: '' + schema_string: aten::rnn_relu.data(Tensor data, Tensor batch_sizes, Tensor hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::TensorList, bool, int64_t, double, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: lstm_cell + operator_name: lstm_cell + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lstm_cell(Tensor input, Tensor[] hx, Tensor w_ih, Tensor w_hh, Tensor? b_ih=None, Tensor? b_hh=None) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_ih + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_hh + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::TensorList, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_ih + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_hh + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gru_cell + operator_name: gru_cell + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::gru_cell(Tensor input, Tensor hx, Tensor w_ih, Tensor w_hh, Tensor? b_ih=None, Tensor? b_hh=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_ih + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_hh + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_ih + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_hh + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rnn_tanh_cell + operator_name: rnn_tanh_cell + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rnn_tanh_cell(Tensor input, Tensor hx, Tensor w_ih, Tensor w_hh, Tensor? b_ih=None, Tensor? b_hh=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_ih + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_hh + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_ih + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_hh + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: rnn_relu_cell + operator_name: rnn_relu_cell + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rnn_relu_cell(Tensor input, Tensor hx, Tensor w_ih, Tensor w_hh, Tensor? b_ih=None, Tensor? b_hh=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_ih + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_hh + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_ih + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: b_hh + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: quantized_lstm_cell + operator_name: quantized_lstm_cell + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_lstm_cell(Tensor input, Tensor[] hx, Tensor w_ih, Tensor w_hh, Tensor b_ih, Tensor b_hh, Tensor packed_ih, Tensor packed_hh, Tensor col_offsets_ih, Tensor col_offsets_hh, Scalar scale_ih, Scalar scale_hh, Scalar zero_point_ih, Scalar zero_point_hh) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_hh + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_hh + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_hh + type: const at::Scalar & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::TensorList, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_hh + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_hh + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_hh + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: quantized_gru_cell + operator_name: quantized_gru_cell + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_gru_cell(Tensor input, Tensor hx, Tensor w_ih, Tensor w_hh, Tensor b_ih, Tensor b_hh, Tensor packed_ih, Tensor packed_hh, Tensor col_offsets_ih, Tensor col_offsets_hh, Scalar scale_ih, Scalar scale_hh, Scalar zero_point_ih, Scalar zero_point_hh) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_hh + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_hh + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_hh + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_hh + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_hh + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_hh + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: quantized_rnn_relu_cell + operator_name: quantized_rnn_relu_cell + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_rnn_relu_cell(Tensor input, Tensor hx, Tensor w_ih, Tensor w_hh, Tensor b_ih, Tensor b_hh, Tensor packed_ih, Tensor packed_hh, Tensor col_offsets_ih, Tensor col_offsets_hh, Scalar scale_ih, Scalar scale_hh, Scalar zero_point_ih, Scalar zero_point_hh) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_hh + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_hh + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_hh + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_hh + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_hh + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_hh + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: quantized_rnn_tanh_cell + operator_name: quantized_rnn_tanh_cell + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_rnn_tanh_cell(Tensor input, Tensor hx, Tensor w_ih, Tensor w_hh, Tensor b_ih, Tensor b_hh, Tensor packed_ih, Tensor packed_hh, Tensor col_offsets_ih, Tensor col_offsets_hh, Scalar scale_ih, Scalar scale_hh, Scalar zero_point_ih, Scalar zero_point_hh) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_hh + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_hh + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_hh + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: w_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: packed_hh + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_ih + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_offsets_hh + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale_hh + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_ih + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: zero_point_hh + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _pack_padded_sequence + operator_name: _pack_padded_sequence + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pack_padded_sequence(Tensor input, Tensor lengths, bool batch_first) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: lengths + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: lengths + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _pack_padded_sequence_backward + operator_name: _pack_padded_sequence_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pack_padded_sequence_backward(Tensor grad, SymInt[] input_size, Tensor batch_sizes, bool batch_first) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _pad_packed_sequence + operator_name: _pad_packed_sequence + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pad_packed_sequence(Tensor data, Tensor batch_sizes, bool batch_first, Scalar padding_value, int total_length) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: padding_value + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: total_length + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, const at::Scalar &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch_sizes + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: padding_value + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: total_length + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: set_ + operator_name: set_ + overload_name: source_Storage + manual_kernel_registration: false + category_override: '' + schema_string: aten::set_.source_Storage(Tensor(a!) self, Storage source) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::Storage) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set_ + operator_name: set_ + overload_name: source_Storage_storage_offset + manual_kernel_registration: false + category_override: '' + schema_string: aten::set_.source_Storage_storage_offset(Tensor(a!) self, Storage source, SymInt storage_offset, SymInt[] size, SymInt[] stride=[]) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: storage_offset + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::Storage, int64_t, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: storage_offset + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set_ + operator_name: set_ + overload_name: source_Tensor_storage_offset + manual_kernel_registration: false + category_override: '' + schema_string: aten::set_.source_Tensor_storage_offset(Tensor(a!) self, Tensor source, SymInt storage_offset, SymInt[] size, SymInt[] stride=[]) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: storage_offset + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, int64_t, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: storage_offset + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: set_ + operator_name: set_ + overload_name: source_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::set_.source_Tensor(Tensor(a!) self, Tensor source) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set_ + operator_name: set_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::set_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lift + operator_name: lift + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lift(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lift_fresh + operator_name: lift_fresh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lift_fresh(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lift_fresh_copy + operator_name: lift_fresh_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lift_fresh_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: is_set_to + operator_name: is_set_to + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::is_set_to(Tensor self, Tensor tensor) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_fill_ + operator_name: masked_fill_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_fill_.Scalar(Tensor(a!) self, Tensor mask, Scalar value) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_fill + operator_name: masked_fill + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_fill.Scalar(Tensor self, Tensor mask, Scalar value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_fill_ + operator_name: masked_fill_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_fill_.Tensor(Tensor(a!) self, Tensor mask, Tensor value) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_fill + operator_name: masked_fill + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_fill.Tensor(Tensor self, Tensor mask, Tensor value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_scatter_ + operator_name: masked_scatter_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_scatter_(Tensor(a!) self, Tensor mask, Tensor source) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_scatter + operator_name: masked_scatter + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_scatter(Tensor self, Tensor mask, Tensor source) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_scatter_backward + operator_name: masked_scatter_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_scatter_backward(Tensor grad_output, Tensor mask, SymInt[] sizes) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sizes + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sizes + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _masked_softmax + operator_name: _masked_softmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_masked_softmax(Tensor self, Tensor mask, int? dim=None, int? mask_type=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _masked_softmax_backward + operator_name: _masked_softmax_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_masked_softmax_backward(Tensor grad_output, Tensor output, Tensor mask, int? dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: view + operator_name: view + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::view(Tensor(a) self, SymInt[] size) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: view + operator_name: view + overload_name: dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::view.dtype(Tensor(a) self, ScalarType dtype) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::ScalarType) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: put_ + operator_name: put_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::put_(Tensor(a!) self, Tensor index, Tensor source, bool accumulate=False) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: put + operator_name: put + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::put(Tensor self, Tensor index, Tensor source, bool accumulate=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_add_out + operator_name: index_add + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_add.out(Tensor self, int dim, Tensor index, Tensor source, *, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_add_ + operator_name: index_add_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_add_(Tensor(a!) self, int dim, Tensor index, Tensor source, *, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_add + operator_name: index_add + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_add(Tensor self, int dim, Tensor index, Tensor source, *, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_add + operator_name: index_add + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_add.dimname(Tensor self, Dimname dim, Tensor index, Tensor source, *, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: index_reduce_out + operator_name: index_reduce + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_reduce.out(Tensor self, int dim, Tensor index, Tensor source, str reduce, *, bool include_self=True, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, c10::string_view, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_reduce_ + operator_name: index_reduce_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_reduce_(Tensor(a!) self, int dim, Tensor index, Tensor source, str reduce, *, bool include_self=True) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, c10::string_view, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_reduce + operator_name: index_reduce + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_reduce(Tensor self, int dim, Tensor index, Tensor source, str reduce, *, bool include_self=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, c10::string_view, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_fill_ + operator_name: index_fill_ + overload_name: int_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_fill_.int_Scalar(Tensor(a!) self, int dim, Tensor index, Scalar value) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_fill + operator_name: index_fill + overload_name: int_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_fill.int_Scalar(Tensor self, int dim, Tensor index, Scalar value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_fill_ + operator_name: index_fill_ + overload_name: int_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_fill_.int_Tensor(Tensor(a!) self, int dim, Tensor index, Tensor value) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_fill + operator_name: index_fill + overload_name: int_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_fill.int_Tensor(Tensor self, int dim, Tensor index, Tensor value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_fill_ + operator_name: index_fill_ + overload_name: Dimname_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_fill_.Dimname_Scalar(Tensor(a!) self, Dimname dim, Tensor index, Scalar value) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::Dimname, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: index_fill_ + operator_name: index_fill_ + overload_name: Dimname_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_fill_.Dimname_Tensor(Tensor(a!) self, Dimname dim, Tensor index, Tensor value) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::Dimname, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: index_fill + operator_name: index_fill + overload_name: Dimname_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_fill.Dimname_Scalar(Tensor self, Dimname dim, Tensor index, Scalar value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: index_fill + operator_name: index_fill + overload_name: Dimname_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_fill.Dimname_Tensor(Tensor self, Dimname dim, Tensor index, Tensor value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: scatter + operator_name: scatter + overload_name: src + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter.src(Tensor self, int dim, Tensor index, Tensor src) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_ + operator_name: scatter_ + overload_name: src + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_.src(Tensor(a!) self, int dim, Tensor index, Tensor src) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_out + operator_name: scatter + overload_name: src_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter.src_out(Tensor self, int dim, Tensor index, Tensor src, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter + operator_name: scatter + overload_name: value + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter.value(Tensor self, int dim, Tensor index, Scalar value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_ + operator_name: scatter_ + overload_name: value + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_.value(Tensor(a!) self, int dim, Tensor index, Scalar value) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_out + operator_name: scatter + overload_name: value_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter.value_out(Tensor self, int dim, Tensor index, Scalar value, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter + operator_name: scatter + overload_name: reduce + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter.reduce(Tensor self, int dim, Tensor index, Tensor src, *, str reduce) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_ + operator_name: scatter_ + overload_name: reduce + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_.reduce(Tensor(a!) self, int dim, Tensor index, Tensor src, *, str reduce) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_out + operator_name: scatter + overload_name: reduce_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter.reduce_out(Tensor self, int dim, Tensor index, Tensor src, *, str reduce, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, c10::string_view, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter + operator_name: scatter + overload_name: value_reduce + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter.value_reduce(Tensor self, int dim, Tensor index, Scalar value, *, str reduce) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Scalar &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_ + operator_name: scatter_ + overload_name: value_reduce + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_.value_reduce(Tensor(a!) self, int dim, Tensor index, Scalar value, *, str reduce) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Scalar &, c10::string_view) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_out + operator_name: scatter + overload_name: value_reduce_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter.value_reduce_out(Tensor self, int dim, Tensor index, Scalar value, *, str reduce, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Scalar &, c10::string_view, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: reduce + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter + operator_name: scatter + overload_name: dimname_src + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter.dimname_src(Tensor self, Dimname dim, Tensor index, Tensor src) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: scatter + operator_name: scatter + overload_name: dimname_value + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter.dimname_value(Tensor self, Dimname dim, Tensor index, Scalar value) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: scatter_add + operator_name: scatter_add + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_add(Tensor self, int dim, Tensor index, Tensor src) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_add_ + operator_name: scatter_add_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_add_(Tensor(a!) self, int dim, Tensor index, Tensor src) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_add_out + operator_name: scatter_add + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_add.out(Tensor self, int dim, Tensor index, Tensor src, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_add + operator_name: scatter_add + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_add.dimname(Tensor self, Dimname dim, Tensor index, Tensor src) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: scatter_reduce + operator_name: scatter_reduce + overload_name: two + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_reduce.two(Tensor self, int dim, Tensor index, Tensor src, str reduce, *, bool include_self=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, c10::string_view, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_reduce_ + operator_name: scatter_reduce_ + overload_name: two + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_reduce_.two(Tensor(a!) self, int dim, Tensor index, Tensor src, str reduce, *, bool include_self=True) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, c10::string_view, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scatter_reduce_out + operator_name: scatter_reduce + overload_name: two_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::scatter_reduce.two_out(Tensor self, int dim, Tensor index, Tensor src, str reduce, *, bool include_self=True, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, c10::string_view, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: include_self + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: eq_ + operator_name: eq_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::eq_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: eq_ + operator_name: eq_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::eq_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_and_out + operator_name: bitwise_and + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_and.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_and_out + operator_name: bitwise_and + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_and.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_and + operator_name: bitwise_and + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_and.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_and + operator_name: bitwise_and + overload_name: Scalar_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_and.Scalar_Tensor(Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_and + operator_name: bitwise_and + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_and.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_and_ + operator_name: bitwise_and_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_and_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_and_ + operator_name: bitwise_and_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_and_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: __and__ + operator_name: __and__ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::__and__.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: __and__ + operator_name: __and__ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::__and__.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: __iand__ + operator_name: __iand__ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::__iand__.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: __iand__ + operator_name: __iand__ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::__iand__.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: bitwise_or_out + operator_name: bitwise_or + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_or.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_or_out + operator_name: bitwise_or + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_or.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_or + operator_name: bitwise_or + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_or.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_or + operator_name: bitwise_or + overload_name: Scalar_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_or.Scalar_Tensor(Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_or + operator_name: bitwise_or + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_or.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_or_ + operator_name: bitwise_or_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_or_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_or_ + operator_name: bitwise_or_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_or_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: __or__ + operator_name: __or__ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::__or__.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: __or__ + operator_name: __or__ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::__or__.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: __ior__ + operator_name: __ior__ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::__ior__.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: __ior__ + operator_name: __ior__ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::__ior__.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: bitwise_xor_out + operator_name: bitwise_xor + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_xor.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_xor_out + operator_name: bitwise_xor + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_xor.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_xor + operator_name: bitwise_xor + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_xor.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_xor + operator_name: bitwise_xor + overload_name: Scalar_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_xor.Scalar_Tensor(Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_xor + operator_name: bitwise_xor + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_xor.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_xor_ + operator_name: bitwise_xor_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_xor_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_xor_ + operator_name: bitwise_xor_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_xor_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: __xor__ + operator_name: __xor__ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::__xor__.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: __xor__ + operator_name: __xor__ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::__xor__.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: __ixor__ + operator_name: __ixor__ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::__ixor__.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: __ixor__ + operator_name: __ixor__ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::__ixor__.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: __lshift__ + operator_name: __lshift__ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::__lshift__.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: __lshift__ + operator_name: __lshift__ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::__lshift__.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: __ilshift__ + operator_name: __ilshift__ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::__ilshift__.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: __ilshift__ + operator_name: __ilshift__ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::__ilshift__.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_left_shift + operator_name: bitwise_left_shift + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_left_shift.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_left_shift_ + operator_name: bitwise_left_shift_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_left_shift_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_left_shift_out + operator_name: bitwise_left_shift + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_left_shift.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_left_shift + operator_name: bitwise_left_shift + overload_name: Tensor_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_left_shift.Tensor_Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_left_shift_ + operator_name: bitwise_left_shift_ + overload_name: Tensor_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_left_shift_.Tensor_Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_left_shift_out + operator_name: bitwise_left_shift + overload_name: Tensor_Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_left_shift.Tensor_Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_left_shift + operator_name: bitwise_left_shift + overload_name: Scalar_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_left_shift.Scalar_Tensor(Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: __rshift__ + operator_name: __rshift__ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::__rshift__.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: __rshift__ + operator_name: __rshift__ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::__rshift__.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: __irshift__ + operator_name: __irshift__ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::__irshift__.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: __irshift__ + operator_name: __irshift__ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::__irshift__.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_right_shift + operator_name: bitwise_right_shift + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_right_shift.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_right_shift_ + operator_name: bitwise_right_shift_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_right_shift_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_right_shift_out + operator_name: bitwise_right_shift + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_right_shift.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_right_shift + operator_name: bitwise_right_shift + overload_name: Tensor_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_right_shift.Tensor_Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_right_shift_ + operator_name: bitwise_right_shift_ + overload_name: Tensor_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_right_shift_.Tensor_Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_right_shift_out + operator_name: bitwise_right_shift + overload_name: Tensor_Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_right_shift.Tensor_Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_right_shift + operator_name: bitwise_right_shift + overload_name: Scalar_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_right_shift.Scalar_Tensor(Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tril_ + operator_name: tril_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::tril_(Tensor(a!) self, SymInt diagonal=0) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: triu_ + operator_name: triu_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::triu_(Tensor(a!) self, SymInt diagonal=0) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: digamma_ + operator_name: digamma_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::digamma_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lerp_ + operator_name: lerp_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::lerp_.Scalar(Tensor(a!) self, Tensor end, Scalar weight) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lerp_ + operator_name: lerp_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::lerp_.Tensor(Tensor(a!) self, Tensor end, Tensor weight) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addbmm_ + operator_name: addbmm_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addbmm_(Tensor(a!) self, Tensor batch1, Tensor batch2, *, Scalar beta=1, Scalar alpha=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addbmm_out + operator_name: addbmm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::addbmm.out(Tensor self, Tensor batch1, Tensor batch2, *, Scalar beta=1, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addbmm + operator_name: addbmm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addbmm(Tensor self, Tensor batch1, Tensor batch2, *, Scalar beta=1, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: batch2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: random_ + operator_name: random_ + overload_name: from + manual_kernel_registration: false + category_override: '' + schema_string: aten::random_.from(Tensor(a!) self, int from, int? to, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: from + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: to + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: from + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: to + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: random_ + operator_name: random_ + overload_name: to + manual_kernel_registration: false + category_override: '' + schema_string: aten::random_.to(Tensor(a!) self, int to, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: to + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: to + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: random_ + operator_name: random_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::random_(Tensor(a!) self, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: uniform_ + operator_name: uniform_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::uniform_(Tensor(a!) self, float from=0, float to=1, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: from + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: to + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, double, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: from + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: to + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cauchy_ + operator_name: cauchy_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cauchy_(Tensor(a!) self, float median=0, float sigma=1, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: median + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: sigma + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, double, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: median + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: sigma + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_normal_ + operator_name: log_normal_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_normal_(Tensor(a!) self, float mean=1, float std=2, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, double, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: exponential_ + operator_name: exponential_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::exponential_(Tensor(a!) self, float lambd=1, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: lambd + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: lambd + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: geometric_ + operator_name: geometric_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::geometric_(Tensor(a!) self, float p, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: diag_out + operator_name: diag + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::diag.out(Tensor self, int diagonal=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: diag + operator_name: diag + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::diag(Tensor self, int diagonal=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cross_out + operator_name: cross + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cross.out(Tensor self, Tensor other, int? dim=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: cross + operator_name: cross + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cross(Tensor self, Tensor other, int? dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: triu_out + operator_name: triu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::triu.out(Tensor self, SymInt diagonal=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: triu + operator_name: triu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::triu(Tensor self, SymInt diagonal=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tril_out + operator_name: tril + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::tril.out(Tensor self, SymInt diagonal=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tril + operator_name: tril + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::tril(Tensor self, SymInt diagonal=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: diagonal + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tril_indices + operator_name: tril_indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::tril_indices(int row, int col, int offset=0, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: row + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: col + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: at::kLong + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, int64_t, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: row + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: col + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: at::kLong + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: triu_indices + operator_name: triu_indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::triu_indices(int row, int col, int offset=0, *, ScalarType? dtype=long, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: row + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: col + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: at::kLong + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, int64_t, int64_t, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: row + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: col + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: at::kLong + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: trace + operator_name: trace + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::trace(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: trace_backward + operator_name: trace_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::trace_backward(Tensor grad, SymInt[] sizes) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sizes + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sizes + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: ne_out + operator_name: ne + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ne.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ne + operator_name: ne + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::ne.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ne_out + operator_name: ne + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ne.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ne + operator_name: ne + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::ne.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ne_ + operator_name: ne_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::ne_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ne_ + operator_name: ne_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::ne_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: not_equal_out + operator_name: not_equal + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::not_equal.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: not_equal + operator_name: not_equal + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::not_equal.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: not_equal_out + operator_name: not_equal + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::not_equal.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: not_equal + operator_name: not_equal + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::not_equal.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: not_equal_ + operator_name: not_equal_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::not_equal_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: not_equal_ + operator_name: not_equal_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::not_equal_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: eq_out + operator_name: eq + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::eq.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: eq + operator_name: eq + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::eq.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: eq_out + operator_name: eq + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::eq.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: eq + operator_name: eq + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::eq.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ge_out + operator_name: ge + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ge.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ge + operator_name: ge + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::ge.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ge_out + operator_name: ge + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ge.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ge + operator_name: ge + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::ge.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ge_ + operator_name: ge_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::ge_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ge_ + operator_name: ge_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::ge_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: greater_equal_out + operator_name: greater_equal + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater_equal.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: greater_equal + operator_name: greater_equal + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater_equal.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: greater_equal_out + operator_name: greater_equal + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater_equal.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: greater_equal + operator_name: greater_equal + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater_equal.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: greater_equal_ + operator_name: greater_equal_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater_equal_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: greater_equal_ + operator_name: greater_equal_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater_equal_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: le_out + operator_name: le + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::le.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: le + operator_name: le + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::le.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: le_out + operator_name: le + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::le.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: le + operator_name: le + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::le.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: le_ + operator_name: le_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::le_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: le_ + operator_name: le_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::le_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: less_equal_out + operator_name: less_equal + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::less_equal.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: less_equal + operator_name: less_equal + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::less_equal.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: less_equal_out + operator_name: less_equal + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::less_equal.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: less_equal + operator_name: less_equal + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::less_equal.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: less_equal_ + operator_name: less_equal_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::less_equal_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: less_equal_ + operator_name: less_equal_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::less_equal_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gt_out + operator_name: gt + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::gt.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gt + operator_name: gt + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::gt.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gt_out + operator_name: gt + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::gt.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gt + operator_name: gt + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::gt.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gt_ + operator_name: gt_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::gt_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gt_ + operator_name: gt_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::gt_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: greater_out + operator_name: greater + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: greater + operator_name: greater + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: greater_out + operator_name: greater + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: greater + operator_name: greater + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: greater_ + operator_name: greater_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: greater_ + operator_name: greater_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::greater_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: lt_out + operator_name: lt + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lt.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lt + operator_name: lt + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::lt.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lt_out + operator_name: lt + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lt.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lt + operator_name: lt + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::lt.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lt_ + operator_name: lt_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::lt_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lt_ + operator_name: lt_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::lt_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: less_out + operator_name: less + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::less.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: less + operator_name: less + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::less.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: less_out + operator_name: less + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::less.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: less + operator_name: less + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::less.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: less_ + operator_name: less_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::less_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: less_ + operator_name: less_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::less_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: take_out + operator_name: take + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::take.out(Tensor self, Tensor index, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: take + operator_name: take + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::take(Tensor self, Tensor index) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: take_along_dim_out + operator_name: take_along_dim + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::take_along_dim.out(Tensor self, Tensor indices, int? dim=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: take_along_dim + operator_name: take_along_dim + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::take_along_dim(Tensor self, Tensor indices, int? dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: index_select_out + operator_name: index_select + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_select.out(Tensor self, int dim, Tensor index, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_select + operator_name: index_select + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_select(Tensor self, int dim, Tensor index) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_select_out + operator_name: index_select + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_select.dimname_out(Tensor self, Dimname dim, Tensor index, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Dimname, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: index_select + operator_name: index_select + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_select.dimname(Tensor self, Dimname dim, Tensor index) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: index_select_backward + operator_name: index_select_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_select_backward(Tensor grad, SymInt[] self_sizes, int dim, Tensor index) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: self_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: self_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: masked_select_out + operator_name: masked_select + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_select.out(Tensor self, Tensor mask, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_select + operator_name: masked_select + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_select(Tensor self, Tensor mask) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_select_backward + operator_name: masked_select_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_select_backward(Tensor grad, Tensor input, Tensor mask) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: nonzero_out + operator_name: nonzero + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nonzero.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nonzero + operator_name: nonzero + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nonzero(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nonzero_static_out + operator_name: nonzero_static + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nonzero_static.out(Tensor self, *, SymInt size, int fill_value=-1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: size + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: fill_value + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: size + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: fill_value + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nonzero_static + operator_name: nonzero_static + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nonzero_static(Tensor self, *, SymInt size, int fill_value=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: size + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: fill_value + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: size + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: fill_value + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nonzero_numpy + operator_name: nonzero_numpy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nonzero_numpy(Tensor self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::vector (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: argwhere + operator_name: argwhere + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::argwhere(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gather_out + operator_name: gather + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::gather.out(Tensor self, int dim, Tensor index, *, bool sparse_grad=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: sparse_grad + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: sparse_grad + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gather + operator_name: gather + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::gather(Tensor self, int dim, Tensor index, *, bool sparse_grad=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: sparse_grad + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: sparse_grad + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: gather_backward + operator_name: gather_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::gather_backward(Tensor grad, Tensor self, int dim, Tensor index, bool sparse_grad) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: sparse_grad + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: sparse_grad + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: gather_out + operator_name: gather + overload_name: dimname_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::gather.dimname_out(Tensor self, Dimname dim, Tensor index, *, bool sparse_grad=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: sparse_grad + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Dimname, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: sparse_grad + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: gather + operator_name: gather + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::gather.dimname(Tensor self, Dimname dim, Tensor index, *, bool sparse_grad=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: sparse_grad + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: sparse_grad + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _gather_sparse_backward + operator_name: _gather_sparse_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_gather_sparse_backward(Tensor self, int dim, Tensor index, Tensor grad) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: addcmul_out + operator_name: addcmul + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::addcmul.out(Tensor self, Tensor tensor1, Tensor tensor2, *, Scalar value=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addcmul + operator_name: addcmul + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addcmul(Tensor self, Tensor tensor1, Tensor tensor2, *, Scalar value=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addcmul_ + operator_name: addcmul_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addcmul_(Tensor(a!) self, Tensor tensor1, Tensor tensor2, *, Scalar value=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addcdiv_out + operator_name: addcdiv + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::addcdiv.out(Tensor self, Tensor tensor1, Tensor tensor2, *, Scalar value=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addcdiv + operator_name: addcdiv + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addcdiv(Tensor self, Tensor tensor1, Tensor tensor2, *, Scalar value=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: addcdiv_ + operator_name: addcdiv_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::addcdiv_(Tensor(a!) self, Tensor tensor1, Tensor tensor2, *, Scalar value=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tensor2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: value + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cross_entropy_loss + operator_name: cross_entropy_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cross_entropy_loss(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, SymInt ignore_index=-100, float label_smoothing=0.0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: label_smoothing + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: label_smoothing + type: double + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: triangular_solve_out + operator_name: triangular_solve + overload_name: X + manual_kernel_registration: false + category_override: '' + schema_string: aten::triangular_solve.X(Tensor self, Tensor A, bool upper=True, bool transpose=False, bool unitriangular=False, *, Tensor(a!) X, Tensor(b!) M) -> (Tensor(a!) solution, Tensor(b!) cloned_coefficient) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: solution + is_nullable: false + name: X + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: cloned_coefficient + is_nullable: false + name: M + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: upper + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: unitriangular + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, bool, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: upper + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: unitriangular + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: solution + is_nullable: false + name: X + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: cloned_coefficient + is_nullable: false + name: M + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: solution + name: X + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: cloned_coefficient + name: M + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: triangular_solve + operator_name: triangular_solve + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::triangular_solve(Tensor self, Tensor A, bool upper=True, bool transpose=False, bool unitriangular=False) -> (Tensor solution, Tensor cloned_coefficient) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: upper + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: unitriangular + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: upper + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: unitriangular + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: solution + name: solution + type: at::Tensor + - dynamic_type: at::Tensor + field_name: cloned_coefficient + name: cloned_coefficient + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _linalg_check_errors + operator_name: _linalg_check_errors + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_check_errors(Tensor info, str api_name, *, bool is_matrix) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: info + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: api_name + type: c10::string_view + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_matrix + type: bool + schema_order_cpp_signature: void (const at::Tensor &, c10::string_view, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: info + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: api_name + type: c10::string_view + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_matrix + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_solve_triangular_out + operator_name: linalg_solve_triangular + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_solve_triangular.out(Tensor self, Tensor B, *, bool upper, bool left=True, bool unitriangular=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: unitriangular + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: unitriangular + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_solve_triangular + operator_name: linalg_solve_triangular + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_solve_triangular(Tensor self, Tensor B, *, bool upper, bool left=True, bool unitriangular=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: unitriangular + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: unitriangular + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_vander + operator_name: linalg_vander + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_vander(Tensor x, *, SymInt? N=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: N + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: N + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: svd_out + operator_name: svd + overload_name: U + manual_kernel_registration: false + category_override: '' + schema_string: aten::svd.U(Tensor self, bool some=True, bool compute_uv=True, *, Tensor(a!) U, Tensor(b!) S, Tensor(c!) V) -> (Tensor(a!) U, Tensor(b!) S, Tensor(c!) V) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: U + is_nullable: false + name: U + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: S + is_nullable: false + name: S + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: V + is_nullable: false + name: V + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: some + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_uv + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: some + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_uv + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: U + is_nullable: false + name: U + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: S + is_nullable: false + name: S + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: V + is_nullable: false + name: V + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: U + name: U + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: S + name: S + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: V + name: V + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: svd + operator_name: svd + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::svd(Tensor self, bool some=True, bool compute_uv=True) -> (Tensor U, Tensor S, Tensor V) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: some + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_uv + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: some + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_uv + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: U + name: U + type: at::Tensor + - dynamic_type: at::Tensor + field_name: S + name: S + type: at::Tensor + - dynamic_type: at::Tensor + field_name: V + name: V + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: swapaxes + operator_name: swapaxes + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::swapaxes(Tensor(a) self, int axis0, int axis1) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis1 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis1 + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: swapaxes_ + operator_name: swapaxes_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::swapaxes_(Tensor(a!) self, int axis0, int axis1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis1 + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis1 + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: swapdims + operator_name: swapdims + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::swapdims(Tensor(a) self, int dim0, int dim1) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: swapdims_ + operator_name: swapdims_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::swapdims_(Tensor(a!) self, int dim0, int dim1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: cholesky_out + operator_name: cholesky + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cholesky.out(Tensor self, bool upper=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cholesky + operator_name: cholesky + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cholesky(Tensor self, bool upper=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cholesky_solve_out + operator_name: cholesky_solve + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cholesky_solve.out(Tensor self, Tensor input2, bool upper=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cholesky_solve + operator_name: cholesky_solve + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cholesky_solve(Tensor self, Tensor input2, bool upper=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cholesky_solve_helper + operator_name: _cholesky_solve_helper + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cholesky_solve_helper(Tensor self, Tensor A, bool upper) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: upper + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: upper + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cholesky_inverse + operator_name: cholesky_inverse + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cholesky_inverse(Tensor self, bool upper=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: cholesky_inverse_out + operator_name: cholesky_inverse + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cholesky_inverse.out(Tensor self, bool upper=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: upper + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: qr_out + operator_name: qr + overload_name: Q + manual_kernel_registration: false + category_override: '' + schema_string: aten::qr.Q(Tensor self, bool some=True, *, Tensor(a!) Q, Tensor(b!) R) -> (Tensor(a!) Q, Tensor(b!) R) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: Q + is_nullable: false + name: Q + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: R + is_nullable: false + name: R + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: some + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: some + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: Q + is_nullable: false + name: Q + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: R + is_nullable: false + name: R + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: Q + name: Q + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: R + name: R + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: qr + operator_name: qr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::qr(Tensor self, bool some=True) -> (Tensor Q, Tensor R) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: some + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: some + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: Q + name: Q + type: at::Tensor + - dynamic_type: at::Tensor + field_name: R + name: R + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: geqrf_out + operator_name: geqrf + overload_name: a + manual_kernel_registration: false + category_override: '' + schema_string: aten::geqrf.a(Tensor self, *, Tensor(a!) a, Tensor(b!) tau) -> (Tensor(a!) a, Tensor(b!) tau) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: a + is_nullable: false + name: a + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: tau + is_nullable: false + name: tau + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: a + is_nullable: false + name: a + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: tau + is_nullable: false + name: tau + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: a + name: a + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: tau + name: tau + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: geqrf + operator_name: geqrf + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::geqrf(Tensor self) -> (Tensor a, Tensor tau) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: a + name: a + type: at::Tensor + - dynamic_type: at::Tensor + field_name: tau + name: tau + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: orgqr + operator_name: orgqr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::orgqr(Tensor self, Tensor input2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: orgqr_out + operator_name: orgqr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::orgqr.out(Tensor self, Tensor input2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: ormqr_out + operator_name: ormqr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ormqr.out(Tensor self, Tensor input2, Tensor input3, bool left=True, bool transpose=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input3 + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input3 + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ormqr + operator_name: ormqr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::ormqr(Tensor self, Tensor input2, Tensor input3, bool left=True, bool transpose=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input3 + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input3 + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: transpose + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _lu_with_info + operator_name: _lu_with_info + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_lu_with_info(Tensor self, bool pivot=True, bool check_errors=True) -> (Tensor LU, Tensor pivots, Tensor info) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: pivot + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: pivot + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: check_errors + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: lu_solve_out + operator_name: lu_solve + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lu_solve.out(Tensor self, Tensor LU_data, Tensor LU_pivots, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_pivots + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_pivots + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: lu_solve + operator_name: lu_solve + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lu_solve(Tensor self, Tensor LU_data, Tensor LU_pivots) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_pivots + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_pivots + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: lu_unpack + operator_name: lu_unpack + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lu_unpack(Tensor LU_data, Tensor LU_pivots, bool unpack_data=True, bool unpack_pivots=True) -> (Tensor P, Tensor L, Tensor U) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_pivots + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unpack_data + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unpack_pivots + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_pivots + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unpack_data + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unpack_pivots + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: P + name: P + type: at::Tensor + - dynamic_type: at::Tensor + field_name: L + name: L + type: at::Tensor + - dynamic_type: at::Tensor + field_name: U + name: U + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lu_unpack_out + operator_name: lu_unpack + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lu_unpack.out(Tensor LU_data, Tensor LU_pivots, bool unpack_data=True, bool unpack_pivots=True, *, Tensor(a!) P, Tensor(b!) L, Tensor(c!) U) -> (Tensor(a!) P, Tensor(b!) L, Tensor(c!) U) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: P + is_nullable: false + name: P + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: L + is_nullable: false + name: L + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: U + is_nullable: false + name: U + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_pivots + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unpack_data + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unpack_pivots + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_data + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU_pivots + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unpack_data + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: unpack_pivots + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: P + is_nullable: false + name: P + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: L + is_nullable: false + name: L + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: U + is_nullable: false + name: U + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: P + name: P + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: L + name: L + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: U + name: U + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: multinomial_out + operator_name: multinomial + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::multinomial.out(Tensor self, SymInt num_samples, bool replacement=False, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_samples + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: replacement + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_samples + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: replacement + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: multinomial + operator_name: multinomial + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::multinomial(Tensor self, SymInt num_samples, bool replacement=False, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_samples + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: replacement + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_samples + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: replacement + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lgamma_out + operator_name: lgamma + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lgamma.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lgamma_ + operator_name: lgamma_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lgamma_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lgamma + operator_name: lgamma + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::lgamma(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: digamma_out + operator_name: digamma + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::digamma.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: digamma + operator_name: digamma + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::digamma(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: polygamma_out + operator_name: polygamma + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::polygamma.out(int n, Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (int64_t, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: polygamma + operator_name: polygamma + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::polygamma(int n, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: polygamma_ + operator_name: polygamma_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::polygamma_(Tensor(a!) self, int n) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: erfinv + operator_name: erfinv + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::erfinv(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: erfinv_ + operator_name: erfinv_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::erfinv_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: erfinv_out + operator_name: erfinv + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::erfinv.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: i0 + operator_name: i0 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::i0(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: i0_ + operator_name: i0_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::i0_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: i0_out + operator_name: i0 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::i0.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sign + operator_name: sign + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sign(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sign_ + operator_name: sign_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sign_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sign_out + operator_name: sign + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sign.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: signbit + operator_name: signbit + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::signbit(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: signbit_out + operator_name: signbit + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::signbit.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: dist + operator_name: dist + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::dist(Tensor self, Tensor other, Scalar p=2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: atan2_out + operator_name: atan2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::atan2.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: atan2_ + operator_name: atan2_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::atan2_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: atan2 + operator_name: atan2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::atan2(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: arctan2 + operator_name: arctan2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arctan2(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arctan2_out + operator_name: arctan2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::arctan2.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: arctan2_ + operator_name: arctan2_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::arctan2_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: lerp_out + operator_name: lerp + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lerp.Scalar_out(Tensor self, Tensor end, Scalar weight, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lerp_out + operator_name: lerp + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lerp.Tensor_out(Tensor self, Tensor end, Tensor weight, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lerp + operator_name: lerp + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::lerp.Scalar(Tensor self, Tensor end, Scalar weight) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: lerp + operator_name: lerp + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::lerp.Tensor(Tensor self, Tensor end, Tensor weight) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: end + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: histc_out + operator_name: histc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::histc.out(Tensor self, int bins=100, Scalar min=0, Scalar max=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 100 + dynamic_type: int64_t + is_nullable: false + name: bins + type: int64_t + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: min + type: const at::Scalar & + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: max + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 100 + dynamic_type: int64_t + is_nullable: false + name: bins + type: int64_t + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: min + type: const at::Scalar & + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: max + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: histc + operator_name: histc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::histc(Tensor self, int bins=100, Scalar min=0, Scalar max=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 100 + dynamic_type: int64_t + is_nullable: false + name: bins + type: int64_t + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: min + type: const at::Scalar & + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: max + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 100 + dynamic_type: int64_t + is_nullable: false + name: bins + type: int64_t + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: min + type: const at::Scalar & + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: max + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: histogram_out + operator_name: histogram + overload_name: bins_tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::histogram.bins_tensor_out(Tensor self, Tensor bins, *, Tensor? weight=None, bool density=False, Tensor(a!) hist, Tensor(b!) bin_edges) -> (Tensor(a!) hist, Tensor(b!) bin_edges) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: hist + is_nullable: false + name: hist + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: bin_edges + is_nullable: false + name: bin_edges + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bins + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bins + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: hist + is_nullable: false + name: hist + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: bin_edges + is_nullable: false + name: bin_edges + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: hist + name: hist + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: bin_edges + name: bin_edges + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: histogram + operator_name: histogram + overload_name: bins_tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::histogram.bins_tensor(Tensor self, Tensor bins, *, Tensor? weight=None, bool density=False) -> (Tensor hist, Tensor bin_edges) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bins + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bins + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: hist + name: hist + type: at::Tensor + - dynamic_type: at::Tensor + field_name: bin_edges + name: bin_edges + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: histogram_out + operator_name: histogram + overload_name: bin_ct_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::histogram.bin_ct_out(Tensor self, int bins=100, *, float[]? range=None, Tensor? weight=None, bool density=False, Tensor(a!) hist, Tensor(b!) bin_edges) -> (Tensor(a!) hist, Tensor(b!) bin_edges) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: hist + is_nullable: false + name: hist + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: bin_edges + is_nullable: false + name: bin_edges + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 100 + dynamic_type: int64_t + is_nullable: false + name: bins + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, ::std::optional>, const ::std::optional &, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 100 + dynamic_type: int64_t + is_nullable: false + name: bins + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: hist + is_nullable: false + name: hist + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: bin_edges + is_nullable: false + name: bin_edges + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: hist + name: hist + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: bin_edges + name: bin_edges + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: histogram + operator_name: histogram + overload_name: bin_ct + manual_kernel_registration: false + category_override: '' + schema_string: aten::histogram.bin_ct(Tensor self, int bins=100, *, float[]? range=None, Tensor? weight=None, bool density=False) -> (Tensor hist, Tensor bin_edges) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 100 + dynamic_type: int64_t + is_nullable: false + name: bins + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, ::std::optional>, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 100 + dynamic_type: int64_t + is_nullable: false + name: bins + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: hist + name: hist + type: at::Tensor + - dynamic_type: at::Tensor + field_name: bin_edges + name: bin_edges + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _histogramdd_bin_edges + operator_name: _histogramdd_bin_edges + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_histogramdd_bin_edges(Tensor self, int[] bins, *, float[]? range=None, Tensor? weight=None, bool density=False) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: bins + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::IntArrayRef, ::std::optional>, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: bins + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _histogramdd_from_bin_cts + operator_name: _histogramdd_from_bin_cts + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_histogramdd_from_bin_cts(Tensor self, int[] bins, *, float[]? range=None, Tensor? weight=None, bool density=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: bins + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional>, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: bins + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _histogramdd_from_bin_tensors + operator_name: _histogramdd_from_bin_tensors + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_histogramdd_from_bin_tensors(Tensor self, Tensor[] bins, *, Tensor? weight=None, bool density=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: bins + type: at::TensorList + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::TensorList, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: bins + type: at::TensorList + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: histogramdd + operator_name: histogramdd + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::histogramdd(Tensor self, int[] bins, float[]? range=None, Tensor? weight=None, bool density=False) -> (Tensor hist, Tensor[] bin_edges) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: bins + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: density + type: bool + schema_order_cpp_signature: ::std::tuple> (const at::Tensor &, at::IntArrayRef, ::std::optional>, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: bins + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: density + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: hist + name: hist + type: at::Tensor + - dynamic_type: at::TensorList + field_name: bin_edges + name: bin_edges + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: histogramdd + operator_name: histogramdd + overload_name: int_bins + manual_kernel_registration: false + category_override: '' + schema_string: aten::histogramdd.int_bins(Tensor self, int bins, float[]? range=None, Tensor? weight=None, bool density=False) -> (Tensor hist, Tensor[] bin_edges) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: bins + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: density + type: bool + schema_order_cpp_signature: ::std::tuple> (const at::Tensor &, int64_t, ::std::optional>, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: bins + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: density + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: hist + name: hist + type: at::Tensor + - dynamic_type: at::TensorList + field_name: bin_edges + name: bin_edges + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: histogramdd + operator_name: histogramdd + overload_name: TensorList_bins + manual_kernel_registration: false + category_override: '' + schema_string: aten::histogramdd.TensorList_bins(Tensor self, Tensor[] bins, float[]? range=None, Tensor? weight=None, bool density=False) -> (Tensor hist, Tensor[] bin_edges) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: bins + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: density + type: bool + schema_order_cpp_signature: ::std::tuple> (const at::Tensor &, at::TensorList, ::std::optional>, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: bins + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: density + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: hist + name: hist + type: at::Tensor + - dynamic_type: at::TensorList + field_name: bin_edges + name: bin_edges + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fmod_out + operator_name: fmod + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fmod.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fmod + operator_name: fmod + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::fmod.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fmod_ + operator_name: fmod_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::fmod_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fmod_out + operator_name: fmod + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fmod.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fmod + operator_name: fmod + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::fmod.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fmod_ + operator_name: fmod_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::fmod_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hypot_out + operator_name: hypot + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hypot.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hypot + operator_name: hypot + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hypot(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hypot_ + operator_name: hypot_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hypot_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: igamma_out + operator_name: igamma + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::igamma.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: igamma + operator_name: igamma + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::igamma(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: igamma_ + operator_name: igamma_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::igamma_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: igammac_out + operator_name: igammac + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::igammac.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: igammac + operator_name: igammac + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::igammac(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: igammac_ + operator_name: igammac_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::igammac_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nextafter_out + operator_name: nextafter + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nextafter.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nextafter + operator_name: nextafter + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nextafter(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nextafter_ + operator_name: nextafter_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nextafter_(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: remainder_out + operator_name: remainder + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::remainder.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: remainder + operator_name: remainder + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::remainder.Scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: remainder_ + operator_name: remainder_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::remainder_.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: remainder_out + operator_name: remainder + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::remainder.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: remainder + operator_name: remainder + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::remainder.Tensor(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: remainder_ + operator_name: remainder_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::remainder_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: remainder + operator_name: remainder + overload_name: Scalar_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::remainder.Scalar_Tensor(Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: min + operator_name: min + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::min(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: min_out + operator_name: min + overload_name: unary_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::min.unary_out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fmin + operator_name: fmin + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fmin(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fmin_out + operator_name: fmin + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fmin.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max + operator_name: max + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fmax + operator_name: fmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fmax(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fmax_out + operator_name: fmax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fmax.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: maximum + operator_name: maximum + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::maximum(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: maximum_out + operator_name: maximum + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::maximum.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max + operator_name: max + overload_name: other + manual_kernel_registration: false + category_override: '' + schema_string: aten::max.other(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: max_out + operator_name: max + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::max.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: max_out + operator_name: max + overload_name: unary_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::max.unary_out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: minimum + operator_name: minimum + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::minimum(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: minimum_out + operator_name: minimum + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::minimum.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: min_out + operator_name: min + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::min.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: min + operator_name: min + overload_name: other + manual_kernel_registration: false + category_override: '' + schema_string: aten::min.other(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: quantile + operator_name: quantile + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantile(Tensor self, Tensor q, int? dim=None, bool keepdim=False, *, str interpolation='linear') -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional, bool, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: quantile_out + operator_name: quantile + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantile.out(Tensor self, Tensor q, int? dim=None, bool keepdim=False, *, str interpolation='linear', Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, bool, c10::string_view, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: quantile + operator_name: quantile + overload_name: scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantile.scalar(Tensor self, float q, int? dim=None, bool keepdim=False, *, str interpolation='linear') -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: q + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, ::std::optional, bool, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: q + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: quantile_out + operator_name: quantile + overload_name: scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantile.scalar_out(Tensor self, float q, int? dim=None, bool keepdim=False, *, str interpolation='linear', Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: q + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, ::std::optional, bool, c10::string_view, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: q + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nanquantile + operator_name: nanquantile + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanquantile(Tensor self, Tensor q, int? dim=None, bool keepdim=False, *, str interpolation='linear') -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional, bool, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nanquantile_out + operator_name: nanquantile + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanquantile.out(Tensor self, Tensor q, int? dim=None, bool keepdim=False, *, str interpolation='linear', Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, bool, c10::string_view, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nanquantile + operator_name: nanquantile + overload_name: scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanquantile.scalar(Tensor self, float q, int? dim=None, bool keepdim=False, *, str interpolation='linear') -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: q + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, ::std::optional, bool, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: q + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nanquantile_out + operator_name: nanquantile + overload_name: scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanquantile.scalar_out(Tensor self, float q, int? dim=None, bool keepdim=False, *, str interpolation='linear', Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: q + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, ::std::optional, bool, c10::string_view, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: q + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: '"linear"' + dynamic_type: c10::string_view + is_nullable: false + kwarg_only: true + name: interpolation + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sort_out + operator_name: sort + overload_name: values + manual_kernel_registration: false + category_override: '' + schema_string: aten::sort.values(Tensor self, int dim=-1, bool descending=False, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sort_out + operator_name: sort + overload_name: values_stable + manual_kernel_registration: false + category_override: '' + schema_string: aten::sort.values_stable(Tensor self, *, bool? stable, int dim=-1, bool descending=False, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: stable + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, ::std::optional, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: stable + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sort + operator_name: sort + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sort(Tensor self, int dim=-1, bool descending=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sort + operator_name: sort + overload_name: stable + manual_kernel_registration: false + category_override: '' + schema_string: aten::sort.stable(Tensor self, *, bool? stable, int dim=-1, bool descending=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: stable + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, ::std::optional, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: stable + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sort_out + operator_name: sort + overload_name: dimname_values + manual_kernel_registration: false + category_override: '' + schema_string: aten::sort.dimname_values(Tensor self, Dimname dim, bool descending=False, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sort_out + operator_name: sort + overload_name: dimname_values_stable + manual_kernel_registration: false + category_override: '' + schema_string: aten::sort.dimname_values_stable(Tensor self, *, bool? stable, Dimname dim, bool descending=False, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: stable + type: ::std::optional + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + kwarg_only: true + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, ::std::optional, at::Dimname, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: stable + type: ::std::optional + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + kwarg_only: true + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sort + operator_name: sort + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::sort.dimname(Tensor self, Dimname dim, bool descending=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: sort + operator_name: sort + overload_name: dimname_stable + manual_kernel_registration: false + category_override: '' + schema_string: aten::sort.dimname_stable(Tensor self, *, bool? stable, Dimname dim, bool descending=False) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: stable + type: ::std::optional + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + kwarg_only: true + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, ::std::optional, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: stable + type: ::std::optional + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + kwarg_only: true + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: msort_out + operator_name: msort + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::msort.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: msort + operator_name: msort + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::msort(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: argsort + operator_name: argsort + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::argsort(Tensor self, int dim=-1, bool descending=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: argsort + operator_name: argsort + overload_name: stable + manual_kernel_registration: false + category_override: '' + schema_string: aten::argsort.stable(Tensor self, *, bool stable, int dim=-1, bool descending=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: stable + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: stable + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: argsort_out + operator_name: argsort + overload_name: stable_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::argsort.stable_out(Tensor self, *, bool stable, int dim=-1, bool descending=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: stable + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, bool, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: stable + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: descending + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: argsort + operator_name: argsort + overload_name: dimname + manual_kernel_registration: false + category_override: '' + schema_string: aten::argsort.dimname(Tensor self, Dimname dim, bool descending=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Dimname, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Dimname + is_nullable: false + name: dim + type: at::Dimname + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: descending + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: topk_out + operator_name: topk + overload_name: values + manual_kernel_registration: false + category_override: '' + schema_string: aten::topk.values(Tensor self, SymInt k, int dim=-1, bool largest=True, bool sorted=True, *, Tensor(a!) values, Tensor(b!) indices) -> (Tensor(a!) values, Tensor(b!) indices) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: largest + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, int64_t, bool, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: largest + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: values + is_nullable: false + name: values + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: indices + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: topk + operator_name: topk + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::topk(Tensor self, SymInt k, int dim=-1, bool largest=True, bool sorted=True) -> (Tensor values, Tensor indices) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: largest + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: largest + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: values + name: values + type: at::Tensor + - dynamic_type: at::Tensor + field_name: indices + name: indices + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: all + operator_name: all + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::all(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: all_out + operator_name: all + overload_name: all_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::all.all_out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: any + operator_name: any + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::any(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: any_out + operator_name: any + overload_name: all_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::any.all_out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: renorm_out + operator_name: renorm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::renorm.out(Tensor self, Scalar p, int dim, Scalar maxnorm, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: maxnorm + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, int64_t, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: maxnorm + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: renorm + operator_name: renorm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::renorm(Tensor self, Scalar p, int dim, Scalar maxnorm) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: maxnorm + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, int64_t, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: maxnorm + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: renorm_ + operator_name: renorm_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::renorm_(Tensor(a!) self, Scalar p, int dim, Scalar maxnorm) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: maxnorm + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, int64_t, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: maxnorm + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unfold + operator_name: unfold + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unfold(Tensor(a) self, int dimension, int size, int step) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: unfold_backward + operator_name: unfold_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unfold_backward(Tensor grad_in, SymInt[] input_sizes, int dim, int size, int step) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_in + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_in + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: equal + operator_name: equal + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::equal(Tensor self, Tensor other) -> bool + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: bool (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: bool + name: result + type: bool + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pow_out + operator_name: pow + overload_name: Tensor_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::pow.Tensor_Tensor_out(Tensor self, Tensor exponent, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pow + operator_name: pow + overload_name: Tensor_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::pow.Tensor_Tensor(Tensor self, Tensor exponent) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pow_out + operator_name: pow + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::pow.Scalar_out(Scalar self, Tensor exponent, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pow + operator_name: pow + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::pow.Scalar(Scalar self, Tensor exponent) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pow_out + operator_name: pow + overload_name: Tensor_Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::pow.Tensor_Scalar_out(Tensor self, Scalar exponent, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pow + operator_name: pow + overload_name: Tensor_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::pow.Tensor_Scalar(Tensor self, Scalar exponent) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pow_ + operator_name: pow_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::pow_.Scalar(Tensor(a!) self, Scalar exponent) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pow_ + operator_name: pow_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::pow_.Tensor(Tensor(a!) self, Tensor exponent) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: float_power_out + operator_name: float_power + overload_name: Tensor_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::float_power.Tensor_Tensor_out(Tensor self, Tensor exponent, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: float_power + operator_name: float_power + overload_name: Tensor_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::float_power.Tensor_Tensor(Tensor self, Tensor exponent) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: float_power_out + operator_name: float_power + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::float_power.Scalar_out(Scalar self, Tensor exponent, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: float_power + operator_name: float_power + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::float_power.Scalar(Scalar self, Tensor exponent) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: float_power_out + operator_name: float_power + overload_name: Tensor_Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::float_power.Tensor_Scalar_out(Tensor self, Scalar exponent, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: float_power + operator_name: float_power + overload_name: Tensor_Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::float_power.Tensor_Scalar(Tensor self, Scalar exponent) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: float_power_ + operator_name: float_power_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::float_power_.Scalar(Tensor(a!) self, Scalar exponent) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: float_power_ + operator_name: float_power_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::float_power_.Tensor(Tensor(a!) self, Tensor exponent) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: exponent + type: const at::Tensor & + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: normal_ + operator_name: normal_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal_(Tensor(a!) self, float mean=0, float std=1, *, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, double, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: normal_functional + operator_name: normal_functional + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal_functional(Tensor self, float mean=0, float std=1, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: normal_out + operator_name: normal + overload_name: Tensor_float_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal.Tensor_float_out(Tensor mean, float std=1, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: normal + operator_name: normal + overload_name: Tensor_float + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal.Tensor_float(Tensor mean, float std=1, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: normal_out + operator_name: normal + overload_name: float_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal.float_Tensor_out(float mean, Tensor std, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: std + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (double, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: std + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: normal + operator_name: normal + overload_name: float_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal.float_Tensor(float mean, Tensor std, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: std + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (double, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: std + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: normal_out + operator_name: normal + overload_name: Tensor_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal.Tensor_Tensor_out(Tensor mean, Tensor std, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: std + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: std + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: normal + operator_name: normal + overload_name: Tensor_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal.Tensor_Tensor(Tensor mean, Tensor std, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: std + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: std + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: normal + operator_name: normal + overload_name: float_float + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal.float_float(float mean, float std, SymInt[] size, *, Generator? generator=None, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (double, double, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: normal_out + operator_name: normal + overload_name: float_float_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal.float_float_out(float mean, float std, SymInt[] size, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (double, double, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: alias + operator_name: alias + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::alias(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _amp_foreach_non_finite_check_and_unscale_ + operator_name: _amp_foreach_non_finite_check_and_unscale_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_amp_foreach_non_finite_check_and_unscale_(Tensor(a!)[] self, Tensor(b!) found_inf, Tensor inv_scale) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: inv_scale + type: const at::Tensor & + schema_order_cpp_signature: void (at::TensorList, at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: inv_scale + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _amp_update_scale_ + operator_name: _amp_update_scale_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_amp_update_scale_(Tensor(a!) self, Tensor(b!) growth_tracker, Tensor found_inf, float scale_growth_factor, float scale_backoff_factor, int growth_interval) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: growth_tracker + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_growth_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_backoff_factor + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: growth_interval + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::Tensor &, const at::Tensor &, double, double, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: growth_tracker + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_growth_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_backoff_factor + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: growth_interval + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add + operator_name: _foreach_add + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add.Scalar(Tensor[] self, Scalar scalar) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add_ + operator_name: _foreach_add_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add_.Scalar(Tensor(a!)[] self, Scalar scalar) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add + operator_name: _foreach_add + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add.List(Tensor[] self, Tensor[] other, *, Scalar alpha=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add_ + operator_name: _foreach_add_ + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add_.List(Tensor(a!)[] self, Tensor[] other, *, Scalar alpha=1) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add + operator_name: _foreach_add + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add.ScalarList(Tensor[] self, Scalar[] scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add_ + operator_name: _foreach_add_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add_.ScalarList(Tensor(a!)[] self, Scalar[] scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add + operator_name: _foreach_add + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add.Tensor(Tensor[] self, Tensor other, *, Scalar alpha=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add_ + operator_name: _foreach_add_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add_.Tensor(Tensor(a!)[] self, Tensor other, *, Scalar alpha=1) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sub + operator_name: _foreach_sub + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sub.Scalar(Tensor[] self, Scalar scalar) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sub_ + operator_name: _foreach_sub_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sub_.Scalar(Tensor(a!)[] self, Scalar scalar) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sub + operator_name: _foreach_sub + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sub.List(Tensor[] self, Tensor[] other, *, Scalar alpha=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sub_ + operator_name: _foreach_sub_ + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sub_.List(Tensor(a!)[] self, Tensor[] other, *, Scalar alpha=1) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sub + operator_name: _foreach_sub + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sub.ScalarList(Tensor[] self, Scalar[] scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sub_ + operator_name: _foreach_sub_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sub_.ScalarList(Tensor(a!)[] self, Scalar[] scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul + operator_name: _foreach_mul + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul.Scalar(Tensor[] self, Scalar scalar) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul_ + operator_name: _foreach_mul_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul_.Scalar(Tensor(a!)[] self, Scalar scalar) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul + operator_name: _foreach_mul + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul.List(Tensor[] self, Tensor[] other) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul_ + operator_name: _foreach_mul_ + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul_.List(Tensor(a!)[] self, Tensor[] other) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul + operator_name: _foreach_mul + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul.ScalarList(Tensor[] self, Scalar[] scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul_ + operator_name: _foreach_mul_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul_.ScalarList(Tensor(a!)[] self, Scalar[] scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul + operator_name: _foreach_mul + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul.Tensor(Tensor[] self, Tensor other) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul_ + operator_name: _foreach_mul_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul_.Tensor(Tensor(a!)[] self, Tensor other) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: void (at::TensorList, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div + operator_name: _foreach_div + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div.Scalar(Tensor[] self, Scalar scalar) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div_ + operator_name: _foreach_div_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div_.Scalar(Tensor(a!)[] self, Scalar scalar) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div + operator_name: _foreach_div + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div.List(Tensor[] self, Tensor[] other) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div_ + operator_name: _foreach_div_ + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div_.List(Tensor(a!)[] self, Tensor[] other) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div + operator_name: _foreach_div + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div.ScalarList(Tensor[] self, Scalar[] scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div_ + operator_name: _foreach_div_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div_.ScalarList(Tensor(a!)[] self, Scalar[] scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div + operator_name: _foreach_div + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div.Tensor(Tensor[] self, Tensor other) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div_ + operator_name: _foreach_div_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div_.Tensor(Tensor(a!)[] self, Tensor other) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: void (at::TensorList, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_max + operator_name: _foreach_clamp_max + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_max.Scalar(Tensor[] self, Scalar scalar) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_max_ + operator_name: _foreach_clamp_max_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_max_.Scalar(Tensor(a!)[] self, Scalar scalar) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_max + operator_name: _foreach_clamp_max + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_max.List(Tensor[] self, Tensor[] other) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_max_ + operator_name: _foreach_clamp_max_ + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_max_.List(Tensor(a!)[] self, Tensor[] other) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_max + operator_name: _foreach_clamp_max + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_max.ScalarList(Tensor[] self, Scalar[] scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_max_ + operator_name: _foreach_clamp_max_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_max_.ScalarList(Tensor(a!)[] self, Scalar[] scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_min + operator_name: _foreach_clamp_min + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_min.Scalar(Tensor[] self, Scalar scalar) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_min_ + operator_name: _foreach_clamp_min_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_min_.Scalar(Tensor(a!)[] self, Scalar scalar) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_min + operator_name: _foreach_clamp_min + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_min.List(Tensor[] self, Tensor[] other) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_min_ + operator_name: _foreach_clamp_min_ + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_min_.List(Tensor(a!)[] self, Tensor[] other) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_min + operator_name: _foreach_clamp_min + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_min.ScalarList(Tensor[] self, Scalar[] scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_min_ + operator_name: _foreach_clamp_min_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_min_.ScalarList(Tensor(a!)[] self, Scalar[] scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_maximum + operator_name: _foreach_maximum + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_maximum.Scalar(Tensor[] self, Scalar scalar) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_maximum_ + operator_name: _foreach_maximum_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_maximum_.Scalar(Tensor(a!)[] self, Scalar scalar) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_maximum + operator_name: _foreach_maximum + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_maximum.List(Tensor[] self, Tensor[] other) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_maximum_ + operator_name: _foreach_maximum_ + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_maximum_.List(Tensor(a!)[] self, Tensor[] other) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_maximum + operator_name: _foreach_maximum + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_maximum.ScalarList(Tensor[] self, Scalar[] scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_maximum_ + operator_name: _foreach_maximum_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_maximum_.ScalarList(Tensor(a!)[] self, Scalar[] scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_minimum + operator_name: _foreach_minimum + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_minimum.Scalar(Tensor[] self, Scalar scalar) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_minimum_ + operator_name: _foreach_minimum_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_minimum_.Scalar(Tensor(a!)[] self, Scalar scalar) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_minimum + operator_name: _foreach_minimum + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_minimum.List(Tensor[] self, Tensor[] other) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_minimum_ + operator_name: _foreach_minimum_ + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_minimum_.List(Tensor(a!)[] self, Tensor[] other) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_minimum + operator_name: _foreach_minimum + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_minimum.ScalarList(Tensor[] self, Scalar[] scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_minimum_ + operator_name: _foreach_minimum_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_minimum_.ScalarList(Tensor(a!)[] self, Scalar[] scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcdiv + operator_name: _foreach_addcdiv + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcdiv.Scalar(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar value=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcdiv + operator_name: _foreach_addcdiv + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcdiv.ScalarList(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar[] scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcdiv + operator_name: _foreach_addcdiv + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcdiv.Tensor(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Tensor scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, at::TensorList, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcdiv_ + operator_name: _foreach_addcdiv_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcdiv_.Scalar(Tensor(a!)[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar value=1) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcdiv_ + operator_name: _foreach_addcdiv_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcdiv_.ScalarList(Tensor(a!)[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar[] scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcdiv_ + operator_name: _foreach_addcdiv_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcdiv_.Tensor(Tensor(a!)[] self, Tensor[] tensor1, Tensor[] tensor2, Tensor scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcmul + operator_name: _foreach_addcmul + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcmul.Scalar(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar value=1) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcmul + operator_name: _foreach_addcmul + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcmul.ScalarList(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar[] scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcmul + operator_name: _foreach_addcmul + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcmul.Tensor(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Tensor scalars) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, at::TensorList, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcmul_ + operator_name: _foreach_addcmul_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcmul_.Scalar(Tensor(a!)[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar value=1) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcmul_ + operator_name: _foreach_addcmul_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcmul_.ScalarList(Tensor(a!)[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar[] scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcmul_ + operator_name: _foreach_addcmul_ + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcmul_.Tensor(Tensor(a!)[] self, Tensor[] tensor1, Tensor[] tensor2, Tensor scalars) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, const at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_abs + operator_name: _foreach_abs + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_abs(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_abs_ + operator_name: _foreach_abs_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_abs_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_acos + operator_name: _foreach_acos + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_acos(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_acos_ + operator_name: _foreach_acos_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_acos_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_asin + operator_name: _foreach_asin + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_asin(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_asin_ + operator_name: _foreach_asin_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_asin_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_atan + operator_name: _foreach_atan + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_atan(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_atan_ + operator_name: _foreach_atan_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_atan_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_ceil + operator_name: _foreach_ceil + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_ceil(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_ceil_ + operator_name: _foreach_ceil_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_ceil_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_cos + operator_name: _foreach_cos + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_cos(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_cos_ + operator_name: _foreach_cos_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_cos_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_cosh + operator_name: _foreach_cosh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_cosh(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_cosh_ + operator_name: _foreach_cosh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_cosh_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_erf + operator_name: _foreach_erf + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_erf(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_erf_ + operator_name: _foreach_erf_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_erf_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_erfc + operator_name: _foreach_erfc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_erfc(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_erfc_ + operator_name: _foreach_erfc_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_erfc_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_exp + operator_name: _foreach_exp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_exp(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_exp_ + operator_name: _foreach_exp_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_exp_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_expm1 + operator_name: _foreach_expm1 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_expm1(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_expm1_ + operator_name: _foreach_expm1_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_expm1_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_floor + operator_name: _foreach_floor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_floor(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_floor_ + operator_name: _foreach_floor_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_floor_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_frac + operator_name: _foreach_frac + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_frac(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_frac_ + operator_name: _foreach_frac_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_frac_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lerp + operator_name: _foreach_lerp + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lerp.List(Tensor[] self, Tensor[] tensors1, Tensor[] weights) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weights + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weights + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lerp_ + operator_name: _foreach_lerp_ + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lerp_.List(Tensor(a!)[] self, Tensor[] tensors1, Tensor[] weights) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weights + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weights + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lerp + operator_name: _foreach_lerp + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lerp.Scalar(Tensor[] self, Tensor[] tensors1, Scalar weight) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lerp_ + operator_name: _foreach_lerp_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lerp_.Scalar(Tensor(a!)[] self, Tensor[] tensors1, Scalar weight) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lerp + operator_name: _foreach_lerp + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lerp.ScalarList(Tensor[] self, Tensor[] tensors1, Scalar[] weight) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: weight + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: weight + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lerp_ + operator_name: _foreach_lerp_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lerp_.ScalarList(Tensor(a!)[] self, Tensor[] tensors1, Scalar[] weight) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: weight + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: weight + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lgamma + operator_name: _foreach_lgamma + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lgamma(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lgamma_ + operator_name: _foreach_lgamma_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lgamma_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log + operator_name: _foreach_log + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log_ + operator_name: _foreach_log_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log10 + operator_name: _foreach_log10 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log10(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log10_ + operator_name: _foreach_log10_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log10_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log1p + operator_name: _foreach_log1p + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log1p(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log1p_ + operator_name: _foreach_log1p_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log1p_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log2 + operator_name: _foreach_log2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log2(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log2_ + operator_name: _foreach_log2_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log2_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_max + operator_name: _foreach_max + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_max(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_neg + operator_name: _foreach_neg + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_neg(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_neg_ + operator_name: _foreach_neg_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_neg_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_norm + operator_name: _foreach_norm + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_norm.Scalar(Tensor[] self, Scalar ord=2, ScalarType? dtype=None) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Scalar &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_pow + operator_name: _foreach_pow + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_pow.List(Tensor[] self, Tensor[] exponent) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exponent + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exponent + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_pow + operator_name: _foreach_pow + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_pow.Scalar(Tensor[] self, Scalar exponent) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + schema_order_cpp_signature: ::std::vector (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_pow + operator_name: _foreach_pow + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_pow.ScalarList(Tensor[] self, Scalar[] exponent) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: exponent + type: at::ArrayRef + schema_order_cpp_signature: ::std::vector (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: exponent + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_pow + operator_name: _foreach_pow + overload_name: ScalarAndTensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_pow.ScalarAndTensor(Scalar self, Tensor[] exponent) -> Tensor[] + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exponent + type: at::TensorList + schema_order_cpp_signature: ::std::vector (const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exponent + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_pow_ + operator_name: _foreach_pow_ + overload_name: List + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_pow_.List(Tensor(a!)[] self, Tensor[] exponent) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exponent + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exponent + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_pow_ + operator_name: _foreach_pow_ + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_pow_.Scalar(Tensor(a!)[] self, Scalar exponent) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_pow_ + operator_name: _foreach_pow_ + overload_name: ScalarList + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_pow_.ScalarList(Tensor(a!)[] self, Scalar[] exponent) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: exponent + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: exponent + type: at::ArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_reciprocal + operator_name: _foreach_reciprocal + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_reciprocal(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_reciprocal_ + operator_name: _foreach_reciprocal_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_reciprocal_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_round + operator_name: _foreach_round + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_round(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_round_ + operator_name: _foreach_round_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_round_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_rsqrt + operator_name: _foreach_rsqrt + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_rsqrt(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_rsqrt_ + operator_name: _foreach_rsqrt_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_rsqrt_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sigmoid + operator_name: _foreach_sigmoid + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sigmoid(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sigmoid_ + operator_name: _foreach_sigmoid_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sigmoid_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sign + operator_name: _foreach_sign + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sign(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sign_ + operator_name: _foreach_sign_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sign_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sin + operator_name: _foreach_sin + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sin(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sin_ + operator_name: _foreach_sin_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sin_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sinh + operator_name: _foreach_sinh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sinh(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sinh_ + operator_name: _foreach_sinh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sinh_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sqrt + operator_name: _foreach_sqrt + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sqrt(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sqrt_ + operator_name: _foreach_sqrt_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sqrt_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_tan + operator_name: _foreach_tan + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_tan(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_tan_ + operator_name: _foreach_tan_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_tan_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_tanh + operator_name: _foreach_tanh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_tanh(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_tanh_ + operator_name: _foreach_tanh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_tanh_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_trunc + operator_name: _foreach_trunc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_trunc(Tensor[] self) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_trunc_ + operator_name: _foreach_trunc_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_trunc_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_zero_ + operator_name: _foreach_zero_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_zero_(Tensor(a!)[] self) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_copy_ + operator_name: _foreach_copy_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_copy_(Tensor(a!)[] self, Tensor[] src, bool non_blocking=False) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: src + type: at::TensorList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: void (at::TensorList, at::TensorList, bool) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: src + type: at::TensorList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_copy + operator_name: _foreach_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_copy(Tensor[] self, Tensor[] src, bool non_blocking=False) -> Tensor[] self_out + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: src + type: at::TensorList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: ::std::vector (at::TensorList, at::TensorList, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: src + type: at::TensorList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bucketize + operator_name: bucketize + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bucketize.Tensor(Tensor self, Tensor boundaries, *, bool out_int32=False, bool right=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: boundaries + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: boundaries + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bucketize_out + operator_name: bucketize + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bucketize.Tensor_out(Tensor self, Tensor boundaries, *, bool out_int32=False, bool right=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: boundaries + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: boundaries + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: bucketize + operator_name: bucketize + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::bucketize.Scalar(Scalar self, Tensor boundaries, *, bool out_int32=False, bool right=False) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: boundaries + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: boundaries + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: searchsorted + operator_name: searchsorted + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::searchsorted.Tensor(Tensor sorted_sequence, Tensor self, *, bool out_int32=False, bool right=False, str? side=None, Tensor? sorter=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sorted_sequence + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: side + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: sorter + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool, bool, ::std::optional, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sorted_sequence + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: side + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: sorter + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: searchsorted_out + operator_name: searchsorted + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::searchsorted.Tensor_out(Tensor sorted_sequence, Tensor self, *, bool out_int32=False, bool right=False, str? side=None, Tensor? sorter=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sorted_sequence + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: side + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: sorter + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, bool, ::std::optional, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sorted_sequence + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: side + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: sorter + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: searchsorted + operator_name: searchsorted + overload_name: Scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::searchsorted.Scalar(Tensor sorted_sequence, Scalar self, *, bool out_int32=False, bool right=False, str? side=None, Tensor? sorter=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sorted_sequence + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: side + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: sorter + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, bool, bool, ::std::optional, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sorted_sequence + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: side + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: sorter + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: searchsorted_out + operator_name: searchsorted + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::searchsorted.Scalar_out(Tensor sorted_sequence, Scalar self, *, bool out_int32=False, bool right=False, str? side=None, Tensor? sorter=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sorted_sequence + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: side + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: sorter + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, bool, bool, ::std::optional, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sorted_sequence + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: side + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: sorter + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _convert_indices_from_coo_to_csr + operator_name: _convert_indices_from_coo_to_csr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convert_indices_from_coo_to_csr(Tensor self, int size, *, bool out_int32=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _convert_indices_from_coo_to_csr_out + operator_name: _convert_indices_from_coo_to_csr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convert_indices_from_coo_to_csr.out(Tensor self, int size, *, bool out_int32=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _convert_indices_from_csr_to_coo + operator_name: _convert_indices_from_csr_to_coo + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convert_indices_from_csr_to_coo(Tensor crow_indices, Tensor col_indices, *, bool out_int32=False, bool transpose=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: transpose + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: transpose + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _convert_indices_from_csr_to_coo_out + operator_name: _convert_indices_from_csr_to_coo + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convert_indices_from_csr_to_coo.out(Tensor crow_indices, Tensor col_indices, *, bool out_int32=False, bool transpose=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: transpose + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: crow_indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: col_indices + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: transpose + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mse_loss_out + operator_name: mse_loss + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mse_loss.out(Tensor self, Tensor target, int reduction=Mean, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mse_loss + operator_name: mse_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mse_loss(Tensor self, Tensor target, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mse_loss_backward_out + operator_name: mse_loss_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::mse_loss_backward.grad_input(Tensor grad_output, Tensor self, Tensor target, int reduction, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mse_loss_backward + operator_name: mse_loss_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mse_loss_backward(Tensor grad_output, Tensor self, Tensor target, int reduction) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: l1_loss + operator_name: l1_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::l1_loss(Tensor self, Tensor target, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: multi_margin_loss_out + operator_name: multi_margin_loss + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::multi_margin_loss.out(Tensor self, Tensor target, Scalar p=1, Scalar margin=1, Tensor? weight=None, int reduction=Mean, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: margin + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, const ::std::optional &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: margin + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: multi_margin_loss + operator_name: multi_margin_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::multi_margin_loss(Tensor self, Tensor target, Scalar p=1, Scalar margin=1, Tensor? weight=None, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: margin + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, const ::std::optional &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: margin + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: multi_margin_loss_backward_out + operator_name: multi_margin_loss_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::multi_margin_loss_backward.grad_input(Tensor grad_output, Tensor self, Tensor target, Scalar p, Scalar margin, Tensor? weight=None, int reduction=Mean, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: margin + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, const ::std::optional &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: margin + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: multi_margin_loss_backward + operator_name: multi_margin_loss_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::multi_margin_loss_backward(Tensor grad_output, Tensor self, Tensor target, Scalar p, Scalar margin, Tensor? weight=None, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: margin + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, const ::std::optional &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: margin + type: const at::Scalar & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: multilabel_margin_loss_out + operator_name: multilabel_margin_loss + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::multilabel_margin_loss.out(Tensor self, Tensor target, int reduction=Mean, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: multilabel_margin_loss + operator_name: multilabel_margin_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::multilabel_margin_loss(Tensor self, Tensor target, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: multilabel_margin_loss_forward_out + operator_name: multilabel_margin_loss_forward + overload_name: output + manual_kernel_registration: false + category_override: '' + schema_string: aten::multilabel_margin_loss_forward.output(Tensor self, Tensor target, int reduction, *, Tensor(a!) output, Tensor(b!) is_target) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: is_target + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: is_target + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: output + type: at::Tensor & + - dynamic_type: at::Tensor + name: is_target + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: multilabel_margin_loss_forward + operator_name: multilabel_margin_loss_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::multilabel_margin_loss_forward(Tensor self, Tensor target, int reduction) -> (Tensor output, Tensor is_target) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: is_target + name: is_target + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: multilabel_margin_loss_backward_out + operator_name: multilabel_margin_loss_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::multilabel_margin_loss_backward.grad_input(Tensor grad_output, Tensor self, Tensor target, int reduction, Tensor is_target, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: is_target + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: is_target + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: multilabel_margin_loss_backward + operator_name: multilabel_margin_loss_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::multilabel_margin_loss_backward(Tensor grad_output, Tensor self, Tensor target, int reduction, Tensor is_target) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: is_target + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: is_target + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nll_loss_out + operator_name: nll_loss + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss.out(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, SymInt ignore_index=-100, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nll_loss_nd + operator_name: nll_loss_nd + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss_nd(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, SymInt ignore_index=-100) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nll_loss + operator_name: nll_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, SymInt ignore_index=-100) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nll_loss_forward_out + operator_name: nll_loss_forward + overload_name: output + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss_forward.output(Tensor self, Tensor target, Tensor? weight, int reduction, SymInt ignore_index, *, Tensor(a!) output, Tensor(b!) total_weight) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: output + type: at::Tensor & + - dynamic_type: at::Tensor + name: total_weight + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nll_loss_forward + operator_name: nll_loss_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss_forward(Tensor self, Tensor target, Tensor? weight, int reduction, SymInt ignore_index) -> (Tensor output, Tensor total_weight) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: total_weight + name: total_weight + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nll_loss_backward_out + operator_name: nll_loss_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss_backward.grad_input(Tensor grad_output, Tensor self, Tensor target, Tensor? weight, int reduction, SymInt ignore_index, Tensor total_weight, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nll_loss_backward + operator_name: nll_loss_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss_backward(Tensor grad_output, Tensor self, Tensor target, Tensor? weight, int reduction, SymInt ignore_index, Tensor total_weight) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nll_loss2d_out + operator_name: nll_loss2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss2d.out(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, SymInt ignore_index=-100, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nll_loss2d + operator_name: nll_loss2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss2d(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, SymInt ignore_index=-100) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: -100 + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nll_loss2d_forward_out + operator_name: nll_loss2d_forward + overload_name: output + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss2d_forward.output(Tensor self, Tensor target, Tensor? weight, int reduction, SymInt ignore_index, *, Tensor(a!) output, Tensor(b!) total_weight) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: output + type: at::Tensor & + - dynamic_type: at::Tensor + name: total_weight + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nll_loss2d_forward + operator_name: nll_loss2d_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss2d_forward(Tensor self, Tensor target, Tensor? weight, int reduction, SymInt ignore_index) -> (Tensor output, Tensor total_weight) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: total_weight + name: total_weight + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nll_loss2d_backward_out + operator_name: nll_loss2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss2d_backward.grad_input(Tensor grad_output, Tensor self, Tensor target, Tensor? weight, int reduction, SymInt ignore_index, Tensor total_weight, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: nll_loss2d_backward + operator_name: nll_loss2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nll_loss2d_backward(Tensor grad_output, Tensor self, Tensor target, Tensor? weight, int reduction, SymInt ignore_index, Tensor total_weight) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ignore_index + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total_weight + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: smooth_l1_loss_out + operator_name: smooth_l1_loss + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::smooth_l1_loss.out(Tensor self, Tensor target, int reduction=Mean, float beta=1.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: beta + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: beta + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: smooth_l1_loss + operator_name: smooth_l1_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::smooth_l1_loss(Tensor self, Tensor target, int reduction=Mean, float beta=1.0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: beta + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: beta + type: double + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: smooth_l1_loss_backward_out + operator_name: smooth_l1_loss_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::smooth_l1_loss_backward.grad_input(Tensor grad_output, Tensor self, Tensor target, int reduction, float beta, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: smooth_l1_loss_backward + operator_name: smooth_l1_loss_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::smooth_l1_loss_backward(Tensor grad_output, Tensor self, Tensor target, int reduction, float beta) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: huber_loss_out + operator_name: huber_loss + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::huber_loss.out(Tensor self, Tensor target, int reduction=Mean, float delta=1.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: delta + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: delta + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: huber_loss + operator_name: huber_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::huber_loss(Tensor self, Tensor target, int reduction=Mean, float delta=1.0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: delta + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: delta + type: double + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: huber_loss_backward_out + operator_name: huber_loss_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::huber_loss_backward.out(Tensor grad_output, Tensor self, Tensor target, int reduction, float delta, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: delta + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: delta + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: huber_loss_backward + operator_name: huber_loss_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::huber_loss_backward(Tensor grad_output, Tensor self, Tensor target, int reduction, float delta) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: delta + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: delta + type: double + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: soft_margin_loss_out + operator_name: soft_margin_loss + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::soft_margin_loss.out(Tensor self, Tensor target, int reduction=Mean, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: soft_margin_loss + operator_name: soft_margin_loss + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::soft_margin_loss(Tensor self, Tensor target, int reduction=Mean) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: soft_margin_loss_backward_out + operator_name: soft_margin_loss_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::soft_margin_loss_backward.grad_input(Tensor grad_output, Tensor self, Tensor target, int reduction, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: soft_margin_loss_backward + operator_name: soft_margin_loss_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::soft_margin_loss_backward(Tensor grad_output, Tensor self, Tensor target, int reduction) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: elu_out + operator_name: elu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::elu.out(Tensor self, Scalar alpha=1, Scalar scale=1, Scalar input_scale=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: scale + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: input_scale + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: scale + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: input_scale + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: elu + operator_name: elu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::elu(Tensor self, Scalar alpha=1, Scalar scale=1, Scalar input_scale=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: scale + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: input_scale + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: scale + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: input_scale + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: elu_backward_out + operator_name: elu_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::elu_backward.grad_input(Tensor grad_output, Scalar alpha, Scalar scale, Scalar input_scale, bool is_result, Tensor self_or_result, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: input_scale + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_result + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self_or_result + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Scalar &, bool, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: input_scale + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_result + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self_or_result + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: elu_backward + operator_name: elu_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::elu_backward(Tensor grad_output, Scalar alpha, Scalar scale, Scalar input_scale, bool is_result, Tensor self_or_result) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: input_scale + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_result + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self_or_result + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Scalar &, bool, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scale + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: input_scale + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_result + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self_or_result + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: elu_ + operator_name: elu_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::elu_(Tensor(a!) self, Scalar alpha=1, Scalar scale=1, Scalar input_scale=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: scale + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: input_scale + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: scale + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: input_scale + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: glu_out + operator_name: glu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::glu.out(Tensor self, int dim=-1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: glu + operator_name: glu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::glu(Tensor self, int dim=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: glu_backward_out + operator_name: glu_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::glu_backward.grad_input(Tensor grad_output, Tensor self, int dim, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: glu_backward + operator_name: glu_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::glu_backward(Tensor grad_output, Tensor self, int dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: glu_jvp + operator_name: glu_jvp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::glu_jvp(Tensor glu, Tensor x, Tensor dx, int dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dx + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dx + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: glu_backward_jvp + operator_name: glu_backward_jvp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::glu_backward_jvp(Tensor grad_x, Tensor grad_glu, Tensor x, Tensor dgrad_glu, Tensor dx, int dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dgrad_glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dx + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dgrad_glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dx + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardsigmoid_out + operator_name: hardsigmoid + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardsigmoid.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardsigmoid + operator_name: hardsigmoid + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardsigmoid(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardsigmoid_ + operator_name: hardsigmoid_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardsigmoid_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardsigmoid_backward_out + operator_name: hardsigmoid_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardsigmoid_backward.grad_input(Tensor grad_output, Tensor self, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardsigmoid_backward + operator_name: hardsigmoid_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardsigmoid_backward(Tensor grad_output, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardtanh_out + operator_name: hardtanh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardtanh.out(Tensor self, Scalar min_val=-1, Scalar max_val=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: const at::Scalar & + is_nullable: false + name: min_val + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: max_val + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: const at::Scalar & + is_nullable: false + name: min_val + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: max_val + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardtanh + operator_name: hardtanh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardtanh(Tensor self, Scalar min_val=-1, Scalar max_val=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: const at::Scalar & + is_nullable: false + name: min_val + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: max_val + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: const at::Scalar & + is_nullable: false + name: min_val + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: max_val + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardtanh_backward_out + operator_name: hardtanh_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardtanh_backward.grad_input(Tensor grad_output, Tensor self, Scalar min_val, Scalar max_val, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: min_val + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: max_val + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: min_val + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: max_val + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardtanh_backward + operator_name: hardtanh_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardtanh_backward(Tensor grad_output, Tensor self, Scalar min_val, Scalar max_val) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: min_val + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: max_val + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: min_val + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: max_val + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardtanh_ + operator_name: hardtanh_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardtanh_(Tensor(a!) self, Scalar min_val=-1, Scalar max_val=1) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: -1 + dynamic_type: const at::Scalar & + is_nullable: false + name: min_val + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: max_val + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: -1 + dynamic_type: const at::Scalar & + is_nullable: false + name: min_val + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: max_val + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardswish_out + operator_name: hardswish + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardswish.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardswish + operator_name: hardswish + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardswish(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardswish_ + operator_name: hardswish_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardswish_(Tensor(a!) self) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::Tensor &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardswish_backward + operator_name: hardswish_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardswish_backward(Tensor grad_output, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: leaky_relu_out + operator_name: leaky_relu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::leaky_relu.out(Tensor self, Scalar negative_slope=0.01, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.01 + dynamic_type: const at::Scalar & + is_nullable: false + name: negative_slope + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.01 + dynamic_type: const at::Scalar & + is_nullable: false + name: negative_slope + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: leaky_relu + operator_name: leaky_relu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::leaky_relu(Tensor self, Scalar negative_slope=0.01) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.01 + dynamic_type: const at::Scalar & + is_nullable: false + name: negative_slope + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.01 + dynamic_type: const at::Scalar & + is_nullable: false + name: negative_slope + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: leaky_relu_backward_out + operator_name: leaky_relu_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::leaky_relu_backward.grad_input(Tensor grad_output, Tensor self, Scalar negative_slope, bool self_is_result, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: negative_slope + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: self_is_result + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: negative_slope + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: self_is_result + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: leaky_relu_backward + operator_name: leaky_relu_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::leaky_relu_backward(Tensor grad_output, Tensor self, Scalar negative_slope, bool self_is_result) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: negative_slope + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: self_is_result + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: negative_slope + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: self_is_result + type: bool + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: leaky_relu_ + operator_name: leaky_relu_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::leaky_relu_(Tensor(a!) self, Scalar negative_slope=0.01) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0.01 + dynamic_type: const at::Scalar & + is_nullable: false + name: negative_slope + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + default: 0.01 + dynamic_type: const at::Scalar & + is_nullable: false + name: negative_slope + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_sigmoid_out + operator_name: log_sigmoid + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_sigmoid.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: log_sigmoid + operator_name: log_sigmoid + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_sigmoid(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: log_sigmoid_forward_out + operator_name: log_sigmoid_forward + overload_name: output + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_sigmoid_forward.output(Tensor self, *, Tensor(a!) output, Tensor(b!) buffer) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: buffer + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: buffer + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: output + type: at::Tensor & + - dynamic_type: at::Tensor + name: buffer + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_sigmoid_forward + operator_name: log_sigmoid_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_sigmoid_forward(Tensor self) -> (Tensor output, Tensor buffer) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: buffer + name: buffer + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_sigmoid_backward_out + operator_name: log_sigmoid_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_sigmoid_backward.grad_input(Tensor grad_output, Tensor self, Tensor buffer, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: buffer + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: buffer + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_sigmoid_backward + operator_name: log_sigmoid_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_sigmoid_backward(Tensor grad_output, Tensor self, Tensor buffer) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: buffer + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: buffer + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rrelu_with_noise_out + operator_name: rrelu_with_noise + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rrelu_with_noise.out(Tensor self, Tensor(b!) noise, Scalar lower=0.125, Scalar upper=0.3333333333333333, bool training=False, Generator? generator=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &, const at::Scalar &, const at::Scalar &, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rrelu_with_noise + operator_name: rrelu_with_noise + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rrelu_with_noise(Tensor self, Tensor(b!) noise, Scalar lower=0.125, Scalar upper=0.3333333333333333, bool training=False, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Tensor &, const at::Scalar &, const at::Scalar &, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rrelu_with_noise_backward + operator_name: rrelu_with_noise_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rrelu_with_noise_backward(Tensor grad_output, Tensor self, Tensor noise, Scalar lower, Scalar upper, bool training, bool self_is_result) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: self_is_result + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: self_is_result + type: bool + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: rrelu_with_noise_ + operator_name: rrelu_with_noise_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rrelu_with_noise_(Tensor(a!) self, Tensor(b!) noise, Scalar lower=0.125, Scalar upper=0.3333333333333333, bool training=False, Generator? generator=None) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::Tensor &, at::Tensor &, const at::Scalar &, const at::Scalar &, bool, ::std::optional) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: softplus_out + operator_name: softplus + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::softplus.out(Tensor self, Scalar beta=1, Scalar threshold=20, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: beta + type: const at::Scalar & + - annotation: null + default: 20 + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: beta + type: const at::Scalar & + - annotation: null + default: 20 + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: softplus + operator_name: softplus + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::softplus(Tensor self, Scalar beta=1, Scalar threshold=20) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: beta + type: const at::Scalar & + - annotation: null + default: 20 + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: beta + type: const at::Scalar & + - annotation: null + default: 20 + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: softplus_backward_out + operator_name: softplus_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::softplus_backward.grad_input(Tensor grad_output, Tensor self, Scalar beta, Scalar threshold, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: beta + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: beta + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: softplus_backward + operator_name: softplus_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::softplus_backward(Tensor grad_output, Tensor self, Scalar beta, Scalar threshold) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: beta + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: beta + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: threshold + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: softshrink_out + operator_name: softshrink + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::softshrink.out(Tensor self, Scalar lambd=0.5, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: softshrink + operator_name: softshrink + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::softshrink(Tensor self, Scalar lambd=0.5) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: softshrink_backward_out + operator_name: softshrink_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::softshrink_backward.grad_input(Tensor grad_output, Tensor self, Scalar lambd, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: softshrink_backward + operator_name: softshrink_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::softshrink_backward(Tensor grad_output, Tensor self, Scalar lambd) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lambd + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_avg_pool2d_out + operator_name: adaptive_avg_pool2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_avg_pool2d.out(Tensor self, SymInt[2] output_size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_avg_pool2d + operator_name: adaptive_avg_pool2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_avg_pool2d(Tensor self, SymInt[2] output_size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: mkldnn_adaptive_avg_pool2d + operator_name: mkldnn_adaptive_avg_pool2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_adaptive_avg_pool2d(Tensor self, int[2] output_size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_adaptive_avg_pool2d_out + operator_name: mkldnn_adaptive_avg_pool2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_adaptive_avg_pool2d.out(Tensor self, int[2] output_size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_adaptive_avg_pool2d_backward + operator_name: mkldnn_adaptive_avg_pool2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_adaptive_avg_pool2d_backward(Tensor grad_output, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _adaptive_avg_pool2d + operator_name: _adaptive_avg_pool2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_adaptive_avg_pool2d(Tensor self, SymInt[2] output_size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _adaptive_avg_pool2d_backward + operator_name: _adaptive_avg_pool2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_adaptive_avg_pool2d_backward(Tensor grad_output, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_avg_pool3d_out + operator_name: adaptive_avg_pool3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_avg_pool3d.out(Tensor self, SymInt[3] output_size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_avg_pool3d + operator_name: adaptive_avg_pool3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_avg_pool3d(Tensor self, SymInt[3] output_size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _adaptive_avg_pool3d + operator_name: _adaptive_avg_pool3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_adaptive_avg_pool3d(Tensor self, SymInt[3] output_size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_avg_pool3d_backward_out + operator_name: adaptive_avg_pool3d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_avg_pool3d_backward.grad_input(Tensor grad_output, Tensor self, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _adaptive_avg_pool3d_backward + operator_name: _adaptive_avg_pool3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_adaptive_avg_pool3d_backward(Tensor grad_output, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_max_pool2d_out + operator_name: adaptive_max_pool2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_max_pool2d.out(Tensor self, int[2] output_size, *, Tensor(a!) out, Tensor(b!) indices) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + - dynamic_type: at::Tensor + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_max_pool2d + operator_name: adaptive_max_pool2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_max_pool2d(Tensor self, int[2] output_size) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_max_pool2d_backward_out + operator_name: adaptive_max_pool2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_max_pool2d_backward.grad_input(Tensor grad_output, Tensor self, Tensor indices, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_max_pool2d_backward + operator_name: adaptive_max_pool2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_max_pool2d_backward(Tensor grad_output, Tensor self, Tensor indices) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_max_pool3d_out + operator_name: adaptive_max_pool3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_max_pool3d.out(Tensor self, int[3] output_size, *, Tensor(a!) out, Tensor(b!) indices) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + - dynamic_type: at::Tensor + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_max_pool3d + operator_name: adaptive_max_pool3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_max_pool3d(Tensor self, int[3] output_size) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_max_pool3d_backward_out + operator_name: adaptive_max_pool3d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_max_pool3d_backward.grad_input(Tensor grad_output, Tensor self, Tensor indices, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_max_pool3d_backward + operator_name: adaptive_max_pool3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_max_pool3d_backward(Tensor grad_output, Tensor self, Tensor indices) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: avg_pool2d_out + operator_name: avg_pool2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::avg_pool2d.out(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, bool ceil_mode=False, bool count_include_pad=True, int? divisor_override=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: avg_pool2d + operator_name: avg_pool2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::avg_pool2d(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, bool ceil_mode=False, bool count_include_pad=True, int? divisor_override=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: avg_pool2d_backward_out + operator_name: avg_pool2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::avg_pool2d_backward.grad_input(Tensor grad_output, Tensor self, int[2] kernel_size, int[2] stride, int[2] padding, bool ceil_mode, bool count_include_pad, int? divisor_override, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: avg_pool2d_backward + operator_name: avg_pool2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::avg_pool2d_backward(Tensor grad_output, Tensor self, int[2] kernel_size, int[2] stride, int[2] padding, bool ceil_mode, bool count_include_pad, int? divisor_override) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: avg_pool3d_out + operator_name: avg_pool3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::avg_pool3d.out(Tensor self, int[3] kernel_size, int[3] stride=[], int[3] padding=0, bool ceil_mode=False, bool count_include_pad=True, int? divisor_override=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: avg_pool3d + operator_name: avg_pool3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::avg_pool3d(Tensor self, int[3] kernel_size, int[3] stride=[], int[3] padding=0, bool ceil_mode=False, bool count_include_pad=True, int? divisor_override=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: avg_pool3d_backward_out + operator_name: avg_pool3d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::avg_pool3d_backward.grad_input(Tensor grad_output, Tensor self, int[3] kernel_size, int[3] stride, int[3] padding, bool ceil_mode, bool count_include_pad, int? divisor_override, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: avg_pool3d_backward + operator_name: avg_pool3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::avg_pool3d_backward(Tensor grad_output, Tensor self, int[3] kernel_size, int[3] stride, int[3] padding, bool ceil_mode, bool count_include_pad, int? divisor_override) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: divisor_override + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fractional_max_pool2d_out + operator_name: fractional_max_pool2d + overload_name: output + manual_kernel_registration: false + category_override: '' + schema_string: aten::fractional_max_pool2d.output(Tensor self, int[2] kernel_size, int[2] output_size, Tensor random_samples, *, Tensor(a!) output, Tensor(b!) indices) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: random_samples + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: random_samples + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: output + type: at::Tensor & + - dynamic_type: at::Tensor + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fractional_max_pool2d + operator_name: fractional_max_pool2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fractional_max_pool2d(Tensor self, int[2] kernel_size, int[2] output_size, Tensor random_samples) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: random_samples + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: random_samples + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fractional_max_pool2d_backward_out + operator_name: fractional_max_pool2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::fractional_max_pool2d_backward.grad_input(Tensor grad_output, Tensor self, int[2] kernel_size, int[2] output_size, Tensor indices, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fractional_max_pool2d_backward + operator_name: fractional_max_pool2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fractional_max_pool2d_backward(Tensor grad_output, Tensor self, int[2] kernel_size, int[2] output_size, Tensor indices) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fractional_max_pool3d_out + operator_name: fractional_max_pool3d + overload_name: output + manual_kernel_registration: false + category_override: '' + schema_string: aten::fractional_max_pool3d.output(Tensor self, int[3] kernel_size, int[3] output_size, Tensor random_samples, *, Tensor(a!) output, Tensor(b!) indices) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: random_samples + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: random_samples + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: output + type: at::Tensor & + - dynamic_type: at::Tensor + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fractional_max_pool3d + operator_name: fractional_max_pool3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fractional_max_pool3d(Tensor self, int[3] kernel_size, int[3] output_size, Tensor random_samples) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: random_samples + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: random_samples + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fractional_max_pool3d_backward_out + operator_name: fractional_max_pool3d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::fractional_max_pool3d_backward.grad_input(Tensor grad_output, Tensor self, int[3] kernel_size, int[3] output_size, Tensor indices, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fractional_max_pool3d_backward + operator_name: fractional_max_pool3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fractional_max_pool3d_backward(Tensor grad_output, Tensor self, int[3] kernel_size, int[3] output_size, Tensor indices) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool2d_with_indices_out + operator_name: max_pool2d_with_indices + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool2d_with_indices.out(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False, *, Tensor(a!) out, Tensor(b!) indices) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + - dynamic_type: at::Tensor + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool2d_with_indices + operator_name: max_pool2d_with_indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool2d_with_indices(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool2d_with_indices_backward_out + operator_name: max_pool2d_with_indices_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool2d_with_indices_backward.grad_input(Tensor grad_output, Tensor self, int[2] kernel_size, int[2] stride, int[2] padding, int[2] dilation, bool ceil_mode, Tensor indices, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool2d_with_indices_backward + operator_name: max_pool2d_with_indices_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool2d_with_indices_backward(Tensor grad_output, Tensor self, int[2] kernel_size, int[2] stride, int[2] padding, int[2] dilation, bool ceil_mode, Tensor indices) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool3d_with_indices_out + operator_name: max_pool3d_with_indices + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool3d_with_indices.out(Tensor self, int[3] kernel_size, int[3] stride=[], int[3] padding=0, int[3] dilation=1, bool ceil_mode=False, *, Tensor(a!) out, Tensor(b!) indices) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: indices + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + - dynamic_type: at::Tensor + name: indices + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool3d_with_indices + operator_name: max_pool3d_with_indices + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool3d_with_indices(Tensor self, int[3] kernel_size, int[3] stride=[], int[3] padding=0, int[3] dilation=1, bool ceil_mode=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool3d_with_indices_backward_out + operator_name: max_pool3d_with_indices_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool3d_with_indices_backward.grad_input(Tensor grad_output, Tensor self, int[3] kernel_size, int[3] stride, int[3] padding, int[3] dilation, bool ceil_mode, Tensor indices, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool3d_with_indices_backward + operator_name: max_pool3d_with_indices_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool3d_with_indices_backward(Tensor grad_output, Tensor self, int[3] kernel_size, int[3] stride, int[3] padding, int[3] dilation, bool ceil_mode, Tensor indices) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_unpool2d_out + operator_name: max_unpool2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_unpool2d.out(Tensor self, Tensor indices, SymInt[2] output_size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_unpool2d + operator_name: max_unpool2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_unpool2d(Tensor self, Tensor indices, SymInt[2] output_size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_unpool3d_out + operator_name: max_unpool3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_unpool3d.out(Tensor self, Tensor indices, SymInt[3] output_size, int[3] stride, int[3] padding, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_unpool3d + operator_name: max_unpool3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_unpool3d(Tensor self, Tensor indices, SymInt[3] output_size, int[3] stride, int[3] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad1d_out + operator_name: reflection_pad1d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad1d.out(Tensor self, SymInt[2] padding, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad1d + operator_name: reflection_pad1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad1d(Tensor self, SymInt[2] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad1d_backward_out + operator_name: reflection_pad1d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad1d_backward.grad_input(Tensor grad_output, Tensor self, SymInt[2] padding, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad1d_backward + operator_name: reflection_pad1d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad1d_backward(Tensor grad_output, Tensor self, SymInt[2] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad2d_out + operator_name: reflection_pad2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad2d.out(Tensor self, SymInt[4] padding, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad2d + operator_name: reflection_pad2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad2d(Tensor self, SymInt[4] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad2d_backward_out + operator_name: reflection_pad2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad2d_backward.grad_input(Tensor grad_output, Tensor self, SymInt[4] padding, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad2d_backward + operator_name: reflection_pad2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad2d_backward(Tensor grad_output, Tensor self, SymInt[4] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad3d_out + operator_name: reflection_pad3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad3d.out(Tensor self, SymInt[6] padding, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad3d + operator_name: reflection_pad3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad3d(Tensor self, SymInt[6] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad3d_backward_out + operator_name: reflection_pad3d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad3d_backward.grad_input(Tensor grad_output, Tensor self, SymInt[6] padding, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: reflection_pad3d_backward + operator_name: reflection_pad3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::reflection_pad3d_backward(Tensor grad_output, Tensor self, SymInt[6] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad1d_out + operator_name: replication_pad1d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad1d.out(Tensor self, SymInt[2] padding, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad1d + operator_name: replication_pad1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad1d(Tensor self, SymInt[2] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad1d_backward_out + operator_name: replication_pad1d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad1d_backward.grad_input(Tensor grad_output, Tensor self, SymInt[2] padding, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad1d_backward + operator_name: replication_pad1d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad1d_backward(Tensor grad_output, Tensor self, SymInt[2] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad2d_out + operator_name: replication_pad2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad2d.out(Tensor self, SymInt[4] padding, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad2d + operator_name: replication_pad2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad2d(Tensor self, SymInt[4] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad2d_backward_out + operator_name: replication_pad2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad2d_backward.grad_input(Tensor grad_output, Tensor self, SymInt[4] padding, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad2d_backward + operator_name: replication_pad2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad2d_backward(Tensor grad_output, Tensor self, SymInt[4] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 4 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad3d_out + operator_name: replication_pad3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad3d.out(Tensor self, SymInt[6] padding, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad3d + operator_name: replication_pad3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad3d(Tensor self, SymInt[6] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad3d_backward_out + operator_name: replication_pad3d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad3d_backward.grad_input(Tensor grad_output, Tensor self, SymInt[6] padding, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: replication_pad3d_backward + operator_name: replication_pad3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::replication_pad3d_backward(Tensor grad_output, Tensor self, SymInt[6] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 6 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _pad_circular + operator_name: _pad_circular + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pad_circular(Tensor self, SymInt[] pad) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: pad + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: pad + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _pad_enum + operator_name: _pad_enum + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pad_enum(Tensor self, SymInt[] pad, int mode, float? value=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: pad + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: value + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: pad + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: value + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: pad + operator_name: pad + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::pad(Tensor self, SymInt[] pad, str mode="constant", float? value=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: pad + type: at::IntArrayRef + - annotation: null + default: '"constant"' + dynamic_type: c10::string_view + is_nullable: false + name: mode + type: c10::string_view + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: value + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, c10::string_view, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: pad + type: at::IntArrayRef + - annotation: null + default: '"constant"' + dynamic_type: c10::string_view + is_nullable: false + name: mode + type: c10::string_view + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: value + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: upsample_linear1d + operator_name: upsample_linear1d + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_linear1d.vec(Tensor input, SymInt[]? output_size, bool align_corners, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: upsample_bilinear2d + operator_name: upsample_bilinear2d + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bilinear2d.vec(Tensor input, SymInt[]? output_size, bool align_corners, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _upsample_bilinear2d_aa + operator_name: _upsample_bilinear2d_aa + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_bilinear2d_aa.vec(Tensor input, SymInt[]? output_size, bool align_corners, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: upsample_trilinear3d + operator_name: upsample_trilinear3d + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_trilinear3d.vec(Tensor input, SymInt[]? output_size, bool align_corners, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: upsample_bicubic2d + operator_name: upsample_bicubic2d + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bicubic2d.vec(Tensor input, SymInt[]? output_size, bool align_corners, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _upsample_bicubic2d_aa + operator_name: _upsample_bicubic2d_aa + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_bicubic2d_aa.vec(Tensor input, SymInt[]? output_size, bool align_corners, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: upsample_nearest1d + operator_name: upsample_nearest1d + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest1d.vec(Tensor input, SymInt[]? output_size, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _upsample_nearest_exact1d + operator_name: _upsample_nearest_exact1d + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact1d.vec(Tensor input, SymInt[]? output_size, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: upsample_nearest2d + operator_name: upsample_nearest2d + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest2d.vec(Tensor input, SymInt[]? output_size, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _upsample_nearest_exact2d + operator_name: _upsample_nearest_exact2d + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact2d.vec(Tensor input, SymInt[]? output_size, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: upsample_nearest3d + operator_name: upsample_nearest3d + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest3d.vec(Tensor input, SymInt[]? output_size, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _upsample_nearest_exact3d + operator_name: _upsample_nearest_exact3d + overload_name: vec + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact3d.vec(Tensor input, SymInt[]? output_size, float[]? scale_factors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: upsample_linear1d_out + operator_name: upsample_linear1d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_linear1d.out(Tensor self, SymInt[1] output_size, bool align_corners, float? scales=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_linear1d + operator_name: upsample_linear1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_linear1d(Tensor self, SymInt[1] output_size, bool align_corners, float? scales=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_linear1d_backward_out + operator_name: upsample_linear1d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_linear1d_backward.grad_input(Tensor grad_output, SymInt[1] output_size, SymInt[3] input_size, bool align_corners, float? scales=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_linear1d_backward + operator_name: upsample_linear1d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_linear1d_backward(Tensor grad_output, SymInt[1] output_size, SymInt[3] input_size, bool align_corners, float? scales=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_bilinear2d_out + operator_name: upsample_bilinear2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bilinear2d.out(Tensor self, SymInt[2] output_size, bool align_corners, float? scales_h=None, float? scales_w=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_bilinear2d + operator_name: upsample_bilinear2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bilinear2d(Tensor self, SymInt[2] output_size, bool align_corners, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_bilinear2d_backward_out + operator_name: upsample_bilinear2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bilinear2d_backward.grad_input(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, bool align_corners, float? scales_h=None, float? scales_w=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_bilinear2d_backward + operator_name: upsample_bilinear2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bilinear2d_backward(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, bool align_corners, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_bilinear2d_aa_out + operator_name: _upsample_bilinear2d_aa + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_bilinear2d_aa.out(Tensor self, SymInt[2] output_size, bool align_corners, float? scales_h=None, float? scales_w=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_bilinear2d_aa + operator_name: _upsample_bilinear2d_aa + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_bilinear2d_aa(Tensor self, SymInt[2] output_size, bool align_corners, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_bilinear2d_aa_backward_out + operator_name: _upsample_bilinear2d_aa_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_bilinear2d_aa_backward.grad_input(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, bool align_corners, float? scales_h=None, float? scales_w=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_bilinear2d_aa_backward + operator_name: _upsample_bilinear2d_aa_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_bilinear2d_aa_backward(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, bool align_corners, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_bicubic2d_out + operator_name: upsample_bicubic2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bicubic2d.out(Tensor self, SymInt[2] output_size, bool align_corners, float? scales_h=None, float? scales_w=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_bicubic2d + operator_name: upsample_bicubic2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bicubic2d(Tensor self, SymInt[2] output_size, bool align_corners, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_bicubic2d_backward_out + operator_name: upsample_bicubic2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bicubic2d_backward.grad_input(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, bool align_corners, float? scales_h=None, float? scales_w=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_bicubic2d_backward + operator_name: upsample_bicubic2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bicubic2d_backward(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, bool align_corners, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_bicubic2d_aa_out + operator_name: _upsample_bicubic2d_aa + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_bicubic2d_aa.out(Tensor self, SymInt[2] output_size, bool align_corners, float? scales_h=None, float? scales_w=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_bicubic2d_aa + operator_name: _upsample_bicubic2d_aa + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_bicubic2d_aa(Tensor self, SymInt[2] output_size, bool align_corners, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_bicubic2d_aa_backward_out + operator_name: _upsample_bicubic2d_aa_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_bicubic2d_aa_backward.grad_input(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, bool align_corners, float? scales_h=None, float? scales_w=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_bicubic2d_aa_backward + operator_name: _upsample_bicubic2d_aa_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_bicubic2d_aa_backward(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, bool align_corners, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_trilinear3d_out + operator_name: upsample_trilinear3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_trilinear3d.out(Tensor self, SymInt[3] output_size, bool align_corners, float? scales_d=None, float? scales_h=None, float? scales_w=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_trilinear3d + operator_name: upsample_trilinear3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_trilinear3d(Tensor self, SymInt[3] output_size, bool align_corners, float? scales_d=None, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_trilinear3d_backward_out + operator_name: upsample_trilinear3d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_trilinear3d_backward.grad_input(Tensor grad_output, SymInt[3] output_size, SymInt[5] input_size, bool align_corners, float? scales_d=None, float? scales_h=None, float? scales_w=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_trilinear3d_backward + operator_name: upsample_trilinear3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_trilinear3d_backward(Tensor grad_output, SymInt[3] output_size, SymInt[5] input_size, bool align_corners, float? scales_d=None, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, bool, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest1d_out + operator_name: upsample_nearest1d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest1d.out(Tensor self, SymInt[1] output_size, float? scales=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact1d_out + operator_name: _upsample_nearest_exact1d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact1d.out(Tensor self, SymInt[1] output_size, float? scales=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest1d + operator_name: upsample_nearest1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest1d(Tensor self, SymInt[1] output_size, float? scales=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact1d + operator_name: _upsample_nearest_exact1d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact1d(Tensor self, SymInt[1] output_size, float? scales=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest1d_backward_out + operator_name: upsample_nearest1d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest1d_backward.grad_input(Tensor grad_output, SymInt[1] output_size, SymInt[3] input_size, float? scales=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact1d_backward_out + operator_name: _upsample_nearest_exact1d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact1d_backward.grad_input(Tensor grad_output, SymInt[1] output_size, SymInt[3] input_size, float? scales=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest1d_backward + operator_name: upsample_nearest1d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest1d_backward(Tensor grad_output, SymInt[1] output_size, SymInt[3] input_size, float? scales=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact1d_backward + operator_name: _upsample_nearest_exact1d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact1d_backward(Tensor grad_output, SymInt[1] output_size, SymInt[3] input_size, float? scales=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest2d_out + operator_name: upsample_nearest2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest2d.out(Tensor self, SymInt[2] output_size, float? scales_h=None, float? scales_w=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact2d_out + operator_name: _upsample_nearest_exact2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact2d.out(Tensor self, SymInt[2] output_size, float? scales_h=None, float? scales_w=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest2d + operator_name: upsample_nearest2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest2d(Tensor self, SymInt[2] output_size, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact2d + operator_name: _upsample_nearest_exact2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact2d(Tensor self, SymInt[2] output_size, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest2d_backward_out + operator_name: upsample_nearest2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest2d_backward.grad_input(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, float? scales_h=None, float? scales_w=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact2d_backward_out + operator_name: _upsample_nearest_exact2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact2d_backward.grad_input(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, float? scales_h=None, float? scales_w=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest2d_backward + operator_name: upsample_nearest2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest2d_backward(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact2d_backward + operator_name: _upsample_nearest_exact2d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact2d_backward(Tensor grad_output, SymInt[2] output_size, SymInt[4] input_size, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 4 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest3d_out + operator_name: upsample_nearest3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest3d.out(Tensor self, SymInt[3] output_size, float? scales_d=None, float? scales_h=None, float? scales_w=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact3d_out + operator_name: _upsample_nearest_exact3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact3d.out(Tensor self, SymInt[3] output_size, float? scales_d=None, float? scales_h=None, float? scales_w=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest3d + operator_name: upsample_nearest3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest3d(Tensor self, SymInt[3] output_size, float? scales_d=None, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact3d + operator_name: _upsample_nearest_exact3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact3d(Tensor self, SymInt[3] output_size, float? scales_d=None, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest3d_backward_out + operator_name: upsample_nearest3d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest3d_backward.grad_input(Tensor grad_output, SymInt[3] output_size, SymInt[5] input_size, float? scales_d=None, float? scales_h=None, float? scales_w=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact3d_backward_out + operator_name: _upsample_nearest_exact3d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact3d_backward.grad_input(Tensor grad_output, SymInt[3] output_size, SymInt[5] input_size, float? scales_d=None, float? scales_h=None, float? scales_w=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest3d_backward + operator_name: upsample_nearest3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest3d_backward(Tensor grad_output, SymInt[3] output_size, SymInt[5] input_size, float? scales_d=None, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _upsample_nearest_exact3d_backward + operator_name: _upsample_nearest_exact3d_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_upsample_nearest_exact3d_backward(Tensor grad_output, SymInt[3] output_size, SymInt[5] input_size, float? scales_d=None, float? scales_h=None, float? scales_w=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + size: 5 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_d + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_h + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: scales_w + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sigmoid_backward_out + operator_name: sigmoid_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::sigmoid_backward.grad_input(Tensor grad_output, Tensor output, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: sigmoid_backward + operator_name: sigmoid_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sigmoid_backward(Tensor grad_output, Tensor output) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logit_backward_out + operator_name: logit_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::logit_backward.grad_input(Tensor grad_output, Tensor self, float? eps=None, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: logit_backward + operator_name: logit_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logit_backward(Tensor grad_output, Tensor self, float? eps=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tanh_backward_out + operator_name: tanh_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::tanh_backward.grad_input(Tensor grad_output, Tensor output, *, Tensor(a!) grad_input) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: tanh_backward + operator_name: tanh_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::tanh_backward(Tensor grad_output, Tensor output) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: slow_conv_transpose2d_out + operator_name: slow_conv_transpose2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv_transpose2d.out(Tensor self, Tensor weight, SymInt[2] kernel_size, Tensor? bias=None, SymInt[2] stride=1, SymInt[2] padding=0, SymInt[2] output_padding=0, SymInt[2] dilation=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: slow_conv_transpose2d + operator_name: slow_conv_transpose2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv_transpose2d(Tensor self, Tensor weight, SymInt[2] kernel_size, Tensor? bias=None, SymInt[2] stride=1, SymInt[2] padding=0, SymInt[2] output_padding=0, SymInt[2] dilation=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: slow_conv_transpose3d_out + operator_name: slow_conv_transpose3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv_transpose3d.out(Tensor self, Tensor weight, SymInt[3] kernel_size, Tensor? bias=None, SymInt[3] stride=1, SymInt[3] padding=0, SymInt[3] output_padding=0, SymInt[3] dilation=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: slow_conv_transpose3d + operator_name: slow_conv_transpose3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv_transpose3d(Tensor self, Tensor weight, SymInt[3] kernel_size, Tensor? bias=None, SymInt[3] stride=1, SymInt[3] padding=0, SymInt[3] output_padding=0, SymInt[3] dilation=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: thnn_conv2d_out + operator_name: thnn_conv2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::thnn_conv2d.out(Tensor self, Tensor weight, SymInt[2] kernel_size, Tensor? bias=None, SymInt[2] stride=1, SymInt[2] padding=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: thnn_conv2d + operator_name: thnn_conv2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::thnn_conv2d(Tensor self, Tensor weight, SymInt[2] kernel_size, Tensor? bias=None, SymInt[2] stride=1, SymInt[2] padding=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _slow_conv2d_forward_out + operator_name: _slow_conv2d_forward + overload_name: output + manual_kernel_registration: false + category_override: '' + schema_string: aten::_slow_conv2d_forward.output(Tensor self, Tensor weight, SymInt[2] kernel_size, Tensor? bias, SymInt[2] stride, SymInt[2] padding, *, Tensor(a!) output) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: output + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _slow_conv2d_forward + operator_name: _slow_conv2d_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_slow_conv2d_forward(Tensor self, Tensor weight, SymInt[2] kernel_size, Tensor? bias, SymInt[2] stride, SymInt[2] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _slow_conv2d_backward_out + operator_name: _slow_conv2d_backward + overload_name: grad_input + manual_kernel_registration: false + category_override: '' + schema_string: aten::_slow_conv2d_backward.grad_input(Tensor grad_output, Tensor self, Tensor weight, SymInt[2] kernel_size, SymInt[2] stride, SymInt[2] padding, *, Tensor(a!) grad_input, Tensor(b!) grad_weight, Tensor(c!) grad_bias) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: grad_weight + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: grad_bias + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: grad_input + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: grad_weight + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: grad_bias + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: grad_input + type: at::Tensor & + - dynamic_type: at::Tensor + name: grad_weight + type: at::Tensor & + - dynamic_type: at::Tensor + name: grad_bias + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _slow_conv2d_backward + operator_name: _slow_conv2d_backward + overload_name: output_mask + manual_kernel_registration: false + category_override: '' + schema_string: aten::_slow_conv2d_backward.output_mask(Tensor grad_output, Tensor self, Tensor weight, SymInt[2] kernel_size, SymInt[2] stride, SymInt[2] padding, bool[3] output_mask) -> (Tensor grad_input, Tensor grad_weight, Tensor grad_bias) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, ::std::array) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + field_name: grad_input + name: grad_input + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_weight + name: grad_weight + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_bias + name: grad_bias + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _conv_depthwise2d_out + operator_name: _conv_depthwise2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_conv_depthwise2d.out(Tensor self, Tensor weight, SymInt[2] kernel_size, Tensor? bias, SymInt[2] stride, SymInt[2] padding, SymInt[2] dilation, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _conv_depthwise2d + operator_name: _conv_depthwise2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_conv_depthwise2d(Tensor self, Tensor weight, SymInt[2] kernel_size, Tensor? bias, SymInt[2] stride, SymInt[2] padding, SymInt[2] dilation) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: conv_depthwise3d + operator_name: conv_depthwise3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv_depthwise3d(Tensor self, Tensor weight, SymInt[3] kernel_size, Tensor? bias, SymInt[3] stride, SymInt[3] padding, SymInt[3] dilation) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: slow_conv3d_out + operator_name: slow_conv3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv3d.out(Tensor self, Tensor weight, SymInt[3] kernel_size, Tensor? bias=None, SymInt[3] stride=1, SymInt[3] padding=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: slow_conv3d + operator_name: slow_conv3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv3d(Tensor self, Tensor weight, SymInt[3] kernel_size, Tensor? bias=None, SymInt[3] stride=1, SymInt[3] padding=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: slow_conv3d_forward_out + operator_name: slow_conv3d_forward + overload_name: output + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv3d_forward.output(Tensor self, Tensor weight, SymInt[3] kernel_size, Tensor? bias, SymInt[3] stride, SymInt[3] padding, *, Tensor(a!) output) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: output + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: output + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: slow_conv3d_forward + operator_name: slow_conv3d_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv3d_forward(Tensor self, Tensor weight, SymInt[3] kernel_size, Tensor? bias, SymInt[3] stride, SymInt[3] padding) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: slow_conv_dilated2d + operator_name: slow_conv_dilated2d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv_dilated2d(Tensor self, Tensor weight, SymInt[2] kernel_size, Tensor? bias=None, SymInt[2] stride=1, SymInt[2] padding=0, SymInt[2] dilation=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: slow_conv_dilated3d + operator_name: slow_conv_dilated3d + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv_dilated3d(Tensor self, Tensor weight, SymInt[3] kernel_size, Tensor? bias=None, SymInt[3] stride=1, SymInt[3] padding=0, SymInt[3] dilation=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: col2im_out + operator_name: col2im + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::col2im.out(Tensor self, SymInt[2] output_size, int[2] kernel_size, int[2] dilation, int[2] padding, int[2] stride, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: col2im + operator_name: col2im + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::col2im(Tensor self, SymInt[2] output_size, int[2] kernel_size, int[2] dilation, int[2] padding, int[2] stride) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: column_stack + operator_name: column_stack + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::column_stack(Tensor[] tensors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: column_stack_out + operator_name: column_stack + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::column_stack.out(Tensor[] tensors, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: im2col_out + operator_name: im2col + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::im2col.out(Tensor self, int[2] kernel_size, int[2] dilation, int[2] padding, int[2] stride, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: im2col + operator_name: im2col + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::im2col(Tensor self, int[2] kernel_size, int[2] dilation, int[2] padding, int[2] stride) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isfinite + operator_name: isfinite + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::isfinite(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: true +- name: isinf + operator_name: isinf + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::isinf(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: record_stream + operator_name: record_stream + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::record_stream(Tensor(a!) self, Stream s) -> () + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Stream + is_nullable: false + name: s + type: at::Stream + schema_order_cpp_signature: void (at::Tensor &, at::Stream) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: at::Stream + is_nullable: false + name: s + type: at::Stream + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isposinf + operator_name: isposinf + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::isposinf(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isposinf_out + operator_name: isposinf + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::isposinf.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isneginf + operator_name: isneginf + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::isneginf(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: isneginf_out + operator_name: isneginf + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::isneginf.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _add_batch_dim + operator_name: _add_batch_dim + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_add_batch_dim(Tensor self, int batch_dim, int level) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: batch_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: batch_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _remove_batch_dim + operator_name: _remove_batch_dim + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_remove_batch_dim(Tensor self, int level, SymInt batch_size, int out_dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: batch_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: out_dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: batch_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: out_dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_entr + operator_name: special_entr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_entr(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_entr_out + operator_name: special_entr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_entr.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_ndtri + operator_name: special_ndtri + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_ndtri(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_ndtri_out + operator_name: special_ndtri + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_ndtri.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_log_ndtr + operator_name: special_log_ndtr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_log_ndtr(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_log_ndtr_out + operator_name: special_log_ndtr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_log_ndtr.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_expm1 + operator_name: special_expm1 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_expm1(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_expm1_out + operator_name: special_expm1 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_expm1.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_exp2 + operator_name: special_exp2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_exp2(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_exp2_out + operator_name: special_exp2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_exp2.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_psi + operator_name: special_psi + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_psi(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_psi_out + operator_name: special_psi + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_psi.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_digamma + operator_name: special_digamma + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_digamma(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_digamma_out + operator_name: special_digamma + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_digamma.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_gammaln + operator_name: special_gammaln + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_gammaln(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_gammaln_out + operator_name: special_gammaln + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_gammaln.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_erf + operator_name: special_erf + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_erf(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_erf_out + operator_name: special_erf + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_erf.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_erfc + operator_name: special_erfc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_erfc(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_erfc_out + operator_name: special_erfc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_erfc.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_erfcx + operator_name: special_erfcx + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_erfcx(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_erfcx_out + operator_name: special_erfcx + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_erfcx.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_erfinv + operator_name: special_erfinv + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_erfinv(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_erfinv_out + operator_name: special_erfinv + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_erfinv.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_ndtr + operator_name: special_ndtr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_ndtr(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_ndtr_out + operator_name: special_ndtr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_ndtr.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_xlog1py + operator_name: special_xlog1py + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlog1py(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_xlog1py + operator_name: special_xlog1py + overload_name: self_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlog1py.self_scalar(Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_xlog1py + operator_name: special_xlog1py + overload_name: other_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlog1py.other_scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_xlog1py_out + operator_name: special_xlog1py + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlog1py.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_xlog1py_out + operator_name: special_xlog1py + overload_name: self_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlog1py.self_scalar_out(Scalar self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_xlog1py_out + operator_name: special_xlog1py + overload_name: other_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlog1py.other_scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_xlogy + operator_name: special_xlogy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlogy(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_xlogy + operator_name: special_xlogy + overload_name: self_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlogy.self_scalar(Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_xlogy + operator_name: special_xlogy + overload_name: other_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlogy.other_scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_xlogy_out + operator_name: special_xlogy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlogy.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_xlogy_out + operator_name: special_xlogy + overload_name: self_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlogy.self_scalar_out(Scalar self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_xlogy_out + operator_name: special_xlogy + overload_name: other_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_xlogy.other_scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_zeta + operator_name: special_zeta + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_zeta(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_zeta + operator_name: special_zeta + overload_name: self_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_zeta.self_scalar(Scalar self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_zeta + operator_name: special_zeta + overload_name: other_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_zeta.other_scalar(Tensor self, Scalar other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_zeta_out + operator_name: special_zeta + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_zeta.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_zeta_out + operator_name: special_zeta + overload_name: self_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_zeta.self_scalar_out(Scalar self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_zeta_out + operator_name: special_zeta + overload_name: other_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_zeta.other_scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_i0 + operator_name: special_i0 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_i0(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_i0_out + operator_name: special_i0 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_i0.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_i0e + operator_name: special_i0e + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_i0e(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_i0e_out + operator_name: special_i0e + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_i0e.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_i1 + operator_name: special_i1 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_i1(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_i1_out + operator_name: special_i1 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_i1.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_i1e + operator_name: special_i1e + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_i1e(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_i1e_out + operator_name: special_i1e + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_i1e.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_logit + operator_name: special_logit + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_logit(Tensor self, float? eps=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_logit_out + operator_name: special_logit + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_logit.out(Tensor self, float? eps=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: eps + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_polygamma + operator_name: special_polygamma + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_polygamma(int n, Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_polygamma_out + operator_name: special_polygamma + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_polygamma.out(int n, Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (int64_t, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_logsumexp + operator_name: special_logsumexp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_logsumexp(Tensor self, int[1] dim, bool keepdim=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_logsumexp_out + operator_name: special_logsumexp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_logsumexp.out(Tensor self, int[1] dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_expit + operator_name: special_expit + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_expit(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_expit_out + operator_name: special_expit + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_expit.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_sinc + operator_name: special_sinc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_sinc(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_sinc_out + operator_name: special_sinc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_sinc.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_round + operator_name: special_round + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_round(Tensor self, *, int decimals=0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: decimals + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: decimals + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_round_out + operator_name: special_round + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_round.out(Tensor self, *, int decimals=0, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: decimals + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: decimals + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_log1p + operator_name: special_log1p + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_log1p(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_log1p_out + operator_name: special_log1p + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_log1p.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_log_softmax + operator_name: special_log_softmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_log_softmax(Tensor self, int dim, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_gammainc_out + operator_name: special_gammainc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_gammainc.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_gammainc + operator_name: special_gammainc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_gammainc(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_gammaincc_out + operator_name: special_gammaincc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_gammaincc.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_gammaincc + operator_name: special_gammaincc + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_gammaincc(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_multigammaln + operator_name: special_multigammaln + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_multigammaln(Tensor self, int p) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: p + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: p + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_multigammaln_out + operator_name: special_multigammaln + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_multigammaln.out(Tensor self, int p, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: p + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: p + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: special_softmax + operator_name: special_softmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_softmax(Tensor self, int dim, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_fft + operator_name: fft_fft + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_fft(Tensor self, SymInt? n=None, int dim=-1, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_fft_out + operator_name: fft_fft + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_fft.out(Tensor self, SymInt? n=None, int dim=-1, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ifft + operator_name: fft_ifft + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ifft(Tensor self, SymInt? n=None, int dim=-1, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ifft_out + operator_name: fft_ifft + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ifft.out(Tensor self, SymInt? n=None, int dim=-1, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_rfft + operator_name: fft_rfft + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_rfft(Tensor self, SymInt? n=None, int dim=-1, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_rfft_out + operator_name: fft_rfft + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_rfft.out(Tensor self, SymInt? n=None, int dim=-1, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_irfft + operator_name: fft_irfft + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_irfft(Tensor self, SymInt? n=None, int dim=-1, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_irfft_out + operator_name: fft_irfft + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_irfft.out(Tensor self, SymInt? n=None, int dim=-1, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_hfft + operator_name: fft_hfft + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_hfft(Tensor self, SymInt? n=None, int dim=-1, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_hfft_out + operator_name: fft_hfft + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_hfft.out(Tensor self, SymInt? n=None, int dim=-1, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ihfft + operator_name: fft_ihfft + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ihfft(Tensor self, SymInt? n=None, int dim=-1, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ihfft_out + operator_name: fft_ihfft + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ihfft.out(Tensor self, SymInt? n=None, int dim=-1, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: n + type: ::std::optional + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_fft2 + operator_name: fft_fft2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_fft2(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_fft2_out + operator_name: fft_fft2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_fft2.out(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ifft2 + operator_name: fft_ifft2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ifft2(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ifft2_out + operator_name: fft_ifft2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ifft2.out(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_rfft2 + operator_name: fft_rfft2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_rfft2(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_rfft2_out + operator_name: fft_rfft2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_rfft2.out(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_irfft2 + operator_name: fft_irfft2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_irfft2(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_irfft2_out + operator_name: fft_irfft2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_irfft2.out(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_hfft2 + operator_name: fft_hfft2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_hfft2(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_hfft2_out + operator_name: fft_hfft2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_hfft2.out(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ihfft2 + operator_name: fft_ihfft2 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ihfft2(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ihfft2_out + operator_name: fft_ihfft2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ihfft2.out(Tensor self, SymInt[1]? s=None, int[1] dim=[-2,-1], str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_fftn + operator_name: fft_fftn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_fftn(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_fftn_out + operator_name: fft_fftn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_fftn.out(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ifftn + operator_name: fft_ifftn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ifftn(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ifftn_out + operator_name: fft_ifftn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ifftn.out(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_rfftn + operator_name: fft_rfftn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_rfftn(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_rfftn_out + operator_name: fft_rfftn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_rfftn.out(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_irfftn + operator_name: fft_irfftn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_irfftn(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_irfftn_out + operator_name: fft_irfftn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_irfftn.out(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_hfftn + operator_name: fft_hfftn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_hfftn(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_hfftn_out + operator_name: fft_hfftn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_hfftn.out(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ihfftn + operator_name: fft_ihfftn + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ihfftn(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ihfftn_out + operator_name: fft_ihfftn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ihfftn.out(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::OptionalIntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: s + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + name: norm + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_fftfreq + operator_name: fft_fftfreq + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_fftfreq(int n, float d=1.0, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: d + type: double + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, double, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: d + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fft_fftfreq_out + operator_name: fft_fftfreq + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_fftfreq.out(int n, float d=1.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: d + type: double + schema_order_cpp_signature: at::Tensor & (int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: d + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fft_rfftfreq + operator_name: fft_rfftfreq + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_rfftfreq(int n, float d=1.0, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: d + type: double + - annotation: null + default: '{}' + dynamic_type: at::TensorOptions + is_nullable: false + kwarg_only: true + name: options + type: at::TensorOptions + schema_order_cpp_signature: at::Tensor (int64_t, double, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: d + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + kwarg_only: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: true + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fft_rfftfreq_out + operator_name: fft_rfftfreq + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_rfftfreq.out(int n, float d=1.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: d + type: double + schema_order_cpp_signature: at::Tensor & (int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: d + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: fft_fftshift + operator_name: fft_fftshift + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_fftshift(Tensor self, int[1]? dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: fft_ifftshift + operator_name: fft_ifftshift + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::fft_ifftshift(Tensor self, int[1]? dim=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: fft + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_cholesky_ex + operator_name: linalg_cholesky_ex + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_cholesky_ex(Tensor self, *, bool upper=False, bool check_errors=False) -> (Tensor L, Tensor info) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: L + name: L + type: at::Tensor + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_cholesky_ex_out + operator_name: linalg_cholesky_ex + overload_name: L + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_cholesky_ex.L(Tensor self, *, bool upper=False, bool check_errors=False, Tensor(a!) L, Tensor(b!) info) -> (Tensor(a!) L, Tensor(b!) info) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: L + is_nullable: false + name: L + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: L + is_nullable: false + name: L + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: L + name: L + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_cholesky + operator_name: linalg_cholesky + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_cholesky(Tensor self, *, bool upper=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_cholesky_out + operator_name: linalg_cholesky + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_cholesky.out(Tensor self, *, bool upper=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: upper + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_cross + operator_name: linalg_cross + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_cross(Tensor self, Tensor other, *, int dim=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_cross_out + operator_name: linalg_cross + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_cross.out(Tensor self, Tensor other, *, int dim=-1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_lu_factor + operator_name: linalg_lu_factor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_lu_factor(Tensor A, *, bool pivot=True) -> (Tensor LU, Tensor pivots) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_lu_factor_out + operator_name: linalg_lu_factor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_lu_factor.out(Tensor A, *, bool pivot=True, Tensor(a!) LU, Tensor(b!) pivots) -> (Tensor(a!) LU, Tensor(b!) pivots) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: LU + is_nullable: false + name: LU + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: LU + is_nullable: false + name: LU + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_lu_factor_ex + operator_name: linalg_lu_factor_ex + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_lu_factor_ex(Tensor A, *, bool pivot=True, bool check_errors=False) -> (Tensor LU, Tensor pivots, Tensor info) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_lu_factor_ex_out + operator_name: linalg_lu_factor_ex + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_lu_factor_ex.out(Tensor A, *, bool pivot=True, bool check_errors=False, Tensor(a!) LU, Tensor(b!) pivots, Tensor(c!) info) -> (Tensor(a!) LU, Tensor(b!) pivots, Tensor(c!) info) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: LU + is_nullable: false + name: LU + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: LU + is_nullable: false + name: LU + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_lu + operator_name: linalg_lu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_lu(Tensor A, *, bool pivot=True) -> (Tensor P, Tensor L, Tensor U) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: P + name: P + type: at::Tensor + - dynamic_type: at::Tensor + field_name: L + name: L + type: at::Tensor + - dynamic_type: at::Tensor + field_name: U + name: U + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_lu_out + operator_name: linalg_lu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_lu.out(Tensor A, *, bool pivot=True, Tensor(a!) P, Tensor(b!) L, Tensor(c!) U) -> (Tensor(a!) P, Tensor(b!) L, Tensor(c!) U) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: P + is_nullable: false + name: P + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: L + is_nullable: false + name: L + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: U + is_nullable: false + name: U + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: pivot + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: P + is_nullable: false + name: P + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: L + is_nullable: false + name: L + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: U + is_nullable: false + name: U + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: P + name: P + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: L + name: L + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: U + name: U + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_lu_solve + operator_name: linalg_lu_solve + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_lu_solve(Tensor LU, Tensor pivots, Tensor B, *, bool left=True, bool adjoint=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pivots + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: adjoint + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pivots + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: adjoint + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_lu_solve_out + operator_name: linalg_lu_solve + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_lu_solve.out(Tensor LU, Tensor pivots, Tensor B, *, bool left=True, bool adjoint=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pivots + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: adjoint + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LU + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pivots + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: adjoint + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _linalg_det + operator_name: _linalg_det + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_det(Tensor A) -> (Tensor result, Tensor LU, Tensor pivots) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: result + name: result + type: at::Tensor + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _linalg_det_out + operator_name: _linalg_det + overload_name: result + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_det.result(Tensor A, *, Tensor(a!) result, Tensor(b!) LU, Tensor(c!) pivots) -> (Tensor(a!) result, Tensor(b!) LU, Tensor(c!) pivots) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: result + is_nullable: false + name: result + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: LU + is_nullable: false + name: LU + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: result + is_nullable: false + name: result + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: LU + is_nullable: false + name: LU + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: result + name: result + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_det + operator_name: linalg_det + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_det(Tensor A) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_det_out + operator_name: linalg_det + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_det.out(Tensor A, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: det + operator_name: det + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::det(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_ldl_factor_ex + operator_name: linalg_ldl_factor_ex + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_ldl_factor_ex(Tensor self, *, bool hermitian=False, bool check_errors=False) -> (Tensor LD, Tensor pivots, Tensor info) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: LD + name: LD + type: at::Tensor + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_ldl_factor_ex_out + operator_name: linalg_ldl_factor_ex + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_ldl_factor_ex.out(Tensor self, *, bool hermitian=False, bool check_errors=False, Tensor(a!) LD, Tensor(b!) pivots, Tensor(c!) info) -> (Tensor(a!) LD, Tensor(b!) pivots, Tensor(c!) info) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: LD + is_nullable: false + name: LD + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: LD + is_nullable: false + name: LD + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: LD + name: LD + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_ldl_factor + operator_name: linalg_ldl_factor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_ldl_factor(Tensor self, *, bool hermitian=False) -> (Tensor LD, Tensor pivots) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: LD + name: LD + type: at::Tensor + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_ldl_factor_out + operator_name: linalg_ldl_factor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_ldl_factor.out(Tensor self, *, bool hermitian=False, Tensor(a!) LD, Tensor(b!) pivots) -> (Tensor(a!) LD, Tensor(b!) pivots) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: LD + is_nullable: false + name: LD + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: LD + is_nullable: false + name: LD + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: LD + name: LD + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_ldl_solve + operator_name: linalg_ldl_solve + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_ldl_solve(Tensor LD, Tensor pivots, Tensor B, *, bool hermitian=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LD + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pivots + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LD + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pivots + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_ldl_solve_out + operator_name: linalg_ldl_solve + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_ldl_solve.out(Tensor LD, Tensor pivots, Tensor B, *, bool hermitian=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LD + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pivots + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: LD + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pivots + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_lstsq + operator_name: linalg_lstsq + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_lstsq(Tensor self, Tensor b, float? rcond=None, *, str? driver=None) -> (Tensor solution, Tensor residuals, Tensor rank, Tensor singular_values) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: rcond + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: rcond + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: solution + name: solution + type: at::Tensor + - dynamic_type: at::Tensor + field_name: residuals + name: residuals + type: at::Tensor + - dynamic_type: at::Tensor + field_name: rank + name: rank + type: at::Tensor + - dynamic_type: at::Tensor + field_name: singular_values + name: singular_values + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_lstsq_out + operator_name: linalg_lstsq + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_lstsq.out(Tensor self, Tensor b, float? rcond=None, *, str? driver=None, Tensor(a!) solution, Tensor(b!) residuals, Tensor(c!) rank, Tensor(d!) singular_values) -> (Tensor(a!) solution, Tensor(b!) residuals, Tensor(c!) rank, Tensor(d!) singular_values) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: solution + is_nullable: false + name: solution + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: residuals + is_nullable: false + name: residuals + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: rank + is_nullable: false + name: rank + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + field_name: singular_values + is_nullable: false + name: singular_values + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: rcond + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: b + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + name: rcond + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: solution + is_nullable: false + name: solution + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: residuals + is_nullable: false + name: residuals + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: rank + is_nullable: false + name: rank + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + field_name: singular_values + is_nullable: false + name: singular_values + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: solution + name: solution + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: residuals + name: residuals + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: rank + name: rank + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: singular_values + name: singular_values + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_matmul + operator_name: linalg_matmul + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matmul(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matmul_out + operator_name: linalg_matmul + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matmul.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_vecdot + operator_name: linalg_vecdot + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_vecdot(Tensor x, Tensor y, *, int dim=-1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_vecdot_out + operator_name: linalg_vecdot + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_vecdot.out(Tensor x, Tensor y, *, int dim=-1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: y + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_exp + operator_name: linalg_matrix_exp + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_exp(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _linalg_slogdet + operator_name: _linalg_slogdet + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_slogdet(Tensor A) -> (Tensor sign, Tensor logabsdet, Tensor LU, Tensor pivots) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: sign + name: sign + type: at::Tensor + - dynamic_type: at::Tensor + field_name: logabsdet + name: logabsdet + type: at::Tensor + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _linalg_slogdet_out + operator_name: _linalg_slogdet + overload_name: sign + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_slogdet.sign(Tensor A, *, Tensor(a!) sign, Tensor(b!) logabsdet, Tensor(c!) LU, Tensor(d!) pivots) -> (Tensor(a!) sign, Tensor(b!) logabsdet, Tensor(c!) LU, Tensor(d!) pivots) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: sign + is_nullable: false + name: sign + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: logabsdet + is_nullable: false + name: logabsdet + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: LU + is_nullable: false + name: LU + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: sign + is_nullable: false + name: sign + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: logabsdet + is_nullable: false + name: logabsdet + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: LU + is_nullable: false + name: LU + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: sign + name: sign + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: logabsdet + name: logabsdet + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_slogdet + operator_name: linalg_slogdet + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_slogdet(Tensor A) -> (Tensor sign, Tensor logabsdet) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: sign + name: sign + type: at::Tensor + - dynamic_type: at::Tensor + field_name: logabsdet + name: logabsdet + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_slogdet_out + operator_name: linalg_slogdet + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_slogdet.out(Tensor A, *, Tensor(a!) sign, Tensor(b!) logabsdet) -> (Tensor(a!) sign, Tensor(b!) logabsdet) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: sign + is_nullable: false + name: sign + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: logabsdet + is_nullable: false + name: logabsdet + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: sign + is_nullable: false + name: sign + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: logabsdet + is_nullable: false + name: logabsdet + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: sign + name: sign + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: logabsdet + name: logabsdet + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: slogdet + operator_name: slogdet + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::slogdet(Tensor self) -> (Tensor sign, Tensor logabsdet) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: sign + name: sign + type: at::Tensor + - dynamic_type: at::Tensor + field_name: logabsdet + name: logabsdet + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: slogdet_out + operator_name: slogdet + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::slogdet.out(Tensor self, *, Tensor(a!) sign, Tensor(b!) logabsdet) -> (Tensor(a!) sign, Tensor(b!) logabsdet) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: sign + is_nullable: false + name: sign + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: logabsdet + is_nullable: false + name: logabsdet + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: sign + is_nullable: false + name: sign + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: logabsdet + is_nullable: false + name: logabsdet + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: sign + name: sign + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: logabsdet + name: logabsdet + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: logdet + operator_name: logdet + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::logdet(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_eig + operator_name: linalg_eig + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_eig(Tensor self) -> (Tensor eigenvalues, Tensor eigenvectors) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: eigenvalues + name: eigenvalues + type: at::Tensor + - dynamic_type: at::Tensor + field_name: eigenvectors + name: eigenvectors + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_eig_out + operator_name: linalg_eig + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_eig.out(Tensor self, *, Tensor(a!) eigenvalues, Tensor(b!) eigenvectors) -> (Tensor(a!) eigenvalues, Tensor(b!) eigenvectors) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: eigenvalues + is_nullable: false + name: eigenvalues + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: eigenvectors + is_nullable: false + name: eigenvectors + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: eigenvalues + is_nullable: false + name: eigenvalues + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: eigenvectors + is_nullable: false + name: eigenvectors + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: eigenvalues + name: eigenvalues + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: eigenvectors + name: eigenvectors + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _linalg_eigvals + operator_name: _linalg_eigvals + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_eigvals(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_eigvals + operator_name: linalg_eigvals + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_eigvals(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_eigvals_out + operator_name: linalg_eigvals + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_eigvals.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _linalg_eigh + operator_name: _linalg_eigh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_eigh(Tensor A, str UPLO="L", bool compute_v=True) -> (Tensor eigenvalues, Tensor eigenvectors) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_v + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, c10::string_view, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_v + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: eigenvalues + name: eigenvalues + type: at::Tensor + - dynamic_type: at::Tensor + field_name: eigenvectors + name: eigenvectors + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _linalg_eigh_out + operator_name: _linalg_eigh + overload_name: eigenvalues + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_eigh.eigenvalues(Tensor A, str UPLO="L", bool compute_v=True, *, Tensor(a!) eigenvalues, Tensor(b!) eigenvectors) -> (Tensor(a!) eigenvalues, Tensor(b!) eigenvectors) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: eigenvalues + is_nullable: false + name: eigenvalues + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: eigenvectors + is_nullable: false + name: eigenvectors + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_v + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, c10::string_view, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_v + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: eigenvalues + is_nullable: false + name: eigenvalues + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: eigenvectors + is_nullable: false + name: eigenvectors + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: eigenvalues + name: eigenvalues + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: eigenvectors + name: eigenvectors + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_eigh + operator_name: linalg_eigh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_eigh(Tensor self, str UPLO="L") -> (Tensor eigenvalues, Tensor eigenvectors) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: eigenvalues + name: eigenvalues + type: at::Tensor + - dynamic_type: at::Tensor + field_name: eigenvectors + name: eigenvectors + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_eigh_out + operator_name: linalg_eigh + overload_name: eigvals + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_eigh.eigvals(Tensor self, str UPLO="L", *, Tensor(a!) eigvals, Tensor(b!) eigvecs) -> (Tensor(a!) eigenvalues, Tensor(b!) eigenvectors) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: eigenvalues + is_nullable: false + name: eigvals + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: eigenvectors + is_nullable: false + name: eigvecs + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, c10::string_view, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: eigenvalues + is_nullable: false + name: eigvals + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: eigenvectors + is_nullable: false + name: eigvecs + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: eigenvalues + name: eigvals + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: eigenvectors + name: eigvecs + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_eigvalsh + operator_name: linalg_eigvalsh + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_eigvalsh(Tensor self, str UPLO="L") -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_eigvalsh_out + operator_name: linalg_eigvalsh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_eigvalsh.out(Tensor self, str UPLO="L", *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, c10::string_view, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"L"' + dynamic_type: c10::string_view + is_nullable: false + name: UPLO + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_householder_product + operator_name: linalg_householder_product + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_householder_product(Tensor input, Tensor tau) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tau + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tau + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_householder_product_out + operator_name: linalg_householder_product + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_householder_product.out(Tensor input, Tensor tau, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tau + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tau + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_inv_ex + operator_name: linalg_inv_ex + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_inv_ex(Tensor A, *, bool check_errors=False) -> (Tensor inverse, Tensor info) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: inverse + name: inverse + type: at::Tensor + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_inv_ex_out + operator_name: linalg_inv_ex + overload_name: inverse + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_inv_ex.inverse(Tensor A, *, bool check_errors=False, Tensor(a!) inverse, Tensor(b!) info) -> (Tensor(a!) inverse, Tensor(b!) info) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: inverse + is_nullable: false + name: inverse + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: inverse + is_nullable: false + name: inverse + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: inverse + name: inverse + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_inv + operator_name: linalg_inv + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_inv(Tensor A) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_inv_out + operator_name: linalg_inv + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_inv.out(Tensor A, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: inverse + operator_name: inverse + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::inverse(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: inverse_out + operator_name: inverse + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::inverse.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: inner + operator_name: inner + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::inner(Tensor self, Tensor other) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: inner_out + operator_name: inner + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::inner.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: outer + operator_name: outer + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::outer(Tensor self, Tensor vec2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: outer_out + operator_name: outer + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::outer.out(Tensor self, Tensor vec2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: ger + operator_name: ger + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::ger(Tensor self, Tensor vec2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + method_of: + - Type + - Tensor + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: ger_out + operator_name: ger + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ger.out(Tensor self, Tensor vec2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: vec2 + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_norm + operator_name: linalg_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_norm(Tensor self, Scalar? ord=None, int[1]? dim=None, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: ord + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, at::OptionalIntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: ord + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_norm + operator_name: linalg_norm + overload_name: ord_str + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_norm.ord_str(Tensor self, str ord, int[1]? dim=None, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: ord + type: c10::string_view + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, c10::string_view, at::OptionalIntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: ord + type: c10::string_view + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_norm_out + operator_name: linalg_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_norm.out(Tensor self, Scalar? ord=None, int[1]? dim=None, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: ord + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, at::OptionalIntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: ord + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_norm_out + operator_name: linalg_norm + overload_name: ord_str_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_norm.ord_str_out(Tensor self, str ord, int[1]? dim=None, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: ord + type: c10::string_view + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, c10::string_view, at::OptionalIntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: ord + type: c10::string_view + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_vector_norm + operator_name: linalg_vector_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_vector_norm(Tensor self, Scalar ord=2, int[1]? dim=None, bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, at::OptionalIntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_vector_norm_out + operator_name: linalg_vector_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_vector_norm.out(Tensor self, Scalar ord=2, int[1]? dim=None, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::OptionalIntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_matrix_norm + operator_name: linalg_matrix_norm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_norm(Tensor self, Scalar ord, int[] dim=[-2,-1], bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &, at::IntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_norm_out + operator_name: linalg_matrix_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_norm.out(Tensor self, Scalar ord, int[] dim=[-2,-1], bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::IntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_norm + operator_name: linalg_matrix_norm + overload_name: str_ord + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_norm.str_ord(Tensor self, str ord='fro', int[] dim=[-2,-1], bool keepdim=False, *, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"fro"' + dynamic_type: c10::string_view + is_nullable: false + name: ord + type: c10::string_view + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, c10::string_view, at::IntArrayRef, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"fro"' + dynamic_type: c10::string_view + is_nullable: false + name: ord + type: c10::string_view + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_norm_out + operator_name: linalg_matrix_norm + overload_name: str_ord_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_norm.str_ord_out(Tensor self, str ord='fro', int[] dim=[-2,-1], bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"fro"' + dynamic_type: c10::string_view + is_nullable: false + name: ord + type: c10::string_view + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, c10::string_view, at::IntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '"fro"' + dynamic_type: c10::string_view + is_nullable: false + name: ord + type: c10::string_view + - annotation: null + default: '{-2,-1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _linalg_svd + operator_name: _linalg_svd + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_svd(Tensor A, bool full_matrices=False, bool compute_uv=True, *, str? driver=None) -> (Tensor U, Tensor S, Tensor Vh) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: full_matrices + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_uv + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: full_matrices + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_uv + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: U + name: U + type: at::Tensor + - dynamic_type: at::Tensor + field_name: S + name: S + type: at::Tensor + - dynamic_type: at::Tensor + field_name: Vh + name: Vh + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _linalg_svd_out + operator_name: _linalg_svd + overload_name: U + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_svd.U(Tensor A, bool full_matrices=False, bool compute_uv=True, *, str? driver=None, Tensor(a!) U, Tensor(b!) S, Tensor(c!) Vh) -> (Tensor(a!) U, Tensor(b!) S, Tensor(c!) Vh) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: U + is_nullable: false + name: U + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: S + is_nullable: false + name: S + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: Vh + is_nullable: false + name: Vh + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: full_matrices + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_uv + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, ::std::optional, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: full_matrices + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: compute_uv + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: U + is_nullable: false + name: U + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: S + is_nullable: false + name: S + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: Vh + is_nullable: false + name: Vh + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: U + name: U + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: S + name: S + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: Vh + name: Vh + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_svd + operator_name: linalg_svd + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_svd(Tensor A, bool full_matrices=True, *, str? driver=None) -> (Tensor U, Tensor S, Tensor Vh) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: full_matrices + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: full_matrices + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: U + name: U + type: at::Tensor + - dynamic_type: at::Tensor + field_name: S + name: S + type: at::Tensor + - dynamic_type: at::Tensor + field_name: Vh + name: Vh + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_svd_out + operator_name: linalg_svd + overload_name: U + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_svd.U(Tensor A, bool full_matrices=True, *, str? driver=None, Tensor(a!) U, Tensor(b!) S, Tensor(c!) Vh) -> (Tensor(a!) U, Tensor(b!) S, Tensor(c!) Vh) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: U + is_nullable: false + name: U + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: S + is_nullable: false + name: S + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: Vh + is_nullable: false + name: Vh + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: full_matrices + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, ::std::optional, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: full_matrices + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: U + is_nullable: false + name: U + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: S + is_nullable: false + name: S + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: Vh + is_nullable: false + name: Vh + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: U + name: U + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: S + name: S + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: Vh + name: Vh + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_svdvals + operator_name: linalg_svdvals + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_svdvals(Tensor A, *, str? driver=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_svdvals_out + operator_name: linalg_svdvals + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_svdvals.out(Tensor A, *, str? driver=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: driver + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_cond + operator_name: linalg_cond + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_cond(Tensor self, Scalar? p=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_cond_out + operator_name: linalg_cond + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_cond.out(Tensor self, Scalar? p=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_cond + operator_name: linalg_cond + overload_name: p_str + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_cond.p_str(Tensor self, str p) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: p + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: p + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_cond_out + operator_name: linalg_cond + overload_name: p_str_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_cond.p_str_out(Tensor self, str p, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: p + type: c10::string_view + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, c10::string_view, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: p + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_pinv + operator_name: linalg_pinv + overload_name: atol_rtol_tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_pinv.atol_rtol_tensor(Tensor self, *, Tensor? atol=None, Tensor? rtol=None, bool hermitian=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: atol + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: rtol + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: atol + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: rtol + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_pinv_out + operator_name: linalg_pinv + overload_name: atol_rtol_tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_pinv.atol_rtol_tensor_out(Tensor self, *, Tensor? atol=None, Tensor? rtol=None, bool hermitian=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: atol + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: rtol + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, const ::std::optional &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: atol + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: rtol + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_pinv + operator_name: linalg_pinv + overload_name: atol_rtol_float + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_pinv.atol_rtol_float(Tensor self, *, float? atol=None, float? rtol=None, bool hermitian=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: atol + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: rtol + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: atol + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: rtol + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_pinv_out + operator_name: linalg_pinv + overload_name: atol_rtol_float_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_pinv.atol_rtol_float_out(Tensor self, *, float? atol=None, float? rtol=None, bool hermitian=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: atol + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: rtol + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, ::std::optional, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: atol + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: rtol + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_pinv + operator_name: linalg_pinv + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_pinv(Tensor self, float rcond, bool hermitian=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: rcond + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: rcond + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_pinv + operator_name: linalg_pinv + overload_name: rcond_tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_pinv.rcond_tensor(Tensor self, Tensor rcond, bool hermitian=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rcond + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rcond + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_pinv_out + operator_name: linalg_pinv + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_pinv.out(Tensor self, float rcond, bool hermitian=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: rcond + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: rcond + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_pinv_out + operator_name: linalg_pinv + overload_name: out_rcond_tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_pinv.out_rcond_tensor(Tensor self, Tensor rcond, bool hermitian=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rcond + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rcond + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _linalg_solve_ex + operator_name: _linalg_solve_ex + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_solve_ex(Tensor A, Tensor B, *, bool left=True, bool check_errors=False) -> (Tensor result, Tensor LU, Tensor pivots, Tensor info) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: result + name: result + type: at::Tensor + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _linalg_solve_ex_out + operator_name: _linalg_solve_ex + overload_name: result + manual_kernel_registration: false + category_override: '' + schema_string: aten::_linalg_solve_ex.result(Tensor A, Tensor B, *, bool left=True, bool check_errors=False, Tensor(a!) result, Tensor(b!) LU, Tensor(c!) pivots, Tensor(d!) info) -> (Tensor(a!) result, Tensor(b!) LU, Tensor(c!) pivots, Tensor(d!) info) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: result + is_nullable: false + name: result + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: LU + is_nullable: false + name: LU + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: result + is_nullable: false + name: result + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: LU + is_nullable: false + name: LU + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + field_name: pivots + is_nullable: false + name: pivots + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: result + name: result + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: LU + name: LU + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: pivots + name: pivots + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_solve_ex + operator_name: linalg_solve_ex + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_solve_ex(Tensor A, Tensor B, *, bool left=True, bool check_errors=False) -> (Tensor result, Tensor info) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: result + name: result + type: at::Tensor + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_solve_ex_out + operator_name: linalg_solve_ex + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_solve_ex.out(Tensor A, Tensor B, *, bool left=True, bool check_errors=False, Tensor(a!) result, Tensor(b!) info) -> (Tensor(a!) result, Tensor(b!) info) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: result + is_nullable: false + name: result + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: check_errors + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: result + is_nullable: false + name: result + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: info + is_nullable: false + name: info + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: result + name: result + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: info + name: info + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_solve + operator_name: linalg_solve + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_solve(Tensor A, Tensor B, *, bool left=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _spsolve + operator_name: _spsolve + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_spsolve(Tensor A, Tensor B, *, bool left=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + method_of: + - Type + - namespace + mode: native + python_module: sparse + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_solve_out + operator_name: linalg_solve + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_solve.out(Tensor A, Tensor B, *, bool left=True, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: B + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: left + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_tensorinv + operator_name: linalg_tensorinv + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_tensorinv(Tensor self, int ind=2) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: int64_t + is_nullable: false + name: ind + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: int64_t + is_nullable: false + name: ind + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_tensorinv_out + operator_name: linalg_tensorinv + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_tensorinv.out(Tensor self, int ind=2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: int64_t + is_nullable: false + name: ind + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: int64_t + is_nullable: false + name: ind + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_tensorsolve + operator_name: linalg_tensorsolve + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_tensorsolve(Tensor self, Tensor other, int[]? dims=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dims + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, at::OptionalIntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dims + type: at::OptionalIntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_tensorsolve_out + operator_name: linalg_tensorsolve + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_tensorsolve.out(Tensor self, Tensor other, int[]? dims=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dims + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::OptionalIntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dims + type: at::OptionalIntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_qr + operator_name: linalg_qr + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_qr(Tensor A, str mode='reduced') -> (Tensor Q, Tensor R) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: '"reduced"' + dynamic_type: c10::string_view + is_nullable: false + name: mode + type: c10::string_view + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: '"reduced"' + dynamic_type: c10::string_view + is_nullable: false + name: mode + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: Q + name: Q + type: at::Tensor + - dynamic_type: at::Tensor + field_name: R + name: R + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_qr_out + operator_name: linalg_qr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_qr.out(Tensor A, str mode='reduced', *, Tensor(a!) Q, Tensor(b!) R) -> (Tensor(a!) Q, Tensor(b!) R) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: Q + is_nullable: false + name: Q + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: R + is_nullable: false + name: R + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: '"reduced"' + dynamic_type: c10::string_view + is_nullable: false + name: mode + type: c10::string_view + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, c10::string_view, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + default: '"reduced"' + dynamic_type: c10::string_view + is_nullable: false + name: mode + type: c10::string_view + - allocate: true + annotation: a! + dynamic_type: at::Tensor + field_name: Q + is_nullable: false + name: Q + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + field_name: R + is_nullable: false + name: R + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + field_name: Q + name: Q + type: at::Tensor & + - dynamic_type: at::Tensor + field_name: R + name: R + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_matrix_power + operator_name: linalg_matrix_power + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_power(Tensor self, int n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_power_out + operator_name: linalg_matrix_power + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_power.out(Tensor self, int n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: n + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_rank + operator_name: linalg_matrix_rank + overload_name: atol_rtol_tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_rank.atol_rtol_tensor(Tensor input, *, Tensor? atol=None, Tensor? rtol=None, bool hermitian=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: atol + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: rtol + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const ::std::optional &, const ::std::optional &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: atol + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: rtol + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_rank_out + operator_name: linalg_matrix_rank + overload_name: atol_rtol_tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_rank.atol_rtol_tensor_out(Tensor input, *, Tensor? atol=None, Tensor? rtol=None, bool hermitian=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: atol + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: rtol + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, const ::std::optional &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: atol + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: rtol + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_rank + operator_name: linalg_matrix_rank + overload_name: atol_rtol_float + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_rank.atol_rtol_float(Tensor self, *, float? atol=None, float? rtol=None, bool hermitian=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: atol + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: rtol + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: atol + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: rtol + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_rank_out + operator_name: linalg_matrix_rank + overload_name: atol_rtol_float_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_rank.atol_rtol_float_out(Tensor self, *, float? atol=None, float? rtol=None, bool hermitian=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: atol + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: rtol + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, ::std::optional, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: atol + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: rtol + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: hermitian + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_rank + operator_name: linalg_matrix_rank + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_rank(Tensor self, float tol, bool hermitian=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: tol + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: tol + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_rank_out + operator_name: linalg_matrix_rank + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_rank.out(Tensor self, float tol, bool hermitian=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: tol + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: tol + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_rank + operator_name: linalg_matrix_rank + overload_name: tol_tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_rank.tol_tensor(Tensor input, Tensor tol, bool hermitian=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tol + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tol + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_matrix_rank_out + operator_name: linalg_matrix_rank + overload_name: out_tol_tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_rank.out_tol_tensor(Tensor input, Tensor tol, bool hermitian=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tol + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tol + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: hermitian + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_multi_dot + operator_name: linalg_multi_dot + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_multi_dot(Tensor[] tensors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: linalg_multi_dot_out + operator_name: linalg_multi_dot + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_multi_dot.out(Tensor[] tensors, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: linalg + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: nested_to_padded_tensor + operator_name: nested_to_padded_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::nested_to_padded_tensor(Tensor self, float padding, int[]? output_size=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: padding + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, at::OptionalIntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: padding + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nested + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _test_serialization_subcmul + operator_name: _test_serialization_subcmul + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_serialization_subcmul(Tensor self, Tensor other, Scalar alpha=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _test_parallel_materialize + operator_name: _test_parallel_materialize + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_parallel_materialize(Tensor self, int num_parallel, bool skip_first=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_parallel + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: skip_first + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_parallel + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: skip_first + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_optional_intlist + operator_name: _test_optional_intlist + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_optional_intlist(Tensor values, int[]? addends) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: addends + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: addends + type: at::OptionalIntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_optional_filled_intlist + operator_name: _test_optional_filled_intlist + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_optional_filled_intlist(Tensor values, int[2]? addends) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: addends + size: 2 + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::OptionalIntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: addends + size: 2 + type: at::OptionalIntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_optional_floatlist + operator_name: _test_optional_floatlist + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_optional_floatlist(Tensor values, float[]? addends) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: addends + type: ::std::optional> + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional>) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: addends + type: ::std::optional> + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_string_default + operator_name: _test_string_default + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_string_default(Tensor dummy, str a="\"'\\", str b='"\'\\') -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: '"\"''\\"' + dynamic_type: c10::string_view + is_nullable: false + name: a + type: c10::string_view + - annotation: null + default: '"\"''\\"' + dynamic_type: c10::string_view + is_nullable: false + name: b + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, c10::string_view, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: '"\"''\\"' + dynamic_type: c10::string_view + is_nullable: false + name: a + type: c10::string_view + - annotation: null + default: '"\"''\\"' + dynamic_type: c10::string_view + is_nullable: false + name: b + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _test_ambiguous_defaults + operator_name: _test_ambiguous_defaults + overload_name: a + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_ambiguous_defaults.a(Tensor dummy, int a=1, int b=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: a + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: b + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: a + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: b + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _test_ambiguous_defaults + operator_name: _test_ambiguous_defaults + overload_name: b + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_ambiguous_defaults.b(Tensor dummy, int a=2, str b="2") -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: int64_t + is_nullable: false + name: a + type: int64_t + - annotation: null + default: '"2"' + dynamic_type: c10::string_view + is_nullable: false + name: b + type: c10::string_view + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: int64_t + is_nullable: false + name: a + type: int64_t + - annotation: null + default: '"2"' + dynamic_type: c10::string_view + is_nullable: false + name: b + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _test_warn_in_autograd + operator_name: _test_warn_in_autograd + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_warn_in_autograd(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_autograd_multiple_dispatch + operator_name: _test_autograd_multiple_dispatch + overload_name: fullcoverage + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_autograd_multiple_dispatch.fullcoverage(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_autograd_multiple_dispatch + operator_name: _test_autograd_multiple_dispatch + overload_name: ntonly + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_autograd_multiple_dispatch.ntonly(Tensor self, bool b) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: b + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: b + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _test_autograd_multiple_dispatch_view + operator_name: _test_autograd_multiple_dispatch_view + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_autograd_multiple_dispatch_view(Tensor(a) self) -> Tensor(a) + arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: a + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_autograd_multiple_dispatch_view_copy + operator_name: _test_autograd_multiple_dispatch_view_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_autograd_multiple_dispatch_view_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: segment_reduce + operator_name: segment_reduce + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::segment_reduce(Tensor data, str reduce, *, Tensor? lengths=None, Tensor? indices=None, Tensor? offsets=None, int axis=0, bool unsafe=False, Scalar? initial=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: lengths + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: indices + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: offsets + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: unsafe + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: initial + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, c10::string_view, const ::std::optional &, const ::std::optional &, const ::std::optional &, int64_t, bool, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: lengths + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: indices + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: offsets + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: unsafe + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: initial + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _segment_reduce_backward + operator_name: _segment_reduce_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_segment_reduce_backward(Tensor grad, Tensor output, Tensor data, str reduce, *, Tensor? lengths=None, Tensor? offsets=None, int axis=0, Scalar? initial=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: lengths + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: offsets + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: initial + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, c10::string_view, const ::std::optional &, const ::std::optional &, int64_t, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: lengths + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: offsets + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: initial + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: pad_sequence + operator_name: pad_sequence + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::pad_sequence(Tensor[] sequences, bool batch_first=False, float padding_value=0.0, str padding_side="right") -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: sequences + type: at::TensorList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: padding_value + type: double + - annotation: null + default: '"right"' + dynamic_type: c10::string_view + is_nullable: false + name: padding_side + type: c10::string_view + schema_order_cpp_signature: at::Tensor (at::TensorList, bool, double, c10::string_view) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: sequences + type: at::TensorList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: padding_value + type: double + - annotation: null + default: '"right"' + dynamic_type: c10::string_view + is_nullable: false + name: padding_side + type: c10::string_view + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: flatten_dense_tensors + operator_name: flatten_dense_tensors + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::flatten_dense_tensors(Tensor[] tensors) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: unflatten_dense_tensors + operator_name: unflatten_dense_tensors + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unflatten_dense_tensors(Tensor flat, Tensor[] tensors) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: flat + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: flat + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _nested_tensor_from_tensor_list + operator_name: _nested_tensor_from_tensor_list + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_from_tensor_list(Tensor[] list, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: list + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: pin_memory + type: ::std::optional + schema_order_cpp_signature: at::Tensor (at::TensorList, ::std::optional, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: list + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: pin_memory + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fw_primal_copy + operator_name: _fw_primal_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fw_primal_copy(Tensor self, int level) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _make_dual_copy + operator_name: _make_dual_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_make_dual_copy(Tensor primal, Tensor tangent, int level) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: primal + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tangent + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: primal + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tangent + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: view_as_real_copy + operator_name: view_as_real_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_as_real_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: view_as_complex_copy + operator_name: view_as_complex_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_as_complex_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _conj_copy + operator_name: _conj_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_conj_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _neg_view_copy + operator_name: _neg_view_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_neg_view_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: as_strided_copy + operator_name: as_strided_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::as_strided_copy(Tensor self, SymInt[] size, SymInt[] stride, SymInt? storage_offset=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_broadcast_to_copy + operator_name: _sparse_broadcast_to_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_broadcast_to_copy(Tensor self, int[] size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: diagonal_copy + operator_name: diagonal_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::diagonal_copy(Tensor self, int offset=0, int dim1=0, int dim2=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: expand_copy + operator_name: expand_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::expand_copy(Tensor self, SymInt[] size, *, bool implicit=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: implicit + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: implicit + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: permute_copy + operator_name: permute_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::permute_copy(Tensor self, int[] dims) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _reshape_alias_copy + operator_name: _reshape_alias_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_reshape_alias_copy(Tensor self, SymInt[] size, SymInt[] stride) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: select_copy + operator_name: select_copy + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::select_copy.int(Tensor self, int dim, SymInt index) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: detach_copy + operator_name: detach_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::detach_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: slice_copy + operator_name: slice_copy + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::slice_copy.Tensor(Tensor self, int dim=0, SymInt? start=None, SymInt? end=None, SymInt step=1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional, ::std::optional, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: split_copy + operator_name: split_copy + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::split_copy.Tensor(Tensor self, SymInt split_size, int dim=0) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: split_size + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: split_size + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: split_with_sizes_copy + operator_name: split_with_sizes_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::split_with_sizes_copy(Tensor self, SymInt[] split_sizes, int dim=0) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_sizes + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, at::IntArrayRef, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_sizes + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze_copy + operator_name: squeeze_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze_copy + operator_name: squeeze_copy + overload_name: dim + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze_copy.dim(Tensor self, int dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze_copy + operator_name: squeeze_copy + overload_name: dims + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze_copy.dims(Tensor self, int[] dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: t_copy + operator_name: t_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::t_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: transpose_copy + operator_name: transpose_copy + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::transpose_copy.int(Tensor self, int dim0, int dim1) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unsqueeze_copy + operator_name: unsqueeze_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unsqueeze_copy(Tensor self, int dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _indices_copy + operator_name: _indices_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_indices_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _values_copy + operator_name: _values_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_values_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: indices_copy + operator_name: indices_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::indices_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: values_copy + operator_name: values_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::values_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: crow_indices_copy + operator_name: crow_indices_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::crow_indices_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: col_indices_copy + operator_name: col_indices_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::col_indices_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: ccol_indices_copy + operator_name: ccol_indices_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::ccol_indices_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: row_indices_copy + operator_name: row_indices_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::row_indices_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unbind_copy + operator_name: unbind_copy + overload_name: int + manual_kernel_registration: false + category_override: '' + schema_string: aten::unbind_copy.int(Tensor self, int dim=0) -> Tensor[] + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::vector (const at::Tensor &, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + name: result + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unbind_copy_out + operator_name: unbind_copy + overload_name: int_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::unbind_copy.int_out(Tensor self, int dim=0, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: void (const at::Tensor &, int64_t, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: split_copy_out + operator_name: split_copy + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::split_copy.Tensor_out(Tensor self, SymInt split_size, int dim=0, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: split_size + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: void (const at::Tensor &, int64_t, int64_t, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: split_size + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: split_with_sizes_copy_out + operator_name: split_with_sizes_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::split_with_sizes_copy.out(Tensor self, SymInt[] split_sizes, int dim=0, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_sizes + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: void (const at::Tensor &, at::IntArrayRef, int64_t, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_sizes + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: view_copy + operator_name: view_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_copy(Tensor self, SymInt[] size) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: view_copy + operator_name: view_copy + overload_name: dtype + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_copy.dtype(Tensor self, ScalarType dtype) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::ScalarType) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: unfold_copy + operator_name: unfold_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::unfold_copy(Tensor self, int dimension, int size, int step) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: alias_copy + operator_name: alias_copy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::alias_copy(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_padded_tensor + operator_name: to_padded_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_padded_tensor(Tensor self, float padding, SymInt[]? output_size=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: padding + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, at::OptionalIntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: padding + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + method_of: + - Type + - Tensor + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _jagged_to_padded_dense_forward + operator_name: _jagged_to_padded_dense_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_jagged_to_padded_dense_forward(Tensor values, Tensor[] offsets, SymInt[] max_lengths, float padding_value=0.0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: offsets + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: max_lengths + type: at::IntArrayRef + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: padding_value + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::TensorList, at::IntArrayRef, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: offsets + type: at::TensorList + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: max_lengths + type: at::IntArrayRef + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: padding_value + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _padded_dense_to_jagged_forward + operator_name: _padded_dense_to_jagged_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_padded_dense_to_jagged_forward(Tensor dense, Tensor[] offsets, SymInt? total_L=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: offsets + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: total_L + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::TensorList, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dense + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: offsets + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: total_L + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_from_padded_tensor + operator_name: _nested_from_padded_tensor + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_from_padded_tensor(Tensor padded, Tensor offsets, Tensor dummy, int ragged_idx=1, Tensor? min_seqlen=None, Tensor? max_seqlen=None, SymInt? sum_S=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: padded + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: ragged_idx + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min_seqlen + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max_seqlen + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: sum_S + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, const ::std::optional &, const ::std::optional &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: padded + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: ragged_idx + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min_seqlen + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max_seqlen + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: sum_S + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_softmax_with_shape + operator_name: _nested_tensor_softmax_with_shape + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_softmax_with_shape(Tensor self, Tensor query) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _safe_softmax + operator_name: _safe_softmax + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_safe_softmax(Tensor self, int dim, ScalarType? dtype=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _transformer_encoder_layer_fwd + operator_name: _transformer_encoder_layer_fwd + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_transformer_encoder_layer_fwd(Tensor src, int embed_dim, int num_heads, Tensor qkv_weight, Tensor qkv_bias, Tensor proj_weight, Tensor proj_bias, bool use_gelu, bool norm_first, float eps, Tensor norm_weight_1, Tensor norm_bias_1, Tensor norm_weight_2, Tensor norm_bias_2, Tensor ffn_weight_1, Tensor ffn_bias_1, Tensor ffn_weight_2, Tensor ffn_bias_2, Tensor? mask=None, int? mask_type=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_heads + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: use_gelu + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: norm_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_weight_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_bias_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_weight_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_bias_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_weight_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_bias_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_weight_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_bias_2 + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, int64_t, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, bool, double, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_heads + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: use_gelu + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: norm_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_weight_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_bias_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_weight_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_bias_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_weight_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_bias_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_weight_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_bias_2 + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _native_multi_head_attention + operator_name: _native_multi_head_attention + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_native_multi_head_attention(Tensor query, Tensor key, Tensor value, int embed_dim, int num_head, Tensor qkv_weight, Tensor qkv_bias, Tensor proj_weight, Tensor proj_bias, Tensor? mask=None, bool need_weights=True, bool average_attn_weights=True, int? mask_type=None) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_head + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: need_weights + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: average_attn_weights + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_head + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: need_weights + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: average_attn_weights + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: scaled_dot_product_attention + operator_name: scaled_dot_product_attention + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::scaled_dot_product_attention(Tensor query, Tensor key, Tensor value, Tensor? attn_mask=None, float dropout_p=0.0, bool is_causal=False, *, float? scale=None, bool enable_gqa=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: enable_gqa + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, double, bool, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: enable_gqa + type: bool + method_of: + - Type + - namespace + mode: native + python_module: nn + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _fused_sdp_choice + operator_name: _fused_sdp_choice + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_sdp_choice(Tensor query, Tensor key, Tensor value, Tensor? attn_mask=None, float dropout_p=0.0, bool is_causal=False, *, float? scale=None, bool enable_gqa=False) -> int + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: enable_gqa + type: bool + schema_order_cpp_signature: int64_t (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, double, bool, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: enable_gqa + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: int64_t + name: result + type: int64_t + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_attention_math + operator_name: _scaled_dot_product_attention_math + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_attention_math(Tensor query, Tensor key, Tensor value, Tensor? attn_mask=None, float dropout_p=0.0, bool is_causal=False, Tensor? dropout_mask=None, *, float? scale=None, bool enable_gqa=False) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: dropout_mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: enable_gqa + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, double, bool, const ::std::optional &, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: dropout_mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: enable_gqa + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _scaled_dot_product_attention_math_for_mps + operator_name: _scaled_dot_product_attention_math_for_mps + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_attention_math_for_mps(Tensor query, Tensor key, Tensor value, Tensor? attn_mask=None, float dropout_p=0.0, bool is_causal=False, Tensor? dropout_mask=None, *, float? scale=None) -> (Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: dropout_mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, double, bool, const ::std::optional &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: dropout_mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_flash_attention + operator_name: _scaled_dot_product_flash_attention + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_flash_attention(Tensor query, Tensor key, Tensor value, float dropout_p=0.0, bool is_causal=False, bool return_debug_mask=False, *, float? scale=None) -> (Tensor output, Tensor logsumexp, Tensor cum_seq_q, Tensor cum_seq_k, SymInt max_q, SymInt max_k, Tensor rng_state, Tensor unused, Tensor debug_attn_mask) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_debug_mask + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, double, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_debug_mask + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: logsumexp + name: logsumexp + type: at::Tensor + - dynamic_type: at::Tensor + field_name: cum_seq_q + name: cum_seq_q + type: at::Tensor + - dynamic_type: at::Tensor + field_name: cum_seq_k + name: cum_seq_k + type: at::Tensor + - dynamic_type: int64_t + field_name: max_q + name: max_q + type: c10::SymInt + - dynamic_type: int64_t + field_name: max_k + name: max_k + type: c10::SymInt + - dynamic_type: at::Tensor + field_name: rng_state + name: rng_state + type: at::Tensor + - dynamic_type: at::Tensor + field_name: unused + name: unused + type: at::Tensor + - dynamic_type: at::Tensor + field_name: debug_attn_mask + name: debug_attn_mask + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_flash_attention_for_cpu + operator_name: _scaled_dot_product_flash_attention_for_cpu + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_flash_attention_for_cpu(Tensor query, Tensor key, Tensor value, float dropout_p=0.0, bool is_causal=False, *, Tensor? attn_mask=None, float? scale=None) -> (Tensor output, Tensor logsumexp) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, double, bool, const ::std::optional &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: logsumexp + name: logsumexp + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_fused_attention_overrideable + operator_name: _scaled_dot_product_fused_attention_overrideable + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_fused_attention_overrideable(Tensor query, Tensor key, Tensor value, Tensor? attn_bias=None, float dropout_p=0.0, bool is_causal=False, bool return_debug_mask=False, *, float? scale=None) -> (Tensor output, Tensor logsumexp, Tensor cum_seq_q, Tensor cum_seq_k, SymInt max_q, SymInt max_k, Tensor philox_seed, Tensor philox_offset, Tensor debug_attn_mask) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: attn_bias + type: const ::std::optional & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_debug_mask + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, double, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: attn_bias + type: const ::std::optional & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_debug_mask + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: logsumexp + name: logsumexp + type: at::Tensor + - dynamic_type: at::Tensor + field_name: cum_seq_q + name: cum_seq_q + type: at::Tensor + - dynamic_type: at::Tensor + field_name: cum_seq_k + name: cum_seq_k + type: at::Tensor + - dynamic_type: int64_t + field_name: max_q + name: max_q + type: c10::SymInt + - dynamic_type: int64_t + field_name: max_k + name: max_k + type: c10::SymInt + - dynamic_type: at::Tensor + field_name: philox_seed + name: philox_seed + type: at::Tensor + - dynamic_type: at::Tensor + field_name: philox_offset + name: philox_offset + type: at::Tensor + - dynamic_type: at::Tensor + field_name: debug_attn_mask + name: debug_attn_mask + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_flash_attention_backward + operator_name: _scaled_dot_product_flash_attention_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_flash_attention_backward(Tensor grad_out, Tensor query, Tensor key, Tensor value, Tensor out, Tensor logsumexp, Tensor cum_seq_q, Tensor cum_seq_k, SymInt max_q, SymInt max_k, float dropout_p, bool is_causal, Tensor philox_seed, Tensor philox_offset, *, float? scale=None) -> (Tensor grad_query, Tensor grad_key, Tensor grad_value) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_k + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, double, bool, const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_k + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: grad_query + name: grad_query + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_key + name: grad_key + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_value + name: grad_value + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_flash_attention_for_cpu_backward + operator_name: _scaled_dot_product_flash_attention_for_cpu_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_flash_attention_for_cpu_backward(Tensor grad_out, Tensor query, Tensor key, Tensor value, Tensor out, Tensor logsumexp, float dropout_p, bool is_causal, *, Tensor? attn_mask=None, float? scale=None) -> (Tensor grad_query, Tensor grad_key, Tensor grad_value) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, double, bool, const ::std::optional &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: attn_mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: grad_query + name: grad_query + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_key + name: grad_key + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_value + name: grad_value + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_fused_attention_overrideable_backward + operator_name: _scaled_dot_product_fused_attention_overrideable_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_fused_attention_overrideable_backward(Tensor grad_out, Tensor query, Tensor key, Tensor value, Tensor attn_bias, bool[4] grad_input_mask, Tensor out, Tensor logsumexp, Tensor cum_seq_q, Tensor cum_seq_k, SymInt max_q, SymInt max_k, float dropout_p, bool is_causal, Tensor philox_seed, Tensor philox_offset, *, float? scale=None) -> (Tensor grad_query, Tensor grad_key, Tensor grad_value, Tensor grad_attn_bias) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: attn_bias + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: grad_input_mask + type: ::std::array + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_k + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::array, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, double, bool, const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: attn_bias + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: grad_input_mask + type: ::std::array + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_k + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: grad_query + name: grad_query + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_key + name: grad_key + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_value + name: grad_value + type: at::Tensor + - dynamic_type: at::Tensor + field_name: grad_attn_bias + name: grad_attn_bias + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_efficient_attention + operator_name: _scaled_dot_product_efficient_attention + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_efficient_attention(Tensor query, Tensor key, Tensor value, Tensor? attn_bias, bool compute_log_sumexp, float dropout_p=0.0, bool is_causal=False, *, float? scale=None) -> (Tensor output, Tensor log_sumexp, Tensor philox_seed, Tensor philox_offset) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: attn_bias + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: compute_log_sumexp + type: bool + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, bool, double, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: attn_bias + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: compute_log_sumexp + type: bool + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: log_sumexp + name: log_sumexp + type: at::Tensor + - dynamic_type: at::Tensor + field_name: philox_seed + name: philox_seed + type: at::Tensor + - dynamic_type: at::Tensor + field_name: philox_offset + name: philox_offset + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_efficient_attention_backward + operator_name: _scaled_dot_product_efficient_attention_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_efficient_attention_backward(Tensor grad_out_, Tensor query, Tensor key, Tensor value, Tensor attn_bias, Tensor out, Tensor logsumexp, Tensor philox_seed, Tensor philox_offset, float dropout_p, bool[4] grad_input_mask, bool is_causal=False, *, float? scale=None) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: attn_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: grad_input_mask + type: ::std::array + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, double, ::std::array, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: attn_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: grad_input_mask + type: ::std::array + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_cudnn_attention + operator_name: _scaled_dot_product_cudnn_attention + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_cudnn_attention(Tensor query, Tensor key, Tensor value, Tensor? attn_bias, bool compute_log_sumexp, float dropout_p=0.0, bool is_causal=False, bool return_debug_mask=False, *, float? scale=None) -> (Tensor output, Tensor logsumexp, Tensor cum_seq_q, Tensor cum_seq_k, SymInt max_q, SymInt max_k, Tensor philox_seed, Tensor philox_offset, Tensor debug_attn_mask) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: attn_bias + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: compute_log_sumexp + type: bool + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_debug_mask + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, bool, double, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: attn_bias + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: compute_log_sumexp + type: bool + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_debug_mask + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: logsumexp + name: logsumexp + type: at::Tensor + - dynamic_type: at::Tensor + field_name: cum_seq_q + name: cum_seq_q + type: at::Tensor + - dynamic_type: at::Tensor + field_name: cum_seq_k + name: cum_seq_k + type: at::Tensor + - dynamic_type: int64_t + field_name: max_q + name: max_q + type: c10::SymInt + - dynamic_type: int64_t + field_name: max_k + name: max_k + type: c10::SymInt + - dynamic_type: at::Tensor + field_name: philox_seed + name: philox_seed + type: at::Tensor + - dynamic_type: at::Tensor + field_name: philox_offset + name: philox_offset + type: at::Tensor + - dynamic_type: at::Tensor + field_name: debug_attn_mask + name: debug_attn_mask + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _scaled_dot_product_cudnn_attention_backward + operator_name: _scaled_dot_product_cudnn_attention_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_scaled_dot_product_cudnn_attention_backward(Tensor grad_out, Tensor query, Tensor key, Tensor value, Tensor out, Tensor logsumexp, Tensor philox_seed, Tensor philox_offset, Tensor attn_bias, Tensor cum_seq_q, Tensor cum_seq_k, SymInt max_q, SymInt max_k, float dropout_p, bool is_causal, *, float? scale=None) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: attn_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_k + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, double, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: attn_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_k + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _flash_attention_forward + operator_name: _flash_attention_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_flash_attention_forward(Tensor query, Tensor key, Tensor value, Tensor? cum_seq_q, Tensor? cum_seq_k, SymInt max_q, SymInt max_k, float dropout_p, bool is_causal, bool return_debug_mask, *, float? scale=None, SymInt? window_size_left=None, SymInt? window_size_right=None, Tensor? seqused_k=None, Tensor? alibi_slopes=None) -> (Tensor output, Tensor softmax_logsumexp, Tensor rng_state, Tensor unused, Tensor debug_attn_mask) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cum_seq_q + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cum_seq_k + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: return_debug_mask + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size_left + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size_right + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: seqused_k + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: alibi_slopes + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, int64_t, int64_t, double, bool, bool, ::std::optional, ::std::optional, ::std::optional, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cum_seq_q + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cum_seq_k + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: return_debug_mask + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size_left + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size_right + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: seqused_k + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: alibi_slopes + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: softmax_logsumexp + name: softmax_logsumexp + type: at::Tensor + - dynamic_type: at::Tensor + field_name: rng_state + name: rng_state + type: at::Tensor + - dynamic_type: at::Tensor + field_name: unused + name: unused + type: at::Tensor + - dynamic_type: at::Tensor + field_name: debug_attn_mask + name: debug_attn_mask + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _flash_attention_backward + operator_name: _flash_attention_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_flash_attention_backward(Tensor grad_out, Tensor query, Tensor key, Tensor value, Tensor out, Tensor logsumexp, Tensor cum_seq_q, Tensor cum_seq_k, SymInt max_q, SymInt max_k, float dropout_p, bool is_causal, Tensor rng_state, Tensor unused, *, float? scale=None, SymInt? window_size_left=None, SymInt? window_size_right=None) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_k + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rng_state + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: unused + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size_left + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size_right + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, double, bool, const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_k + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rng_state + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: unused + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size_left + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size_right + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _efficient_attention_forward + operator_name: _efficient_attention_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_efficient_attention_forward(Tensor query, Tensor key, Tensor value, Tensor? bias, Tensor? cu_seqlens_q, Tensor? cu_seqlens_k, SymInt? max_seqlen_q, SymInt? max_seqlen_k, float dropout_p, int custom_mask_type, bool compute_log_sumexp=False, *, float? scale=None, Tensor? seqlen_k=None, int? window_size=None) -> (Tensor output, Tensor logsumexp, Tensor philox_seed, Tensor philox_offset, SymInt max_seqlen_batch_q, SymInt max_seqlen_batch_k) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cu_seqlens_q + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cu_seqlens_k + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: max_seqlen_q + type: ::std::optional + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: max_seqlen_k + type: ::std::optional + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: custom_mask_type + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: compute_log_sumexp + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: seqlen_k + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, ::std::optional, ::std::optional, double, int64_t, bool, ::std::optional, const ::std::optional &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cu_seqlens_q + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cu_seqlens_k + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: max_seqlen_q + type: ::std::optional + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: max_seqlen_k + type: ::std::optional + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: custom_mask_type + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: compute_log_sumexp + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: seqlen_k + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: logsumexp + name: logsumexp + type: at::Tensor + - dynamic_type: at::Tensor + field_name: philox_seed + name: philox_seed + type: at::Tensor + - dynamic_type: at::Tensor + field_name: philox_offset + name: philox_offset + type: at::Tensor + - dynamic_type: int64_t + field_name: max_seqlen_batch_q + name: max_seqlen_batch_q + type: c10::SymInt + - dynamic_type: int64_t + field_name: max_seqlen_batch_k + name: max_seqlen_batch_k + type: c10::SymInt + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _efficient_attention_backward + operator_name: _efficient_attention_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_efficient_attention_backward(Tensor grad_out_, Tensor query, Tensor key, Tensor value, Tensor? bias, Tensor out, Tensor? cu_seqlens_q, Tensor? cu_seqlens_k, SymInt max_seqlen_q, SymInt max_seqlen_k, Tensor logsumexp, float dropout_p, Tensor philox_seed, Tensor philox_offset, int custom_mask_type, bool bias_requires_grad, *, float? scale=None, int? num_splits_key=None, int? window_size=None, bool shared_storage_dqdkdv=False) -> (Tensor, Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cu_seqlens_q + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cu_seqlens_k + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_seqlen_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_seqlen_k + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: custom_mask_type + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: bias_requires_grad + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: num_splits_key + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: shared_storage_dqdkdv + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const at::Tensor &, const ::std::optional &, const ::std::optional &, int64_t, int64_t, const at::Tensor &, double, const at::Tensor &, const at::Tensor &, int64_t, bool, ::std::optional, ::std::optional, ::std::optional, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cu_seqlens_q + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cu_seqlens_k + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_seqlen_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_seqlen_k + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: custom_mask_type + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: bias_requires_grad + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: num_splits_key + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: window_size + type: ::std::optional + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: shared_storage_dqdkdv + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_attention_forward + operator_name: _cudnn_attention_forward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_attention_forward(Tensor query, Tensor key, Tensor value, Tensor? attn_bias, Tensor? cum_seq_q, Tensor? cum_seq_k, SymInt max_q, SymInt max_k, bool compute_log_sumexp, float dropout_p=0.0, bool is_causal=False, bool return_debug_mask=False, *, float? scale=None) -> (Tensor output, Tensor logsumexp, Tensor cum_seq_q, Tensor cum_seq_k, SymInt max_q, SymInt max_k, Tensor philox_seed, Tensor philox_offset, Tensor debug_attn_mask) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: attn_bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cum_seq_q + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cum_seq_k + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: compute_log_sumexp + type: bool + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_debug_mask + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, int64_t, int64_t, bool, double, bool, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: attn_bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cum_seq_q + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cum_seq_k + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: compute_log_sumexp + type: bool + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_debug_mask + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: logsumexp + name: logsumexp + type: at::Tensor + - dynamic_type: at::Tensor + field_name: cum_seq_q + name: cum_seq_q_return + type: at::Tensor + - dynamic_type: at::Tensor + field_name: cum_seq_k + name: cum_seq_k_return + type: at::Tensor + - dynamic_type: int64_t + field_name: max_q + name: max_q_return + type: c10::SymInt + - dynamic_type: int64_t + field_name: max_k + name: max_k_return + type: c10::SymInt + - dynamic_type: at::Tensor + field_name: philox_seed + name: philox_seed + type: at::Tensor + - dynamic_type: at::Tensor + field_name: philox_offset + name: philox_offset + type: at::Tensor + - dynamic_type: at::Tensor + field_name: debug_attn_mask + name: debug_attn_mask + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_attention_backward + operator_name: _cudnn_attention_backward + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_attention_backward(Tensor grad_out, Tensor query, Tensor key, Tensor value, Tensor out, Tensor logsumexp, Tensor philox_seed, Tensor philox_offset, Tensor attn_bias, Tensor cum_seq_q, Tensor cum_seq_k, SymInt max_q, SymInt max_k, float dropout_p, bool is_causal, *, float? scale=None) -> (Tensor, Tensor, Tensor) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: attn_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_k + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, double, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: logsumexp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_seed + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: philox_offset + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: attn_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cum_seq_k + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_q + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: max_k + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: is_causal + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: double + is_nullable: true + kwarg_only: true + name: scale + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _triton_scaled_dot_attention + operator_name: _triton_scaled_dot_attention + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_triton_scaled_dot_attention(Tensor q, Tensor k, Tensor v, float dropout_p=0.0) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: k + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: k + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fill_mem_eff_dropout_mask_ + operator_name: _fill_mem_eff_dropout_mask_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fill_mem_eff_dropout_mask_(Tensor(a!) self, float dropout_p, int seed, int offset) -> Tensor(a!) + arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: seed + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + schema_order_cpp_signature: at::Tensor & (at::Tensor &, double, int64_t, int64_t) + schema_order_arguments: + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: self + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: seed + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: self + type: at::Tensor & + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _triton_multi_head_attention + operator_name: _triton_multi_head_attention + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_triton_multi_head_attention(Tensor query, Tensor key, Tensor value, int embed_dim, int num_head, Tensor qkv_weight, Tensor qkv_bias, Tensor proj_weight, Tensor proj_bias, Tensor? mask=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_head + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_head + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_airy_ai + operator_name: special_airy_ai + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_airy_ai(Tensor x) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_airy_ai_out + operator_name: special_airy_ai + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_airy_ai.out(Tensor x, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_bessel_j0 + operator_name: special_bessel_j0 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_bessel_j0(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_bessel_j0_out + operator_name: special_bessel_j0 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_bessel_j0.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_bessel_j1 + operator_name: special_bessel_j1 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_bessel_j1(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_bessel_j1_out + operator_name: special_bessel_j1 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_bessel_j1.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_bessel_y0 + operator_name: special_bessel_y0 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_bessel_y0(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_bessel_y0_out + operator_name: special_bessel_y0 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_bessel_y0.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_bessel_y1 + operator_name: special_bessel_y1 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_bessel_y1(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_bessel_y1_out + operator_name: special_bessel_y1 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_bessel_y1.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_t + operator_name: special_chebyshev_polynomial_t + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_t(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_t + operator_name: special_chebyshev_polynomial_t + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_t.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_t + operator_name: special_chebyshev_polynomial_t + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_t.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_t_out + operator_name: special_chebyshev_polynomial_t + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_t.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_t_out + operator_name: special_chebyshev_polynomial_t + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_t.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_t_out + operator_name: special_chebyshev_polynomial_t + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_t.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_u + operator_name: special_chebyshev_polynomial_u + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_u(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_u + operator_name: special_chebyshev_polynomial_u + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_u.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_u + operator_name: special_chebyshev_polynomial_u + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_u.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_u_out + operator_name: special_chebyshev_polynomial_u + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_u.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_u_out + operator_name: special_chebyshev_polynomial_u + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_u.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_u_out + operator_name: special_chebyshev_polynomial_u + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_u.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_v + operator_name: special_chebyshev_polynomial_v + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_v(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_v + operator_name: special_chebyshev_polynomial_v + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_v.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_v + operator_name: special_chebyshev_polynomial_v + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_v.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_v_out + operator_name: special_chebyshev_polynomial_v + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_v.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_v_out + operator_name: special_chebyshev_polynomial_v + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_v.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_v_out + operator_name: special_chebyshev_polynomial_v + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_v.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_w + operator_name: special_chebyshev_polynomial_w + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_w(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_w + operator_name: special_chebyshev_polynomial_w + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_w.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_w + operator_name: special_chebyshev_polynomial_w + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_w.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_w_out + operator_name: special_chebyshev_polynomial_w + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_w.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_w_out + operator_name: special_chebyshev_polynomial_w + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_w.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_chebyshev_polynomial_w_out + operator_name: special_chebyshev_polynomial_w + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_chebyshev_polynomial_w.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_h + operator_name: special_hermite_polynomial_h + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_h(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_h + operator_name: special_hermite_polynomial_h + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_h.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_h + operator_name: special_hermite_polynomial_h + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_h.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_h_out + operator_name: special_hermite_polynomial_h + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_h.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_h_out + operator_name: special_hermite_polynomial_h + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_h.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_h_out + operator_name: special_hermite_polynomial_h + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_h.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_he + operator_name: special_hermite_polynomial_he + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_he(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_he + operator_name: special_hermite_polynomial_he + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_he.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_he + operator_name: special_hermite_polynomial_he + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_he.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_he_out + operator_name: special_hermite_polynomial_he + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_he.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_he_out + operator_name: special_hermite_polynomial_he + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_he.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_hermite_polynomial_he_out + operator_name: special_hermite_polynomial_he + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_hermite_polynomial_he.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_laguerre_polynomial_l + operator_name: special_laguerre_polynomial_l + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_laguerre_polynomial_l(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_laguerre_polynomial_l + operator_name: special_laguerre_polynomial_l + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_laguerre_polynomial_l.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_laguerre_polynomial_l + operator_name: special_laguerre_polynomial_l + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_laguerre_polynomial_l.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_laguerre_polynomial_l_out + operator_name: special_laguerre_polynomial_l + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_laguerre_polynomial_l.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_laguerre_polynomial_l_out + operator_name: special_laguerre_polynomial_l + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_laguerre_polynomial_l.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_laguerre_polynomial_l_out + operator_name: special_laguerre_polynomial_l + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_laguerre_polynomial_l.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_legendre_polynomial_p + operator_name: special_legendre_polynomial_p + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_legendre_polynomial_p(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_legendre_polynomial_p + operator_name: special_legendre_polynomial_p + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_legendre_polynomial_p.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_legendre_polynomial_p + operator_name: special_legendre_polynomial_p + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_legendre_polynomial_p.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_legendre_polynomial_p_out + operator_name: special_legendre_polynomial_p + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_legendre_polynomial_p.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_legendre_polynomial_p_out + operator_name: special_legendre_polynomial_p + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_legendre_polynomial_p.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_legendre_polynomial_p_out + operator_name: special_legendre_polynomial_p + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_legendre_polynomial_p.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_modified_bessel_i0 + operator_name: special_modified_bessel_i0 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_modified_bessel_i0(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_modified_bessel_i0_out + operator_name: special_modified_bessel_i0 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_modified_bessel_i0.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_modified_bessel_i1 + operator_name: special_modified_bessel_i1 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_modified_bessel_i1(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_modified_bessel_i1_out + operator_name: special_modified_bessel_i1 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_modified_bessel_i1.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_modified_bessel_k0 + operator_name: special_modified_bessel_k0 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_modified_bessel_k0(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_modified_bessel_k0_out + operator_name: special_modified_bessel_k0 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_modified_bessel_k0.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_modified_bessel_k1 + operator_name: special_modified_bessel_k1 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_modified_bessel_k1(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_modified_bessel_k1_out + operator_name: special_modified_bessel_k1 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_modified_bessel_k1.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_scaled_modified_bessel_k0 + operator_name: special_scaled_modified_bessel_k0 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_scaled_modified_bessel_k0(Tensor x) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_scaled_modified_bessel_k0_out + operator_name: special_scaled_modified_bessel_k0 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_scaled_modified_bessel_k0.out(Tensor x, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_scaled_modified_bessel_k1 + operator_name: special_scaled_modified_bessel_k1 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_scaled_modified_bessel_k1(Tensor x) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_scaled_modified_bessel_k1_out + operator_name: special_scaled_modified_bessel_k1 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_scaled_modified_bessel_k1.out(Tensor x, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_t + operator_name: special_shifted_chebyshev_polynomial_t + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_t(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_t + operator_name: special_shifted_chebyshev_polynomial_t + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_t.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_t + operator_name: special_shifted_chebyshev_polynomial_t + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_t.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_t_out + operator_name: special_shifted_chebyshev_polynomial_t + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_t.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_t_out + operator_name: special_shifted_chebyshev_polynomial_t + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_t.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_t_out + operator_name: special_shifted_chebyshev_polynomial_t + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_t.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_u + operator_name: special_shifted_chebyshev_polynomial_u + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_u(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_u + operator_name: special_shifted_chebyshev_polynomial_u + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_u.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_u + operator_name: special_shifted_chebyshev_polynomial_u + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_u.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_u_out + operator_name: special_shifted_chebyshev_polynomial_u + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_u.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_u_out + operator_name: special_shifted_chebyshev_polynomial_u + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_u.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_u_out + operator_name: special_shifted_chebyshev_polynomial_u + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_u.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_v + operator_name: special_shifted_chebyshev_polynomial_v + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_v(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_v + operator_name: special_shifted_chebyshev_polynomial_v + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_v.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_v + operator_name: special_shifted_chebyshev_polynomial_v + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_v.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_v_out + operator_name: special_shifted_chebyshev_polynomial_v + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_v.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_v_out + operator_name: special_shifted_chebyshev_polynomial_v + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_v.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_v_out + operator_name: special_shifted_chebyshev_polynomial_v + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_v.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_w + operator_name: special_shifted_chebyshev_polynomial_w + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_w(Tensor x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_w + operator_name: special_shifted_chebyshev_polynomial_w + overload_name: x_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_w.x_scalar(Scalar x, Tensor n) -> Tensor + arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Scalar &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_w + operator_name: special_shifted_chebyshev_polynomial_w + overload_name: n_scalar + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_w.n_scalar(Tensor x, Scalar n) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Scalar &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_w_out + operator_name: special_shifted_chebyshev_polynomial_w + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_w.out(Tensor x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_w_out + operator_name: special_shifted_chebyshev_polynomial_w + overload_name: x_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_w.x_scalar_out(Scalar x, Tensor n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: x + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: n + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_shifted_chebyshev_polynomial_w_out + operator_name: special_shifted_chebyshev_polynomial_w + overload_name: n_scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_shifted_chebyshev_polynomial_w.n_scalar_out(Tensor x, Scalar n, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: n + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_spherical_bessel_j0 + operator_name: special_spherical_bessel_j0 + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_spherical_bessel_j0(Tensor x) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: special_spherical_bessel_j0_out + operator_name: special_spherical_bessel_j0 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::special_spherical_bessel_j0.out(Tensor x, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: special + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foobar + operator_name: _foobar + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foobar(Tensor self, bool arg1=True, bool arg2=True, *, bool arg3=True) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: arg1 + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: arg2 + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: arg3 + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: arg1 + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: arg2 + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: arg3 + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adam_ + operator_name: _fused_adam_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adam_(Tensor(a!)[] self, Tensor(b!)[] grads, Tensor(c!)[] exp_avgs, Tensor(d!)[] exp_avg_sqs, Tensor(e!)[] max_exp_avg_sqs, Tensor[] state_steps, *, float lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, double, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adam_ + operator_name: _fused_adam_ + overload_name: tensor_lr + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adam_.tensor_lr(Tensor(a!)[] self, Tensor(b!)[] grads, Tensor(c!)[] exp_avgs, Tensor(d!)[] exp_avg_sqs, Tensor(e!)[] max_exp_avg_sqs, Tensor[] state_steps, *, Tensor lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adamw_ + operator_name: _fused_adamw_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adamw_(Tensor(a!)[] self, Tensor(b!)[] grads, Tensor(c!)[] exp_avgs, Tensor(d!)[] exp_avg_sqs, Tensor(e!)[] max_exp_avg_sqs, Tensor[] state_steps, *, float lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, double, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adamw_ + operator_name: _fused_adamw_ + overload_name: tensor_lr + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adamw_.tensor_lr(Tensor(a!)[] self, Tensor(b!)[] grads, Tensor(c!)[] exp_avgs, Tensor(d!)[] exp_avg_sqs, Tensor(e!)[] max_exp_avg_sqs, Tensor[] state_steps, *, Tensor lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_sgd_ + operator_name: _fused_sgd_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_sgd_(Tensor(a!)[] self, Tensor(b!)[] grads, Tensor(c!)[] momentum_buffer_list, *, float weight_decay, float momentum, float lr, float dampening, bool nesterov, bool maximize, bool is_first_step, Tensor? grad_scale=None, Tensor? found_inf=None) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, double, double, double, double, bool, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_sgd_ + operator_name: _fused_sgd_ + overload_name: tensor_lr + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_sgd_.tensor_lr(Tensor(a!)[] self, Tensor(b!)[] grads, Tensor(c!)[] momentum_buffer_list, *, float weight_decay, float momentum, Tensor lr, float dampening, bool nesterov, bool maximize, bool is_first_step, Tensor? grad_scale=None, Tensor? found_inf=None) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, double, double, const at::Tensor &, double, bool, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adagrad_ + operator_name: _fused_adagrad_ + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adagrad_(Tensor(a!)[] self, Tensor(b!)[] grads, Tensor(c!)[] state_sums, Tensor(d!)[] state_steps, *, float lr, float lr_decay, float weight_decay, float eps, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, double, double, double, double, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adagrad_ + operator_name: _fused_adagrad_ + overload_name: tensor_lr + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adagrad_.tensor_lr(Tensor(a!)[] self, Tensor(b!)[] grads, Tensor(c!)[] state_sums, Tensor[] state_steps, *, Tensor lr, float lr_decay, float weight_decay, float eps, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> () + arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, double, double, double, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: true + is_factory_method: false + abstract: true + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: false +- name: _propagate_xla_data + operator_name: _propagate_xla_data + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_propagate_xla_data(Tensor input, Tensor output) -> () + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + schema_order_cpp_signature: void (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: false + device_guard: true + with_gil: false + deprecated: false + has_math_kernel: true +- name: _new_zeros_with_same_feature_meta_out + operator_name: _new_zeros_with_same_feature_meta + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_new_zeros_with_same_feature_meta.out(Tensor self, Tensor other, *, int self_num_batch_dims=0, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: self_num_batch_dims + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: self_num_batch_dims + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_ctc_loss_out + operator_name: _cudnn_ctc_loss + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_ctc_loss.out(Tensor log_probs, Tensor targets, int[] input_lengths, int[] target_lengths, int blank, bool deterministic, bool zero_infinity, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_rnn_flatten_weight_out + operator_name: _cudnn_rnn_flatten_weight + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_rnn_flatten_weight.out(Tensor[] weight_arr, int weight_stride0, SymInt input_size, int mode, SymInt hidden_size, SymInt proj_size, int num_layers, bool batch_first, bool bidirectional, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight_arr + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: input_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + schema_order_cpp_signature: at::Tensor & (at::TensorList, int64_t, int64_t, int64_t, int64_t, int64_t, int64_t, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight_arr + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: input_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_rnn_out + operator_name: _cudnn_rnn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_rnn.out(Tensor input, Tensor[] weight, int weight_stride0, Tensor? weight_buf, Tensor hx, Tensor? cx, int mode, SymInt hidden_size, SymInt proj_size, int num_layers, bool batch_first, float dropout, bool train, bool bidirectional, SymInt[] batch_sizes, Tensor? dropout_state, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3, Tensor(e!) out4) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!), Tensor(e!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out4 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight_buf + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::TensorList, int64_t, const ::std::optional &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, int64_t, int64_t, bool, double, bool, bool, at::IntArrayRef, const ::std::optional &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight_buf + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out4 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out4 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_rnn_backward_out + operator_name: _cudnn_rnn_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_rnn_backward.out(Tensor input, Tensor[] weight, int weight_stride0, Tensor weight_buf, Tensor hx, Tensor? cx, Tensor output, Tensor? grad_output, Tensor? grad_hy, Tensor? grad_cy, int mode, SymInt hidden_size, SymInt proj_size, int num_layers, bool batch_first, float dropout, bool train, bool bidirectional, SymInt[] batch_sizes, Tensor? dropout_state, Tensor reserve, bool[4] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!)[] out3) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: out3 + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_buf + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserve + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: void (const at::Tensor &, at::TensorList, int64_t, const at::Tensor &, const at::Tensor &, const ::std::optional &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, int64_t, int64_t, int64_t, int64_t, bool, double, bool, bool, at::IntArrayRef, const ::std::optional &, const at::Tensor &, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_buf + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: proj_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserve + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: out3 + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cudnn_init_dropout_state_out + operator_name: _cudnn_init_dropout_state + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cudnn_init_dropout_state.out(float dropout, bool train, int dropout_seed, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dropout_seed + type: int64_t + schema_order_cpp_signature: at::Tensor & (double, bool, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dropout_seed + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_dropout_out + operator_name: _fused_dropout + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_dropout.out(Tensor self, float p, Generator? generator=None, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, double, ::std::optional, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _masked_scale_out + operator_name: _masked_scale + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_masked_scale.out(Tensor self, Tensor mask, float scale, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_dropout_out + operator_name: native_dropout + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_dropout.out(Tensor input, float p, bool? train, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: true + name: train + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, double, ::std::optional, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: bool + is_nullable: true + name: train + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_dropout_backward_out + operator_name: native_dropout_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_dropout_backward.out(Tensor grad_output, Tensor mask, float scale, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _conj_physical_out + operator_name: _conj_physical + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_conj_physical.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: avg_pool1d_out + operator_name: avg_pool1d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::avg_pool1d.out(Tensor self, int[1] kernel_size, int[1] stride=[], int[1] padding=0, bool ceil_mode=False, bool count_include_pad=True, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: count_include_pad + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: adaptive_avg_pool1d_out + operator_name: adaptive_avg_pool1d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::adaptive_avg_pool1d.out(Tensor self, int[1] output_size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 1 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _add_relu_out + operator_name: _add_relu + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_add_relu.Scalar_out(Tensor self, Scalar other, Scalar alpha=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: add_out + operator_name: add + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::add.Scalar_out(Tensor self, Scalar other, Scalar alpha=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: affine_grid_generator_out + operator_name: affine_grid_generator + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::affine_grid_generator.out(Tensor theta, SymInt[] size, bool align_corners, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: theta + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: theta + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_functorch_fallback_out + operator_name: _test_functorch_fallback + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_functorch_fallback.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bartlett_window_out + operator_name: bartlett_window + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bartlett_window.out(int window_length, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + schema_order_cpp_signature: at::Tensor & (int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bartlett_window_out + operator_name: bartlett_window + overload_name: periodic_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bartlett_window.periodic_out(int window_length, bool periodic, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + schema_order_cpp_signature: at::Tensor & (int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantized_batch_norm_out + operator_name: quantized_batch_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_batch_norm.out(Tensor input, Tensor? weight, Tensor? bias, Tensor mean, Tensor var, float eps, float output_scale, int output_zero_point, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: var + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: output_scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: output_zero_point + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, double, double, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: var + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: output_scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: output_zero_point + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bernoulli_out + operator_name: bernoulli + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bernoulli.Tensor_out(Tensor self, Tensor p, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: p + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: p + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bernoulli + operator_name: bernoulli + overload_name: Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::bernoulli.Tensor(Tensor self, Tensor p, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: p + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: p + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bernoulli_out + operator_name: bernoulli + overload_name: float_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bernoulli.float_out(Tensor self, float p=0.5, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0.5 + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: binary_cross_entropy_with_logits_out + operator_name: binary_cross_entropy_with_logits + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::binary_cross_entropy_with_logits.out(Tensor self, Tensor target, Tensor? weight=None, Tensor? pos_weight=None, int reduction=Mean, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: pos_weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: pos_weight + type: const ::std::optional & + - annotation: null + default: at::Reduction::Mean + dynamic_type: int64_t + is_nullable: false + name: reduction + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bincount_out + operator_name: bincount + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bincount.out(Tensor self, Tensor? weights=None, SymInt minlength=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weights + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: minlength + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: weights + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: minlength + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: blackman_window_out + operator_name: blackman_window + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::blackman_window.out(int window_length, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + schema_order_cpp_signature: at::Tensor & (int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: blackman_window_out + operator_name: blackman_window + overload_name: periodic_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::blackman_window.periodic_out(int window_length, bool periodic, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + schema_order_cpp_signature: at::Tensor & (int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: block_diag_out + operator_name: block_diag + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::block_diag.out(Tensor[] tensors, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: at::Tensor & (at::TensorList, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: constant_pad_nd_out + operator_name: constant_pad_nd + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::constant_pad_nd.out(Tensor self, SymInt[] pad, Scalar value=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: pad + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: pad + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: convolution_out + operator_name: convolution + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::convolution.out(Tensor input, Tensor weight, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: convolution_backward_out + operator_name: convolution_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::convolution_backward.out(Tensor grad_output, Tensor input, Tensor weight, SymInt[]? bias_sizes, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups, bool[3] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: bias_sizes + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::OptionalIntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: bias_sizes + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: convolution_overrideable_out + operator_name: convolution_overrideable + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::convolution_overrideable.out(Tensor input, Tensor weight, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: convolution_backward_overrideable_out + operator_name: convolution_backward_overrideable + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::convolution_backward_overrideable.out(Tensor grad_output, Tensor input, Tensor weight, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups, bool[3] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _convolution_out + operator_name: _convolution + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_convolution.out(Tensor input, Tensor weight, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, bool transposed, SymInt[] output_padding, SymInt groups, bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::IntArrayRef, int64_t, bool, bool, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: transposed + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: cudnn_enabled + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: conv_tbc_out + operator_name: conv_tbc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv_tbc.out(Tensor self, Tensor weight, Tensor bias, int pad=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: pad + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bias + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: pad + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: copy_out + operator_name: copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::copy.out(Tensor self, Tensor src, bool non_blocking=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _copy_from_out + operator_name: _copy_from + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_copy_from.out(Tensor self, Tensor dst, bool non_blocking=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dst + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dst + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _copy_from_and_resize_out + operator_name: _copy_from_and_resize + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_copy_from_and_resize.out(Tensor self, Tensor dst, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dst + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dst + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: count_nonzero_out + operator_name: count_nonzero + overload_name: dim_IntList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::count_nonzero.dim_IntList_out(Tensor self, int[] dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: count_nonzero_out + operator_name: count_nonzero + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::count_nonzero.out(Tensor self, int? dim=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_affine_grid_generator_out + operator_name: cudnn_affine_grid_generator + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_affine_grid_generator.out(Tensor theta, int N, int C, int H, int W, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: theta + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: H + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: W + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: theta + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: H + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: W + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_affine_grid_generator_backward_out + operator_name: cudnn_affine_grid_generator_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_affine_grid_generator_backward.out(Tensor grad, int N, int C, int H, int W, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: H + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: W + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: H + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: W + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_batch_norm_backward_out + operator_name: cudnn_batch_norm_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_batch_norm_backward.out(Tensor input, Tensor grad_output, Tensor weight, Tensor? running_mean, Tensor? running_var, Tensor? save_mean, Tensor? save_var, float epsilon, Tensor reserveSpace, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserveSpace + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, double, const at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserveSpace + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_convolution_transpose_out + operator_name: cudnn_convolution_transpose + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_convolution_transpose.out(Tensor self, Tensor weight, SymInt[] padding, SymInt[] output_padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool benchmark, bool deterministic, bool allow_tf32, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: allow_tf32 + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _mps_convolution_transpose_out + operator_name: _mps_convolution_transpose + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_mps_convolution_transpose.out(Tensor self, Tensor weight, SymInt[] padding, SymInt[] output_padding, SymInt[] stride, SymInt[] dilation, SymInt groups, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mps_convolution_transpose_backward_out + operator_name: mps_convolution_transpose_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mps_convolution_transpose_backward.out(Tensor self, Tensor grad_output, Tensor weight, SymInt[] padding, SymInt[] output_padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool[2] output_mask, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, ::std::array, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_convolution_relu_out + operator_name: cudnn_convolution_relu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_convolution_relu.out(Tensor self, Tensor weight, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, SymInt groups, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_convolution_add_relu_out + operator_name: cudnn_convolution_add_relu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_convolution_add_relu.out(Tensor self, Tensor weight, Tensor z, Scalar? alpha, Tensor? bias, SymInt[] stride, SymInt[] padding, SymInt[] dilation, SymInt groups, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: z + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: alpha + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: z + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: alpha + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_grid_sampler_out + operator_name: cudnn_grid_sampler + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_grid_sampler.out(Tensor self, Tensor grid, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: cudnn_grid_sampler_backward_out + operator_name: cudnn_grid_sampler_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cudnn_grid_sampler_backward.out(Tensor self, Tensor grid, Tensor grad_output, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _ctc_loss_out + operator_name: _ctc_loss + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_ctc_loss.out(Tensor log_probs, Tensor targets, int[] input_lengths, int[] target_lengths, int blank=0, bool zero_infinity=False, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _ctc_loss_out + operator_name: _ctc_loss + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_ctc_loss.Tensor_out(Tensor log_probs, Tensor targets, Tensor input_lengths, Tensor target_lengths, int blank=0, bool zero_infinity=False, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_lengths + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: target_lengths + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _ctc_loss_backward_out + operator_name: _ctc_loss_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_ctc_loss_backward.out(Tensor grad, Tensor log_probs, Tensor targets, int[] input_lengths, int[] target_lengths, Tensor neg_log_likelihood, Tensor log_alpha, int blank, bool zero_infinity=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: neg_log_likelihood + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_alpha + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, const at::Tensor &, const at::Tensor &, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_probs + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: targets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: target_lengths + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: neg_log_likelihood + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: log_alpha + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: blank + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: zero_infinity + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: diag_embed_out + operator_name: diag_embed + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::diag_embed.out(Tensor self, int offset=0, int dim1=-2, int dim2=-1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: -2 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: -2 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: diagonal_backward_out + operator_name: diagonal_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::diagonal_backward.out(Tensor grad_output, SymInt[] input_sizes, int offset, int dim1, int dim2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: div_out + operator_name: div + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::div.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: div_out + operator_name: div + overload_name: Scalar_mode_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::div.Scalar_mode_out(Tensor self, Scalar other, *, str? rounding_mode, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + dynamic_type: c10::string_view + is_nullable: true + kwarg_only: true + name: rounding_mode + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: embedding_out + operator_name: embedding + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding.out(Tensor weight, Tensor indices, SymInt padding_idx=-1, bool scale_grad_by_freq=False, bool sparse=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: embedding_dense_backward_out + operator_name: embedding_dense_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding_dense_backward.out(Tensor grad_output, Tensor indices, SymInt num_weights, SymInt padding_idx, bool scale_grad_by_freq, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: embedding_renorm_out + operator_name: embedding_renorm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding_renorm.out(Tensor self, Tensor indices, float max_norm, float norm_type, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: max_norm + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: norm_type + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, double, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: max_norm + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: norm_type + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: embedding_renorm + operator_name: embedding_renorm + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::embedding_renorm(Tensor self, Tensor indices, float max_norm, float norm_type) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: max_norm + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: norm_type + type: double + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: max_norm + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: norm_type + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _embedding_bag_forward_only_out + operator_name: _embedding_bag_forward_only + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_embedding_bag_forward_only.out(Tensor weight, Tensor indices, Tensor offsets, bool scale_grad_by_freq=False, int mode=0, bool sparse=False, Tensor? per_sample_weights=None, bool include_last_offset=False, int padding_idx=-1, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, int64_t, bool, const ::std::optional &, bool, int64_t, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _embedding_bag_out + operator_name: _embedding_bag + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_embedding_bag.out(Tensor weight, Tensor indices, Tensor offsets, bool scale_grad_by_freq=False, int mode=0, bool sparse=False, Tensor? per_sample_weights=None, bool include_last_offset=False, int padding_idx=-1, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, int64_t, bool, const ::std::optional &, bool, int64_t, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: sparse + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: include_last_offset + type: bool + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _embedding_bag_dense_backward_out + operator_name: _embedding_bag_dense_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_embedding_bag_dense_backward.out(Tensor grad, Tensor indices, Tensor offset2bag, Tensor bag_size, Tensor maximum_indices, SymInt num_weights, bool scale_grad_by_freq, int mode, Tensor? per_sample_weights, int padding_idx=-1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bag_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: maximum_indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, bool, int64_t, const ::std::optional &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: bag_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: maximum_indices + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_weights + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: scale_grad_by_freq + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: per_sample_weights + type: const ::std::optional & + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _embedding_bag_per_sample_weights_backward_out + operator_name: _embedding_bag_per_sample_weights_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_embedding_bag_per_sample_weights_backward.out(Tensor grad, Tensor weight, Tensor indices, Tensor offsets, Tensor offset2bag, int mode, int padding_idx=-1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offset2bag + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + default: -1 + dynamic_type: int64_t + is_nullable: false + name: padding_idx + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty_out + operator_name: empty + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty.names_out(int[] size, *, Dimname[]? names, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty_permuted_out + operator_name: empty_permuted + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty_permuted.out(SymInt[] size, int[] physical_layout, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: physical_layout + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: physical_layout + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: new_empty_out + operator_name: new_empty + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::new_empty.out(Tensor self, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: new_empty_strided_out + operator_name: new_empty_strided + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::new_empty_strided.out(Tensor self, SymInt[] size, SymInt[] stride, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: new_full_out + operator_name: new_full + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::new_full.out(Tensor self, SymInt[] size, Scalar fill_value, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: new_zeros_out + operator_name: new_zeros + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::new_zeros.out(Tensor self, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: new_ones_out + operator_name: new_ones + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::new_ones.out(Tensor self, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _empty_affine_quantized_out + operator_name: _empty_affine_quantized + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_empty_affine_quantized.out(SymInt[] size, *, float scale=1, int zero_point=0, MemoryFormat? memory_format=contiguous_format, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + kwarg_only: true + name: scale + type: double + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: zero_point + type: int64_t + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, double, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + kwarg_only: true + name: scale + type: double + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: zero_point + type: int64_t + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _empty_per_channel_affine_quantized_out + operator_name: _empty_per_channel_affine_quantized + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_empty_per_channel_affine_quantized.out(SymInt[] size, *, Tensor scales, Tensor zero_points, int axis, MemoryFormat? memory_format=contiguous_format, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, const at::Tensor &, const at::Tensor &, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: c10::MemoryFormat::Contiguous + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: resize_out + operator_name: resize + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::resize.out(Tensor self, SymInt[] size, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: const at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: resize + operator_name: resize + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::resize(Tensor self, SymInt[] size, *, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _resize_output_out + operator_name: _resize_output + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_resize_output.out(Tensor self, SymInt[] size, Device device, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Device + is_nullable: false + name: device + type: at::Device + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Device, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Device + is_nullable: false + name: device + type: at::Device + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: const at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _resize_output + operator_name: _resize_output + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_resize_output(Tensor self, SymInt[] size, Device device) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Device + is_nullable: false + name: device + type: at::Device + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, at::Device) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Device + is_nullable: false + name: device + type: at::Device + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty_quantized_out + operator_name: empty_quantized + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty_quantized.out(int[] size, Tensor qtensor, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qtensor + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qtensor + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty_like_out + operator_name: empty_like + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty_like.out(Tensor self, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: empty_strided_out + operator_name: empty_strided + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::empty_strided.out(SymInt[] size, SymInt[] stride, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: fill_out + operator_name: fill + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fill.Scalar_out(Tensor self, Scalar value, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: fill_out + operator_name: fill + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fill.Tensor_out(Tensor self, Tensor value, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: floor_divide_out + operator_name: floor_divide + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::floor_divide.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: full_out + operator_name: full + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::full.names_out(int[] size, Scalar fill_value, *, Dimname[]? names, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, const at::Scalar &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: full_like_out + operator_name: full_like + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::full_like.out(Tensor self, Scalar fill_value, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: fill_value + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: from_file_out + operator_name: from_file + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::from_file.out(str filename, bool? shared=None, int? size=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: filename + type: c10::string_view + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: shared + type: ::std::optional + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: true + name: size + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (c10::string_view, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: filename + type: c10::string_view + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: shared + type: ::std::optional + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: true + name: size + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: grid_sampler_2d_out + operator_name: grid_sampler_2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::grid_sampler_2d.out(Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: grid_sampler_2d_backward_out + operator_name: grid_sampler_2d_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::grid_sampler_2d_backward.out(Tensor grad_output, Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners, bool[2] output_mask, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool, ::std::array, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _grid_sampler_2d_cpu_fallback_out + operator_name: _grid_sampler_2d_cpu_fallback + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_grid_sampler_2d_cpu_fallback.out(Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: grid_sampler_3d_out + operator_name: grid_sampler_3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::grid_sampler_3d.out(Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: grid_sampler_3d_backward_out + operator_name: grid_sampler_3d_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::grid_sampler_3d_backward.out(Tensor grad_output, Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool align_corners, bool[2] output_mask, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, bool, ::std::array, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grid + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: interpolation_mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: padding_mode + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: hann_window_out + operator_name: hann_window + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hann_window.out(int window_length, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + schema_order_cpp_signature: at::Tensor & (int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: hann_window_out + operator_name: hann_window + overload_name: periodic_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hann_window.periodic_out(int window_length, bool periodic, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + schema_order_cpp_signature: at::Tensor & (int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: hamming_window_out + operator_name: hamming_window + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hamming_window.out(int window_length, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + schema_order_cpp_signature: at::Tensor & (int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: hamming_window_out + operator_name: hamming_window + overload_name: periodic_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hamming_window.periodic_out(int window_length, bool periodic, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + schema_order_cpp_signature: at::Tensor & (int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: hamming_window_out + operator_name: hamming_window + overload_name: periodic_alpha_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hamming_window.periodic_alpha_out(int window_length, bool periodic, float alpha, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: alpha + type: double + schema_order_cpp_signature: at::Tensor & (int64_t, bool, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: alpha + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: hamming_window_out + operator_name: hamming_window + overload_name: periodic_alpha_beta_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hamming_window.periodic_alpha_beta_out(int window_length, bool periodic, float alpha, float beta, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: alpha + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + schema_order_cpp_signature: at::Tensor & (int64_t, bool, double, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: alpha + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: kaiser_window_out + operator_name: kaiser_window + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::kaiser_window.out(int window_length, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + schema_order_cpp_signature: at::Tensor & (int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: kaiser_window_out + operator_name: kaiser_window + overload_name: periodic_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::kaiser_window.periodic_out(int window_length, bool periodic, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + schema_order_cpp_signature: at::Tensor & (int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: kaiser_window_out + operator_name: kaiser_window + overload_name: beta_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::kaiser_window.beta_out(int window_length, bool periodic, float beta, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + schema_order_cpp_signature: at::Tensor & (int64_t, bool, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: window_length + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: periodic + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: beta + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_group_norm_out + operator_name: native_group_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_group_norm.out(Tensor input, Tensor? weight, Tensor? bias, SymInt N, SymInt C, SymInt HxW, int group, float eps, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: HxW + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: group + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, int64_t, int64_t, int64_t, int64_t, double, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: HxW + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: group + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_group_norm_backward_out + operator_name: native_group_norm_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_group_norm_backward.out(Tensor grad_out, Tensor input, Tensor mean, Tensor rstd, Tensor? weight, SymInt N, SymInt C, SymInt HxW, int group, bool[3] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: HxW + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: group + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, int64_t, int64_t, int64_t, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: N + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: C + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: HxW + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: group + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_put_out + operator_name: index_put + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_put.out(Tensor self, Tensor?[] indices, Tensor values, bool accumulate=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const c10::List<::std::optional> &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _index_put_impl_out + operator_name: _index_put_impl + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_index_put_impl.out(Tensor self, Tensor?[] indices, Tensor values, bool accumulate=False, bool unsafe=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: unsafe + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const c10::List<::std::optional> &, const at::Tensor &, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: unsafe + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _index_put_impl + operator_name: _index_put_impl + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_index_put_impl(Tensor self, Tensor?[] indices, Tensor values, bool accumulate=False, bool unsafe=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: unsafe + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const c10::List<::std::optional> &, const at::Tensor &, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const c10::List<::std::optional> & + is_nullable: true + name: indices + type: const c10::List<::std::optional> & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: unsafe + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: isnan_out + operator_name: isnan + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::isnan.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_layer_norm_out + operator_name: native_layer_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_layer_norm.out(Tensor input, SymInt[] normalized_shape, Tensor? weight, Tensor? bias, float eps, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::IntArrayRef, const ::std::optional &, const ::std::optional &, double, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_layer_norm_backward_out + operator_name: native_layer_norm_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_layer_norm_backward.out(Tensor grad_out, Tensor input, SymInt[] normalized_shape, Tensor mean, Tensor rstd, Tensor? weight, Tensor? bias, bool[3] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: normalized_shape + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: rstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: linear_backward_out + operator_name: linear_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linear_backward.out(Tensor self, Tensor grad_output, Tensor weight, bool[3] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_linear_out + operator_name: mkldnn_linear + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_linear.out(Tensor self, Tensor weight, Tensor? bias=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_linear_backward_input_out + operator_name: mkldnn_linear_backward_input + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_linear_backward_input.out(int[] input_size, Tensor grad_output, Tensor weight, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_linear_backward_weights_out + operator_name: mkldnn_linear_backward_weights + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_linear_backward_weights.out(Tensor grad_output, Tensor input, Tensor weight, bool bias_defined, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: bias_defined + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: bias_defined + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_linear_backward_out + operator_name: mkldnn_linear_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_linear_backward.out(Tensor self, Tensor grad_output, Tensor weight, bool[3] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: matmul_backward_out + operator_name: matmul_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::matmul_backward.out(Tensor grad, Tensor self, Tensor other, bool[2] mask, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::array, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _aminmax_out + operator_name: _aminmax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_aminmax.out(Tensor self, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _aminmax_out + operator_name: _aminmax + overload_name: dim_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_aminmax.dim_out(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: max_pool2d_backward_out + operator_name: max_pool2d_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::max_pool2d_backward.out(Tensor grad_output, Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_max_pool2d_out + operator_name: mkldnn_max_pool2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_max_pool2d.out(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_max_pool2d_backward_out + operator_name: mkldnn_max_pool2d_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_max_pool2d_backward.out(Tensor grad_output, Tensor output, Tensor input, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_max_pool3d_out + operator_name: mkldnn_max_pool3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_max_pool3d.out(Tensor self, int[3] kernel_size, int[3] stride=[], int[3] padding=0, int[3] dilation=1, bool ceil_mode=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_max_pool3d_backward_out + operator_name: mkldnn_max_pool3d_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_max_pool3d_backward.out(Tensor grad_output, Tensor output, Tensor input, int[3] kernel_size, int[3] stride=[], int[3] padding=0, int[3] dilation=1, bool ceil_mode=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantized_max_pool1d_out + operator_name: quantized_max_pool1d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_max_pool1d.out(Tensor self, int[1] kernel_size, int[1] stride=[], int[1] padding=0, int[1] dilation=1, bool ceil_mode=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 1 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 1 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantized_max_pool2d_out + operator_name: quantized_max_pool2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_max_pool2d.out(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0, int[2] dilation=1, bool ceil_mode=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantized_max_pool3d_out + operator_name: quantized_max_pool3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantized_max_pool3d.out(Tensor self, int[3] kernel_size, int[3] stride=[], int[3] padding=0, int[3] dilation=1, bool ceil_mode=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: ceil_mode + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: median_out + operator_name: median + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::median.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: nanmedian_out + operator_name: nanmedian + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::nanmedian.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _mps_convolution_out + operator_name: _mps_convolution + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_mps_convolution.out(Tensor self, Tensor weight, Tensor? bias, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mps_convolution_backward_out + operator_name: mps_convolution_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mps_convolution_backward.out(Tensor self, Tensor grad_output, Tensor weight, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool[3] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_convolution_out + operator_name: mkldnn_convolution + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_convolution.out(Tensor self, Tensor weight, Tensor? bias, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_rnn_layer_out + operator_name: mkldnn_rnn_layer + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_rnn_layer.out(Tensor input, Tensor weight0, Tensor weight1, Tensor weight2, Tensor weight3, Tensor hx_, Tensor cx_, bool reverse, int[] batch_sizes, int mode, int hidden_size, int num_layers, bool has_biases, bool bidirectional, bool batch_first, bool train, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight0 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx_ + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: reverse + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, at::IntArrayRef, int64_t, int64_t, int64_t, bool, bool, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight0 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx_ + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: reverse + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_rnn_layer_backward_out + operator_name: mkldnn_rnn_layer_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_rnn_layer_backward.out(Tensor input, Tensor weight1, Tensor weight2, Tensor weight3, Tensor weight4, Tensor hx_, Tensor cx_tmp, Tensor output, Tensor hy_, Tensor cy_, Tensor? grad_output, Tensor? grad_hy, Tensor? grad_cy, bool reverse, int mode, int hidden_size, int num_layers, bool has_biases, bool train, bool bidirectional, int[] batch_sizes, bool batch_first, Tensor workspace, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3, Tensor(e!) out4, Tensor(f!) out5, Tensor(g!) out6) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!), Tensor(e!), Tensor(f!), Tensor(g!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out4 + output: true + type: at::Tensor & + - allocate: true + annotation: f! + dynamic_type: at::Tensor + is_nullable: false + name: out5 + output: true + type: at::Tensor & + - allocate: true + annotation: g! + dynamic_type: at::Tensor + is_nullable: false + name: out6 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight4 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx_tmp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hy_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: reverse + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, int64_t, int64_t, int64_t, bool, bool, bool, at::IntArrayRef, bool, const at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight4 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx_tmp + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hy_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy_ + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: reverse + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out4 + output: true + type: at::Tensor & + - allocate: true + annotation: f! + dynamic_type: at::Tensor + is_nullable: false + name: out5 + output: true + type: at::Tensor & + - allocate: true + annotation: g! + dynamic_type: at::Tensor + is_nullable: false + name: out6 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out4 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out5 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out6 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_batch_norm_out + operator_name: miopen_batch_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_batch_norm.out(Tensor input, Tensor weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, bool training, float exponential_average_factor, float epsilon, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: exponential_average_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, double, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: exponential_average_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_batch_norm_backward_out + operator_name: miopen_batch_norm_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_batch_norm_backward.out(Tensor input, Tensor grad_output, Tensor weight, Tensor? running_mean, Tensor? running_var, Tensor? save_mean, Tensor? save_var, float epsilon, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, double, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: epsilon + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_convolution_out + operator_name: miopen_convolution + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_convolution.out(Tensor self, Tensor weight, Tensor? bias, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool benchmark, bool deterministic, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_convolution_transpose_out + operator_name: miopen_convolution_transpose + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_convolution_transpose.out(Tensor self, Tensor weight, Tensor? bias, SymInt[] padding, SymInt[] output_padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool benchmark, bool deterministic, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_depthwise_convolution_out + operator_name: miopen_depthwise_convolution + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_depthwise_convolution.out(Tensor self, Tensor weight, Tensor? bias, SymInt[] padding, SymInt[] stride, SymInt[] dilation, SymInt groups, bool benchmark, bool deterministic, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: benchmark + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: deterministic + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_rnn_out + operator_name: miopen_rnn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_rnn.out(Tensor input, Tensor[] weight, int weight_stride0, Tensor hx, Tensor? cx, int mode, int hidden_size, int num_layers, bool batch_first, float dropout, bool train, bool bidirectional, int[] batch_sizes, Tensor? dropout_state, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3, Tensor(e!) out4) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!), Tensor(e!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out4 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::TensorList, int64_t, const at::Tensor &, const ::std::optional &, int64_t, int64_t, int64_t, bool, double, bool, bool, at::IntArrayRef, const ::std::optional &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out4 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out4 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: miopen_rnn_backward_out + operator_name: miopen_rnn_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::miopen_rnn_backward.out(Tensor input, Tensor[] weight, int weight_stride0, Tensor weight_buf, Tensor hx, Tensor? cx, Tensor output, Tensor? grad_output, Tensor? grad_hy, Tensor? grad_cy, int mode, int hidden_size, int num_layers, bool batch_first, float dropout, bool train, bool bidirectional, int[] batch_sizes, Tensor? dropout_state, Tensor reserve, bool[4] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!)[] out3) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: out3 + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_buf + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserve + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: void (const at::Tensor &, at::TensorList, int64_t, const at::Tensor &, const at::Tensor &, const ::std::optional &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, int64_t, int64_t, int64_t, bool, double, bool, bool, at::IntArrayRef, const ::std::optional &, const at::Tensor &, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weight + type: at::TensorList + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: weight_stride0 + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight_buf + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: cx + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_output + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: mode + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: hidden_size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: batch_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: dropout_state + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: reserve + type: const at::Tensor & + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: out3 + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_sparse_matmul_out + operator_name: _sparse_sparse_matmul + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_sparse_matmul.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mul_out + operator_name: mul + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mul.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _native_batch_norm_legit_functional + operator_name: _native_batch_norm_legit_functional + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_native_batch_norm_legit_functional(Tensor input, Tensor? weight, Tensor? bias, Tensor running_mean, Tensor running_var, bool training, float momentum, float eps) -> (Tensor, Tensor, Tensor, Tensor running_mean_out, Tensor running_var_out) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, bool, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + field_name: running_mean_out + name: running_mean_out + type: at::Tensor + - dynamic_type: at::Tensor + field_name: running_var_out + name: running_var_out + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _native_batch_norm_legit_no_training_out + operator_name: _native_batch_norm_legit_no_training + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_native_batch_norm_legit_no_training.out(Tensor input, Tensor? weight, Tensor? bias, Tensor running_mean, Tensor running_var, float momentum, float eps, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, double, double, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_stats_out + operator_name: batch_norm_stats + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_stats.out(Tensor input, float eps, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, double, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_gather_stats_out + operator_name: batch_norm_gather_stats + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_gather_stats.out(Tensor input, Tensor mean, Tensor invstd, Tensor? running_mean, Tensor? running_var, float momentum, float eps, int count, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: count + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, double, double, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: count + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_gather_stats_with_counts_out + operator_name: batch_norm_gather_stats_with_counts + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_gather_stats_with_counts.out(Tensor input, Tensor mean, Tensor invstd, Tensor? running_mean, Tensor? running_var, float momentum, float eps, Tensor counts, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: counts + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, double, double, const at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: counts + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_batch_norm_backward_out + operator_name: native_batch_norm_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_batch_norm_backward.out(Tensor grad_out, Tensor input, Tensor? weight, Tensor? running_mean, Tensor? running_var, Tensor? save_mean, Tensor? save_invstd, bool train, float eps, bool[3] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_invstd + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, bool, double, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: save_invstd + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_backward_reduce_out + operator_name: batch_norm_backward_reduce + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_backward_reduce.out(Tensor grad_out, Tensor input, Tensor mean, Tensor invstd, Tensor? weight, bool input_g, bool weight_g, bool bias_g, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: input_g + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: weight_g + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bias_g + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, bool, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: bool + is_nullable: false + name: input_g + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: weight_g + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bias_g + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_backward_elemt_out + operator_name: batch_norm_backward_elemt + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_backward_elemt.out(Tensor grad_out, Tensor input, Tensor mean, Tensor invstd, Tensor? weight, Tensor sum_dy, Tensor sum_dy_xmu, Tensor count, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sum_dy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sum_dy_xmu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: count + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_out + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: invstd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sum_dy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: sum_dy_xmu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: count + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: batch_norm_update_stats_out + operator_name: batch_norm_update_stats + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::batch_norm_update_stats.out(Tensor input, Tensor? running_mean, Tensor? running_var, float momentum, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, double, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nnpack_spatial_convolution_out + operator_name: _nnpack_spatial_convolution + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nnpack_spatial_convolution.out(Tensor input, Tensor weight, Tensor? bias, SymInt[2] padding, SymInt[2] stride=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: ones_out + operator_name: ones + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ones.names_out(int[] size, *, Dimname[]? names, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: ones_like_out + operator_name: ones_like + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ones_like.out(Tensor self, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _euclidean_dist_out + operator_name: _euclidean_dist + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_euclidean_dist.out(Tensor x1, Tensor x2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cdist_forward_out + operator_name: _cdist_forward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cdist_forward.out(Tensor x1, Tensor x2, float p, int? compute_mode, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: compute_mode + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, double, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: compute_mode + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cdist_backward_out + operator_name: _cdist_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cdist_backward.out(Tensor grad, Tensor x1, Tensor x2, float p, Tensor cdist, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cdist + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, double, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x2 + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cdist + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _pdist_forward_out + operator_name: _pdist_forward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pdist_forward.out(Tensor self, float p=2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: p + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _pdist_backward_out + operator_name: _pdist_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pdist_backward.out(Tensor grad, Tensor self, float p, Tensor pdist, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pdist + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, double, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: pdist + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: pixel_shuffle_out + operator_name: pixel_shuffle + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::pixel_shuffle.out(Tensor self, int upscale_factor, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: upscale_factor + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: upscale_factor + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: pixel_unshuffle_out + operator_name: pixel_unshuffle + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::pixel_unshuffle.out(Tensor self, int downscale_factor, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: downscale_factor + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: downscale_factor + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: channel_shuffle_out + operator_name: channel_shuffle + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::channel_shuffle.out(Tensor self, SymInt groups, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _pin_memory_out + operator_name: _pin_memory + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pin_memory.out(Tensor self, Device? device=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: scalar_tensor_out + operator_name: scalar_tensor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::scalar_tensor.out(Scalar s, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: s + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: s + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand_out + operator_name: rand + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand.names_out(SymInt[] size, *, Dimname[]? names, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand_out + operator_name: rand + overload_name: generator_with_names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand.generator_with_names_out(SymInt[] size, *, Generator? generator, Dimname[]? names, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand_like_out + operator_name: rand_like + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand_like.out(Tensor self, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rand_like_out + operator_name: rand_like + overload_name: generator_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rand_like.generator_out(Tensor self, *, Generator? generator, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like_out + operator_name: randint_like + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.out(Tensor self, SymInt high, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like_out + operator_name: randint_like + overload_name: generator_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.generator_out(Tensor self, SymInt high, *, Generator? generator, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like_out + operator_name: randint_like + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.Tensor_out(Tensor self, Tensor high, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: high + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: high + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like_out + operator_name: randint_like + overload_name: Tensor_generator_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.Tensor_generator_out(Tensor self, Tensor high, *, Generator? generator, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: high + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: high + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like_out + operator_name: randint_like + overload_name: low_dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.low_dtype_out(Tensor self, SymInt low, SymInt high, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randint_like_out + operator_name: randint_like + overload_name: low_generator_dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randint_like.low_generator_dtype_out(Tensor self, SymInt low, SymInt high, *, Generator? generator, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: low + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: high + type: int64_t + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randn_out + operator_name: randn + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn.names_out(SymInt[] size, *, Dimname[]? names, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randn_out + operator_name: randn + overload_name: generator_with_names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn.generator_with_names_out(SymInt[] size, *, Generator? generator, Dimname[]? names, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randn_like_out + operator_name: randn_like + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn_like.out(Tensor self, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: randn_like_out + operator_name: randn_like + overload_name: generator_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::randn_like.generator_out(Tensor self, *, Generator? generator, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: repeat_out + operator_name: repeat + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::repeat.out(Tensor self, SymInt[] repeats, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: repeats + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: repeats + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: repeat_interleave_out + operator_name: repeat_interleave + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::repeat_interleave.Tensor_out(Tensor repeats, *, SymInt? output_size=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: repeats + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: output_size + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: repeats + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: output_size + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _mkldnn_reshape_out + operator_name: _mkldnn_reshape + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_mkldnn_reshape.out(Tensor self, int[] shape, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shape + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shape + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: relu_out + operator_name: relu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::relu.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: select_backward_out + operator_name: select_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::select_backward.out(Tensor grad_output, SymInt[] input_sizes, int dim, SymInt index, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: celu_out + operator_name: celu + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::celu.out(Tensor self, Scalar alpha=1.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1.0 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1.0 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: slice_backward_out + operator_name: slice_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::slice_backward.out(Tensor grad_output, SymInt[] input_sizes, int dim, SymInt start, SymInt end, SymInt step, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: end + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: start + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: end + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: slice_scatter_out + operator_name: slice_scatter + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::slice_scatter.out(Tensor self, Tensor src, int dim=0, SymInt? start=None, SymInt? end=None, SymInt step=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, ::std::optional, ::std::optional, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: select_scatter_out + operator_name: select_scatter + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::select_scatter.out(Tensor self, Tensor src, int dim, SymInt index, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: diagonal_scatter_out + operator_name: diagonal_scatter + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::diagonal_scatter.out(Tensor self, Tensor src, int offset=0, int dim1=0, int dim2=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: as_strided_scatter_out + operator_name: as_strided_scatter + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::as_strided_scatter.out(Tensor self, Tensor src, SymInt[] size, SymInt[] stride, SymInt? storage_offset=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: unsafe_split_out + operator_name: unsafe_split + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::unsafe_split.Tensor_out(Tensor self, SymInt split_size, int dim=0, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: split_size + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: void (const at::Tensor &, int64_t, int64_t, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: split_size + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: unsafe_split_with_sizes_out + operator_name: unsafe_split_with_sizes + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::unsafe_split_with_sizes.out(Tensor self, SymInt[] split_sizes, int dim=0, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_sizes + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: void (const at::Tensor &, at::IntArrayRef, int64_t, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: split_sizes + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: sum_out + operator_name: sum + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sum.out(Tensor self, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: std_mean_out + operator_name: std_mean + overload_name: correction_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::std_mean.correction_out(Tensor self, int[1]? dim=None, *, Scalar? correction=None, bool keepdim=False, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::OptionalIntArrayRef, const ::std::optional &, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: prod_out + operator_name: prod + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::prod.out(Tensor self, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _mkldnn_transpose_out + operator_name: _mkldnn_transpose + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_mkldnn_transpose.out(Tensor self, int dim0, int dim1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: flip_out + operator_name: flip + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::flip.out(Tensor self, int[] dims, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: roll_out + operator_name: roll + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::roll.out(Tensor self, SymInt[1] shifts, int[1] dims=[], *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shifts + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + size: 1 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shifts + size: 1 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + size: 1 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rot90_out + operator_name: rot90 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rot90.out(Tensor self, int k=1, int[] dims=[0,1], *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: '{0,1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: k + type: int64_t + - annotation: null + default: '{0,1}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _transform_bias_rescale_qkv_out + operator_name: _transform_bias_rescale_qkv + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_transform_bias_rescale_qkv.out(Tensor qkv, Tensor qkv_bias, int num_heads, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_heads + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_heads + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_from_mask_out + operator_name: _nested_tensor_from_mask + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_from_mask.out(Tensor t, Tensor mask, bool mask_check=True, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: t + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: mask_check + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: t + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: mask_check + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_from_padded_out + operator_name: _nested_from_padded + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_from_padded.out(Tensor padded, Tensor cpu_nested_shape_example, bool fuse_transform_0213=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: padded + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cpu_nested_shape_example + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: fuse_transform_0213 + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: padded + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cpu_nested_shape_example + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: fuse_transform_0213 + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_size_out + operator_name: _nested_tensor_size + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_size.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_strides_out + operator_name: _nested_tensor_strides + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_strides.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_storage_offsets_out + operator_name: _nested_tensor_storage_offsets + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_storage_offsets.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_from_padded_and_nested_example_out + operator_name: _nested_from_padded_and_nested_example + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_from_padded_and_nested_example.out(Tensor padded, Tensor nt_example, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: padded + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nt_example + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: padded + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nt_example + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_view_from_buffer_copy_out + operator_name: _nested_view_from_buffer_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_view_from_buffer_copy.out(Tensor self, Tensor nested_size, Tensor nested_strides, Tensor offsets, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_strides + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_size + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: nested_strides + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_view_from_jagged_copy_out + operator_name: _nested_view_from_jagged_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_view_from_jagged_copy.out(Tensor self, Tensor offsets, Tensor dummy, Tensor? lengths=None, int ragged_idx=1, Tensor? min_seqlen=None, Tensor? max_seqlen=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: lengths + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: ragged_idx + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min_seqlen + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max_seqlen + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, int64_t, const ::std::optional &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dummy + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: lengths + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: ragged_idx + type: int64_t + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: min_seqlen + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: max_seqlen + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_get_values_copy_out + operator_name: _nested_get_values_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_get_values_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _trilinear_out + operator_name: _trilinear + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_trilinear.out(Tensor i1, Tensor i2, Tensor i3, int[] expand1, int[] expand2, int[] expand3, int[] sumdim, int unroll_dim=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sumdim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: unroll_dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: i3 + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand1 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: expand3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: sumdim + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: unroll_dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _unique_out + operator_name: _unique + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unique.out(Tensor self, bool sorted=True, bool return_inverse=False, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: unique_dim_out + operator_name: unique_dim + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::unique_dim.out(Tensor self, int dim, bool sorted=True, bool return_inverse=False, bool return_counts=False, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: unique_consecutive_out + operator_name: unique_consecutive + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::unique_consecutive.out(Tensor self, bool return_inverse=False, bool return_counts=False, int? dim=None, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, ::std::optional, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: unique_dim_consecutive_out + operator_name: unique_dim_consecutive + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::unique_dim_consecutive.out(Tensor self, int dim, bool return_inverse=False, bool return_counts=False, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, int64_t, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _unique2_out + operator_name: _unique2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unique2.out(Tensor self, bool sorted=True, bool return_inverse=False, bool return_counts=False, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, bool, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: sorted + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_inverse + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: return_counts + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _unsafe_view_out + operator_name: _unsafe_view + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_unsafe_view.out(Tensor self, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: var_mean_out + operator_name: var_mean + overload_name: correction_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::var_mean.correction_out(Tensor self, int[1]? dim=None, *, Scalar? correction=None, bool keepdim=False, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::OptionalIntArrayRef, const ::std::optional &, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: dim + size: 1 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: correction + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: keepdim + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _weight_norm_interface_out + operator_name: _weight_norm_interface + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_weight_norm_interface.out(Tensor v, Tensor g, int dim=0, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: g + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: g + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _weight_norm_interface_backward_out + operator_name: _weight_norm_interface_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_weight_norm_interface_backward.out(Tensor grad_w, Tensor saved_v, Tensor saved_g, Tensor saved_norms, int dim, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_w + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_g + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_norms + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_w + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_v + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_g + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: saved_norms + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: zeros_out + operator_name: zeros + overload_name: names_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::zeros.names_out(int[] size, *, Dimname[]? names, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::DimnameList + is_nullable: true + kwarg_only: true + name: names + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _efficientzerotensor_out + operator_name: _efficientzerotensor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_efficientzerotensor.out(SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: zeros_like_out + operator_name: zeros_like + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::zeros_like.out(Tensor self, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _standard_gamma_grad_out + operator_name: _standard_gamma_grad + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_standard_gamma_grad.out(Tensor self, Tensor output, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _standard_gamma_out + operator_name: _standard_gamma + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_standard_gamma.out(Tensor self, Generator? generator=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _dirichlet_grad_out + operator_name: _dirichlet_grad + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_dirichlet_grad.out(Tensor x, Tensor alpha, Tensor total, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: alpha + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: alpha + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: total + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sample_dirichlet_out + operator_name: _sample_dirichlet + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sample_dirichlet.out(Tensor self, Generator? generator=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: poisson_out + operator_name: poisson + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::poisson.out(Tensor self, Generator? generator=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: binomial_out + operator_name: binomial + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::binomial.out(Tensor count, Tensor prob, Generator? generator=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: count + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: prob + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: count + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: prob + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_norm_out + operator_name: native_norm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_norm.out(Tensor self, Scalar p=2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: native_norm_out + operator_name: native_norm + overload_name: ScalarOpt_dim_dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::native_norm.ScalarOpt_dim_dtype_out(Tensor self, Scalar? p, int[1] dim, bool keepdim, ScalarType? dtype, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, at::IntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _batch_norm_with_update_functional + operator_name: _batch_norm_with_update_functional + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_batch_norm_with_update_functional(Tensor input, Tensor? weight, Tensor? bias, Tensor running_mean, Tensor running_var, float momentum, float eps) -> (Tensor, Tensor, Tensor, Tensor, Tensor running_mean_out, Tensor running_var_out) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, double, double) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_mean + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_var + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + name: result1 + type: at::Tensor + - dynamic_type: at::Tensor + name: result2 + type: at::Tensor + - dynamic_type: at::Tensor + name: result3 + type: at::Tensor + - dynamic_type: at::Tensor + field_name: running_mean_out + name: running_mean_out + type: at::Tensor + - dynamic_type: at::Tensor + field_name: running_var_out + name: running_var_out + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _batch_norm_no_update_out + operator_name: _batch_norm_no_update + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_batch_norm_no_update.out(Tensor input, Tensor? weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, float momentum, float eps, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const ::std::optional &, const ::std::optional &, const ::std::optional &, const ::std::optional &, double, double, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: weight + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_mean + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: running_var + type: const ::std::optional & + - annotation: null + dynamic_type: double + is_nullable: false + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_sum_out + operator_name: _sparse_sum + overload_name: dim_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_sum.dim_out(Tensor self, int[1] dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_sum_backward_out + operator_name: _sparse_sum_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_sum_backward.out(Tensor grad, Tensor self, int[] dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_csr_sum_out + operator_name: _sparse_csr_sum + overload_name: dim_dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_csr_sum.dim_dtype_out(Tensor self, int[1] dim, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_csr_prod_out + operator_name: _sparse_csr_prod + overload_name: dim_dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_csr_prod.dim_dtype_out(Tensor self, int[1] dim, bool keepdim=False, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + size: 1 + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: keepdim + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + kwarg_only: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_softmax_out + operator_name: _sparse_softmax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_softmax.out(Tensor self, int dim, bool half_to_float, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_softmax_backward_data_out + operator_name: _sparse_softmax_backward_data + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_softmax_backward_data.out(Tensor grad_output, Tensor output, int dim, Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_log_softmax_out + operator_name: _sparse_log_softmax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_log_softmax.out(Tensor self, int dim, bool half_to_float, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: bool + is_nullable: false + name: half_to_float + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_log_softmax_backward_data_out + operator_name: _sparse_log_softmax_backward_data + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_log_softmax_backward_data.out(Tensor grad_output, Tensor output, int dim, Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _spdiags_out + operator_name: _spdiags + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_spdiags.out(Tensor diagonals, Tensor offsets, int[] shape, Layout? layout=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: diagonals + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shape + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + name: layout + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: diagonals + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: offsets + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: shape + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + name: layout + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: norm_out + operator_name: norm + overload_name: ScalarOpt_dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.ScalarOpt_dtype_out(Tensor self, Scalar? p, *, ScalarType dtype, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const ::std::optional &, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: true + name: p + type: const ::std::optional & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + kwarg_only: true + name: dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: norm_out + operator_name: norm + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::norm.Scalar_out(Tensor self, Scalar p=2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: clone_out + operator_name: clone + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::clone.out(Tensor self, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: resize_as_out + operator_name: resize_as + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::resize_as.out(Tensor self, Tensor the_template, *, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: const at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: resize_as + operator_name: resize_as + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::resize_as(Tensor self, Tensor the_template, *, MemoryFormat? memory_format=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: resize_as_sparse_out + operator_name: resize_as_sparse + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::resize_as_sparse.out(Tensor self, Tensor the_template, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: const at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: resize_as_sparse + operator_name: resize_as_sparse + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::resize_as_sparse(Tensor self, Tensor the_template) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: the_template + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: zero_out + operator_name: zero + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::zero.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: zero + operator_name: zero + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::zero(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: sub_out + operator_name: sub + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sub.Scalar_out(Tensor self, Scalar other, Scalar alpha=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rsub_out + operator_name: rsub + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rsub.Tensor_out(Tensor self, Tensor other, *, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rsub_out + operator_name: rsub + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rsub.Scalar_out(Tensor self, Scalar other, Scalar alpha=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_addmm_out + operator_name: _sparse_addmm + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_addmm.out(Tensor self, Tensor mat1, Tensor mat2, *, Scalar beta=1, Scalar alpha=1, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mat2 + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: beta + type: const at::Scalar & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_coo_tensor_out + operator_name: sparse_coo_tensor + overload_name: size_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_coo_tensor.size_out(int[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_coo_tensor_with_dims_out + operator_name: _sparse_coo_tensor_with_dims + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_coo_tensor_with_dims.out(int sparse_dim, int dense_dim, int[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (int64_t, int64_t, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_coo_tensor_with_dims_and_tensors_out + operator_name: _sparse_coo_tensor_with_dims_and_tensors + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_coo_tensor_with_dims_and_tensors.out(int sparse_dim, int dense_dim, SymInt[] size, Tensor indices, Tensor values, *, bool? is_coalesced=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: is_coalesced + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (int64_t, int64_t, at::IntArrayRef, const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: indices + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + kwarg_only: true + name: is_coalesced + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_resize_out + operator_name: sparse_resize + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_resize.out(Tensor self, int[] size, int sparse_dim, int dense_dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: const at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_resize + operator_name: sparse_resize + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_resize(Tensor self, int[] size, int sparse_dim, int dense_dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_resize_and_clear_out + operator_name: sparse_resize_and_clear + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_resize_and_clear.out(Tensor self, int[] size, int sparse_dim, int dense_dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + schema_order_cpp_signature: const at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, int64_t, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: const at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_resize_and_clear + operator_name: sparse_resize_and_clear + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_resize_and_clear(Tensor self, int[] size, int sparse_dim, int dense_dim) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::IntArrayRef, int64_t, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dense_dim + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: sparse_mask_out + operator_name: sparse_mask + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::sparse_mask.out(Tensor self, Tensor mask, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_mask_projection_out + operator_name: _sparse_mask_projection + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_mask_projection.out(Tensor self, Tensor mask, bool accumulate_matches=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate_matches + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate_matches + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _to_dense_out + operator_name: _to_dense + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_dense.out(Tensor self, ScalarType? dtype=None, bool? masked_grad=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: masked_grad + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: masked_grad + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _coalesce_out + operator_name: _coalesce + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_coalesce.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _coalesced_out + operator_name: _coalesced + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_coalesced.out(Tensor self, bool coalesced, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: coalesced + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: coalesced + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _coalesced + operator_name: _coalesced + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_coalesced(Tensor self, bool coalesced) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: coalesced + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: coalesced + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: copy_sparse_to_sparse_out + operator_name: copy_sparse_to_sparse + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::copy_sparse_to_sparse.out(Tensor self, Tensor src, bool non_blocking=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: copy_sparse_to_sparse + operator_name: copy_sparse_to_sparse + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::copy_sparse_to_sparse(Tensor self, Tensor src, bool non_blocking=False) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _to_sparse_out + operator_name: _to_sparse + overload_name: sparse_dim_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse.sparse_dim_out(Tensor self, int sparse_dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: sparse_dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _to_sparse_out + operator_name: _to_sparse + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse.out(Tensor self, *, Layout? layout=None, int[2]? blocksize=None, int? dense_dim=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + kwarg_only: true + name: blocksize + size: 2 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::OptionalIntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + kwarg_only: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + kwarg_only: true + name: blocksize + size: 2 + type: at::OptionalIntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + kwarg_only: true + name: dense_dim + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _to_sparse_csr_out + operator_name: _to_sparse_csr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse_csr.out(Tensor self, int? dense_dim=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _to_sparse_csc_out + operator_name: _to_sparse_csc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse_csc.out(Tensor self, int? dense_dim=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _to_sparse_bsr_out + operator_name: _to_sparse_bsr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse_bsr.out(Tensor self, int[2] blocksize, int? dense_dim=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _to_sparse_bsc_out + operator_name: _to_sparse_bsc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_sparse_bsc.out(Tensor self, int[2] blocksize, int? dense_dim=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: blocksize + size: 2 + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dense_dim + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_mkldnn_out + operator_name: to_mkldnn + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_mkldnn.out(Tensor self, ScalarType? dtype=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_reorder_conv2d_weight_out + operator_name: mkldnn_reorder_conv2d_weight + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_reorder_conv2d_weight.out(Tensor self, SymInt[2] padding=0, SymInt[2] stride=1, SymInt[2] dilation=1, SymInt groups=1, SymInt[]? input_size=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: input_size + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::OptionalIntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: input_size + type: at::OptionalIntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_reorder_conv3d_weight_out + operator_name: mkldnn_reorder_conv3d_weight + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_reorder_conv3d_weight.out(Tensor self, SymInt[3] padding=0, SymInt[3] stride=1, SymInt[3] dilation=1, SymInt groups=1, SymInt[]? input_size=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: input_size + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, int64_t, at::OptionalIntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: groups + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: input_size + type: at::OptionalIntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantize_per_tensor_dynamic_out + operator_name: quantize_per_tensor_dynamic + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantize_per_tensor_dynamic.out(Tensor self, ScalarType dtype, bool reduce_range, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - annotation: null + dynamic_type: bool + is_nullable: false + name: reduce_range + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::ScalarType, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - annotation: null + dynamic_type: bool + is_nullable: false + name: reduce_range + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantize_per_tensor_out + operator_name: quantize_per_tensor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantize_per_tensor.out(Tensor self, float scale, int zero_point, ScalarType dtype, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, int64_t, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantize_per_tensor_out + operator_name: quantize_per_tensor + overload_name: tensor_qparams_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantize_per_tensor.tensor_qparams_out(Tensor self, Tensor scale, Tensor zero_point, ScalarType dtype, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantize_per_tensor_out + operator_name: quantize_per_tensor + overload_name: tensors_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantize_per_tensor.tensors_out(Tensor[] tensors, Tensor scales, Tensor zero_points, ScalarType dtype, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: void (at::TensorList, const at::Tensor &, const at::Tensor &, at::ScalarType, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: quantize_per_channel_out + operator_name: quantize_per_channel + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::quantize_per_channel.out(Tensor self, Tensor scales, Tensor zero_points, int axis, ScalarType dtype, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scales + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_points + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: dequantize_out + operator_name: dequantize + overload_name: self_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::dequantize.self_out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: dequantize_out + operator_name: dequantize + overload_name: tensors_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::dequantize.tensors_out(Tensor[] tensors, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: q_per_channel_scales_out + operator_name: q_per_channel_scales + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::q_per_channel_scales.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: q_per_channel_zero_points_out + operator_name: q_per_channel_zero_points + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::q_per_channel_zero_points.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: int_repr_out + operator_name: int_repr + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::int_repr.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _make_per_tensor_quantized_tensor_out + operator_name: _make_per_tensor_quantized_tensor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_make_per_tensor_quantized_tensor.out(Tensor self, float scale, int zero_point, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _make_per_channel_quantized_tensor_out + operator_name: _make_per_channel_quantized_tensor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_make_per_channel_quantized_tensor.out(Tensor self, Tensor scale, Tensor zero_point, int axis, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: fake_quantize_per_tensor_affine_cachemask_out + operator_name: fake_quantize_per_tensor_affine_cachemask + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fake_quantize_per_tensor_affine_cachemask.out(Tensor self, float scale, int zero_point, int quant_min, int quant_max, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, double, int64_t, int64_t, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: zero_point + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fake_quantize_per_tensor_affine_cachemask_tensor_qparams_out + operator_name: _fake_quantize_per_tensor_affine_cachemask_tensor_qparams + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fake_quantize_per_tensor_affine_cachemask_tensor_qparams.out(Tensor self, Tensor scale, Tensor zero_point, Tensor fake_quant_enabled, int quant_min, int quant_max, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_enabled + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_enabled + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fake_quantize_learnable_per_tensor_affine_out + operator_name: _fake_quantize_learnable_per_tensor_affine + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fake_quantize_learnable_per_tensor_affine.out(Tensor self, Tensor scale, Tensor zero_point, int quant_min, int quant_max, float grad_factor=1.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: fake_quantize_per_channel_affine_cachemask_out + operator_name: fake_quantize_per_channel_affine_cachemask + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::fake_quantize_per_channel_affine_cachemask.out(Tensor self, Tensor scale, Tensor zero_point, int axis, int quant_min, int quant_max, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, int64_t, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fake_quantize_learnable_per_channel_affine_out + operator_name: _fake_quantize_learnable_per_channel_affine + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fake_quantize_learnable_per_channel_affine.out(Tensor self, Tensor scale, Tensor zero_point, int axis, int quant_min, int quant_max, float grad_factor=1.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, int64_t, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: axis + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + default: 1.0 + dynamic_type: double + is_nullable: false + name: grad_factor + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_moving_avg_obs_fq_helper_out + operator_name: _fused_moving_avg_obs_fq_helper + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_moving_avg_obs_fq_helper.out(Tensor self, Tensor observer_on, Tensor fake_quant_on, Tensor(a!) running_min, Tensor(b!) running_max, Tensor(c!) scale, Tensor(d!) zero_point, float averaging_const, int quant_min, int quant_max, int ch_axis, bool per_row_fake_quant=False, bool symmetric_quant=False, *, Tensor(e!) out0, Tensor(f!) out1) -> (Tensor(e!), Tensor(f!)) + arguments: + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: f! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: observer_on + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_on + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_min + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_max + type: at::Tensor & + - annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: at::Tensor & + - annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: averaging_const + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ch_axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: per_row_fake_quant + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: symmetric_quant + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, double, int64_t, int64_t, int64_t, bool, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: observer_on + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_on + type: const at::Tensor & + - annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: running_min + type: at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: running_max + type: at::Tensor & + - annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: at::Tensor & + - annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: averaging_const + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ch_axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: per_row_fake_quant + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: symmetric_quant + type: bool + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: f! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_moving_avg_obs_fq_helper_functional + operator_name: _fused_moving_avg_obs_fq_helper_functional + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_moving_avg_obs_fq_helper_functional(Tensor self, Tensor observer_on, Tensor fake_quant_on, Tensor running_min, Tensor running_max, Tensor scale, Tensor zero_point, float averaging_const, int quant_min, int quant_max, int ch_axis, bool per_row_fake_quant=False, bool symmetric_quant=False) -> (Tensor output, Tensor mask, Tensor running_min_out, Tensor running_max_out, Tensor scale_out, Tensor zero_point_out) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: observer_on + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_on + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_min + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_max + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: averaging_const + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ch_axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: per_row_fake_quant + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: symmetric_quant + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, double, int64_t, int64_t, int64_t, bool, bool) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: observer_on + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: fake_quant_on + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_min + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: running_max + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scale + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: zero_point + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: averaging_const + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_min + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: quant_max + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: ch_axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: per_row_fake_quant + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: symmetric_quant + type: bool + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + field_name: output + name: output + type: at::Tensor + - dynamic_type: at::Tensor + field_name: mask + name: mask + type: at::Tensor + - dynamic_type: at::Tensor + field_name: running_min_out + name: running_min_out + type: at::Tensor + - dynamic_type: at::Tensor + field_name: running_max_out + name: running_max_out + type: at::Tensor + - dynamic_type: at::Tensor + field_name: scale_out + name: scale_out + type: at::Tensor + - dynamic_type: at::Tensor + field_name: zero_point_out + name: zero_point_out + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _to_copy_out + operator_name: _to_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_to_copy.out(Tensor self, *, bool non_blocking=False, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: non_blocking + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, bool, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: non_blocking + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::MemoryFormat + is_nullable: true + kwarg_only: true + name: memory_format + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _lstm_mps_out + operator_name: _lstm_mps + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_lstm_mps.out(Tensor input, Tensor[] hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional, bool batch_first, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3, Tensor(e!) out4, Tensor(f!) out5) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!), Tensor(e!), Tensor(f!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out4 + output: true + type: at::Tensor & + - allocate: true + annotation: f! + dynamic_type: at::Tensor + is_nullable: false + name: out5 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, at::TensorList, at::TensorList, bool, int64_t, double, bool, bool, bool, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out4 + output: true + type: at::Tensor & + - allocate: true + annotation: f! + dynamic_type: at::Tensor + is_nullable: false + name: out5 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out4 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out5 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: lstm_mps_backward_out + operator_name: lstm_mps_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lstm_mps_backward.out(Tensor? grad_y, Tensor? grad_hy, Tensor? grad_cy, Tensor z_state, Tensor cell_state_fwd, Tensor input, Tensor layersOutputs, Tensor[] hx, Tensor[] params, bool has_biases, int num_layers, float dropout, bool train, bool bidirectional, bool batch_first, *, Tensor(a!) out0, Tensor(b!)[] out1, Tensor(c!)[] out2) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: out1 + output: true + type: at::TensorList + - allocate: true + annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: out2 + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_y + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: z_state + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cell_state_fwd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: layersOutputs + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: void (const ::std::optional &, const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, at::TensorList, at::TensorList, bool, int64_t, double, bool, bool, bool, at::Tensor &, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_y + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: z_state + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cell_state_fwd + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: layersOutputs + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: hx + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: params + type: at::TensorList + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_biases + type: bool + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_layers + type: int64_t + - annotation: null + dynamic_type: double + is_nullable: false + name: dropout + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + name: train + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: bidirectional + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: out1 + output: true + type: at::TensorList + - allocate: true + annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: out2 + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _thnn_fused_lstm_cell_out + operator_name: _thnn_fused_lstm_cell + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_fused_lstm_cell.out(Tensor input_gates, Tensor hidden_gates, Tensor cx, Tensor? input_bias=None, Tensor? hidden_bias=None, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _thnn_fused_lstm_cell_backward_impl_out + operator_name: _thnn_fused_lstm_cell_backward_impl + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_fused_lstm_cell_backward_impl.out(Tensor? grad_hy, Tensor? grad_cy, Tensor cx, Tensor cy, Tensor workspace, bool has_bias, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_bias + type: bool + schema_order_cpp_signature: ::std::tuple (const ::std::optional &, const ::std::optional &, const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_hy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: grad_cy + type: const ::std::optional & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cx + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: cy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_bias + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _thnn_fused_gru_cell_out + operator_name: _thnn_fused_gru_cell + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_fused_gru_cell.out(Tensor input_gates, Tensor hidden_gates, Tensor hx, Tensor? input_bias=None, Tensor? hidden_bias=None, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, const ::std::optional &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hidden_gates + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: hx + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: input_bias + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: hidden_bias + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _thnn_fused_gru_cell_backward_out + operator_name: _thnn_fused_gru_cell_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_thnn_fused_gru_cell_backward.out(Tensor grad_hy, Tensor workspace, bool has_bias, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2, Tensor(d!) out3, Tensor(e!) out4) -> (Tensor(a!), Tensor(b!), Tensor(c!), Tensor(d!), Tensor(e!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out4 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_hy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_bias + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_hy + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: workspace + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: has_bias + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - allocate: true + annotation: d! + dynamic_type: at::Tensor + is_nullable: false + name: out3 + output: true + type: at::Tensor & + - allocate: true + annotation: e! + dynamic_type: at::Tensor + is_nullable: false + name: out4 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out3 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out4 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _pack_padded_sequence_out + operator_name: _pack_padded_sequence + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_pack_padded_sequence.out(Tensor input, Tensor lengths, bool batch_first, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: lengths + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, bool, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: lengths + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: batch_first + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set_out + operator_name: set + overload_name: source_Storage_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::set.source_Storage_out(Tensor self, Storage source, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Storage, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set + operator_name: set + overload_name: source_Storage + manual_kernel_registration: false + category_override: '' + schema_string: aten::set.source_Storage(Tensor self, Storage source) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Storage) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set_out + operator_name: set + overload_name: source_Storage_storage_offset_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::set.source_Storage_storage_offset_out(Tensor self, Storage source, SymInt storage_offset, SymInt[] size, SymInt[] stride=[], *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: storage_offset + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Storage, int64_t, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: storage_offset + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set + operator_name: set + overload_name: source_Storage_storage_offset + manual_kernel_registration: false + category_override: '' + schema_string: aten::set.source_Storage_storage_offset(Tensor self, Storage source, SymInt storage_offset, SymInt[] size, SymInt[] stride=[]) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: storage_offset + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor (const at::Tensor &, at::Storage, int64_t, at::IntArrayRef, at::IntArrayRef) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Storage + is_nullable: false + name: source + type: at::Storage + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: storage_offset + type: int64_t + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set_out + operator_name: set + overload_name: source_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::set.source_Tensor_out(Tensor self, Tensor source, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set + operator_name: set + overload_name: source_Tensor + manual_kernel_registration: false + category_override: '' + schema_string: aten::set.source_Tensor(Tensor self, Tensor source) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set_out + operator_name: set + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::set.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: set + operator_name: set + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::set(Tensor self) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor (const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: lift_out + operator_name: lift + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lift.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: lift_fresh_copy_out + operator_name: lift_fresh_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::lift_fresh_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_fill_out + operator_name: masked_fill + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_fill.Scalar_out(Tensor self, Tensor mask, Scalar value, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_fill_out + operator_name: masked_fill + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_fill.Tensor_out(Tensor self, Tensor mask, Tensor value, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: masked_scatter_out + operator_name: masked_scatter + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::masked_scatter.out(Tensor self, Tensor mask, Tensor source, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _masked_softmax_out + operator_name: _masked_softmax + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_masked_softmax.out(Tensor self, Tensor mask, int? dim=None, int? mask_type=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _masked_softmax_backward_out + operator_name: _masked_softmax_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_masked_softmax_backward.out(Tensor grad_output, Tensor output, Tensor mask, int? dim=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: mask + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: dim + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: put_out + operator_name: put + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::put.out(Tensor self, Tensor index, Tensor source, bool accumulate=False, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: source + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: accumulate + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_fill_out + operator_name: index_fill + overload_name: int_Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_fill.int_Scalar_out(Tensor self, int dim, Tensor index, Scalar value, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: index_fill_out + operator_name: index_fill + overload_name: int_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::index_fill.int_Tensor_out(Tensor self, int dim, Tensor index, Tensor value, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: index + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_and_out + operator_name: bitwise_and + overload_name: Scalar_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_and.Scalar_Tensor_out(Scalar self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_or_out + operator_name: bitwise_or + overload_name: Scalar_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_or.Scalar_Tensor_out(Scalar self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_xor_out + operator_name: bitwise_xor + overload_name: Scalar_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_xor.Scalar_Tensor_out(Scalar self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: __lshift___out + operator_name: __lshift__ + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::__lshift__.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: __lshift___out + operator_name: __lshift__ + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::__lshift__.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_left_shift_out + operator_name: bitwise_left_shift + overload_name: Scalar_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_left_shift.Scalar_Tensor_out(Scalar self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: __rshift___out + operator_name: __rshift__ + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::__rshift__.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: other + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: __rshift___out + operator_name: __rshift__ + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::__rshift__.Tensor_out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bitwise_right_shift_out + operator_name: bitwise_right_shift + overload_name: Scalar_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bitwise_right_shift.Scalar_Tensor_out(Scalar self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: random_out + operator_name: random + overload_name: from_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::random.from_out(Tensor self, int from, int? to, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: from + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: to + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: from + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: to + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: random + operator_name: random + overload_name: from + manual_kernel_registration: false + category_override: '' + schema_string: aten::random.from(Tensor self, int from, int? to, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: from + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: to + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: from + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: true + name: to + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: random_out + operator_name: random + overload_name: to_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::random.to_out(Tensor self, int to, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: to + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: to + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: random + operator_name: random + overload_name: to + manual_kernel_registration: false + category_override: '' + schema_string: aten::random.to(Tensor self, int to, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: to + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, int64_t, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: to + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: random_out + operator_name: random + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::random.out(Tensor self, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: random + operator_name: random + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::random(Tensor self, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: uniform_out + operator_name: uniform + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::uniform.out(Tensor self, float from=0, float to=1, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: from + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: to + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, double, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: from + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: to + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: uniform + operator_name: uniform + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::uniform(Tensor self, float from=0, float to=1, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: from + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: to + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: from + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: to + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: cauchy_out + operator_name: cauchy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::cauchy.out(Tensor self, float median=0, float sigma=1, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: median + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: sigma + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, double, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: median + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: sigma + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: cauchy + operator_name: cauchy + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::cauchy(Tensor self, float median=0, float sigma=1, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: median + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: sigma + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: median + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: sigma + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_normal_out + operator_name: log_normal + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_normal.out(Tensor self, float mean=1, float std=2, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, double, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: log_normal + operator_name: log_normal + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::log_normal(Tensor self, float mean=1, float std=2, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 2 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: exponential_out + operator_name: exponential + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::exponential.out(Tensor self, float lambd=1, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: lambd + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: lambd + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: exponential + operator_name: exponential + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::exponential(Tensor self, float lambd=1, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: lambd + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: lambd + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: geometric_out + operator_name: geometric + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::geometric.out(Tensor self, float p, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: geometric + operator_name: geometric + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::geometric(Tensor self, float p, *, Generator? generator=None) -> Tensor + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor (const at::Tensor &, double, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: p + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: tril_indices_out + operator_name: tril_indices + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::tril_indices.out(int row, int col, int offset=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: row + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: col + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + schema_order_cpp_signature: at::Tensor & (int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: row + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: col + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: triu_indices_out + operator_name: triu_indices + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::triu_indices.out(int row, int col, int offset=0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: row + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: col + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + schema_order_cpp_signature: at::Tensor & (int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: row + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: col + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: trace_out + operator_name: trace + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::trace.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _cholesky_solve_helper_out + operator_name: _cholesky_solve_helper + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_cholesky_solve_helper.out(Tensor self, Tensor A, bool upper, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: upper + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: A + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: upper + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: dist_out + operator_name: dist + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::dist.out(Tensor self, Tensor other, Scalar p=2, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Scalar &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: p + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _histogramdd_bin_edges_out + operator_name: _histogramdd_bin_edges + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_histogramdd_bin_edges.out(Tensor self, int[] bins, *, float[]? range=None, Tensor? weight=None, bool density=False, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: bins + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + schema_order_cpp_signature: void (const at::Tensor &, at::IntArrayRef, ::std::optional>, const ::std::optional &, bool, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: bins + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _histogramdd_from_bin_cts_out + operator_name: _histogramdd_from_bin_cts + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_histogramdd_from_bin_cts.out(Tensor self, int[] bins, *, float[]? range=None, Tensor? weight=None, bool density=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: bins + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, ::std::optional>, const ::std::optional &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: bins + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: at::ArrayRef + is_nullable: true + kwarg_only: true + name: range + type: ::std::optional> + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _histogramdd_from_bin_tensors_out + operator_name: _histogramdd_from_bin_tensors + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_histogramdd_from_bin_tensors.out(Tensor self, Tensor[] bins, *, Tensor? weight=None, bool density=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: bins + type: at::TensorList + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::TensorList, const ::std::optional &, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: bins + type: at::TensorList + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: weight + type: const ::std::optional & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: density + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: remainder_out + operator_name: remainder + overload_name: Scalar_Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::remainder.Scalar_Tensor_out(Scalar self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: unfold_backward_out + operator_name: unfold_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::unfold_backward.out(Tensor grad_in, SymInt[] input_sizes, int dim, int size, int step, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_in + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_in + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: input_sizes + type: at::IntArrayRef + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: normal_out + operator_name: normal + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::normal.out(Tensor self, float mean=0, float std=1, *, Generator? generator=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, double, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: double + is_nullable: false + name: mean + type: double + - annotation: null + default: 1 + dynamic_type: double + is_nullable: false + name: std + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + kwarg_only: true + name: generator + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _amp_foreach_non_finite_check_and_unscale_out + operator_name: _amp_foreach_non_finite_check_and_unscale + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_amp_foreach_non_finite_check_and_unscale.out(Tensor[] self, Tensor(b!) found_inf, Tensor inv_scale, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: inv_scale + type: const at::Tensor & + schema_order_cpp_signature: void (at::TensorList, at::Tensor &, const at::Tensor &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: inv_scale + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _amp_foreach_non_finite_check_and_unscale + operator_name: _amp_foreach_non_finite_check_and_unscale + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_amp_foreach_non_finite_check_and_unscale(Tensor[] self, Tensor found_inf, Tensor inv_scale) -> (Tensor[] self_out, Tensor found_inf_out) + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: inv_scale + type: const at::Tensor & + schema_order_cpp_signature: ::std::tuple<::std::vector,at::Tensor> (at::TensorList, const at::Tensor &, const at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: inv_scale + type: const at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + - dynamic_type: at::Tensor + field_name: found_inf_out + name: found_inf_out + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _amp_update_scale_out + operator_name: _amp_update_scale + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_amp_update_scale.out(Tensor self, Tensor(b!) growth_tracker, Tensor found_inf, float scale_growth_factor, float scale_backoff_factor, int growth_interval, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: growth_tracker + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_growth_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_backoff_factor + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: growth_interval + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &, const at::Tensor &, double, double, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: growth_tracker + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_growth_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_backoff_factor + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: growth_interval + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _amp_update_scale + operator_name: _amp_update_scale + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_amp_update_scale(Tensor self, Tensor growth_tracker, Tensor found_inf, float scale_growth_factor, float scale_backoff_factor, int growth_interval) -> (Tensor, Tensor growth_tracker_out) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: growth_tracker + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_growth_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_backoff_factor + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: growth_interval + type: int64_t + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, double, double, int64_t) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: growth_tracker + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: found_inf + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_growth_factor + type: double + - annotation: null + dynamic_type: double + is_nullable: false + name: scale_backoff_factor + type: double + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: growth_interval + type: int64_t + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + field_name: growth_tracker_out + name: growth_tracker_out + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add_out + operator_name: _foreach_add + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add.Scalar_out(Tensor[] self, Scalar scalar, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add_out + operator_name: _foreach_add + overload_name: List_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add.List_out(Tensor[] self, Tensor[] other, *, Scalar alpha=1, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add_out + operator_name: _foreach_add + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add.ScalarList_out(Tensor[] self, Scalar[] scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_add_out + operator_name: _foreach_add + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_add.Tensor_out(Tensor[] self, Tensor other, *, Scalar alpha=1, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Tensor &, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sub_out + operator_name: _foreach_sub + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sub.Scalar_out(Tensor[] self, Scalar scalar, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sub_out + operator_name: _foreach_sub + overload_name: List_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sub.List_out(Tensor[] self, Tensor[] other, *, Scalar alpha=1, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + kwarg_only: true + name: alpha + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sub_out + operator_name: _foreach_sub + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sub.ScalarList_out(Tensor[] self, Scalar[] scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul_out + operator_name: _foreach_mul + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul.Scalar_out(Tensor[] self, Scalar scalar, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul_out + operator_name: _foreach_mul + overload_name: List_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul.List_out(Tensor[] self, Tensor[] other, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul_out + operator_name: _foreach_mul + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul.ScalarList_out(Tensor[] self, Scalar[] scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_mul_out + operator_name: _foreach_mul + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_mul.Tensor_out(Tensor[] self, Tensor other, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: void (at::TensorList, const at::Tensor &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div_out + operator_name: _foreach_div + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div.Scalar_out(Tensor[] self, Scalar scalar, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div_out + operator_name: _foreach_div + overload_name: List_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div.List_out(Tensor[] self, Tensor[] other, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div_out + operator_name: _foreach_div + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div.ScalarList_out(Tensor[] self, Scalar[] scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_div_out + operator_name: _foreach_div + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_div.Tensor_out(Tensor[] self, Tensor other, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + schema_order_cpp_signature: void (at::TensorList, const at::Tensor &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: other + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_max_out + operator_name: _foreach_clamp_max + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_max.Scalar_out(Tensor[] self, Scalar scalar, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_max_out + operator_name: _foreach_clamp_max + overload_name: List_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_max.List_out(Tensor[] self, Tensor[] other, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_max_out + operator_name: _foreach_clamp_max + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_max.ScalarList_out(Tensor[] self, Scalar[] scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_min_out + operator_name: _foreach_clamp_min + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_min.Scalar_out(Tensor[] self, Scalar scalar, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_min_out + operator_name: _foreach_clamp_min + overload_name: List_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_min.List_out(Tensor[] self, Tensor[] other, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_clamp_min_out + operator_name: _foreach_clamp_min + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_clamp_min.ScalarList_out(Tensor[] self, Scalar[] scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_maximum_out + operator_name: _foreach_maximum + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_maximum.Scalar_out(Tensor[] self, Scalar scalar, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_maximum_out + operator_name: _foreach_maximum + overload_name: List_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_maximum.List_out(Tensor[] self, Tensor[] other, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_maximum_out + operator_name: _foreach_maximum + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_maximum.ScalarList_out(Tensor[] self, Scalar[] scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_minimum_out + operator_name: _foreach_minimum + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_minimum.Scalar_out(Tensor[] self, Scalar scalar, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: scalar + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_minimum_out + operator_name: _foreach_minimum + overload_name: List_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_minimum.List_out(Tensor[] self, Tensor[] other, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: other + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_minimum_out + operator_name: _foreach_minimum + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_minimum.ScalarList_out(Tensor[] self, Scalar[] scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcdiv_out + operator_name: _foreach_addcdiv + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcdiv.Scalar_out(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar value=1, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcdiv_out + operator_name: _foreach_addcdiv + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcdiv.ScalarList_out(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar[] scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcdiv_out + operator_name: _foreach_addcdiv + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcdiv.Tensor_out(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Tensor scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcmul_out + operator_name: _foreach_addcmul + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcmul.Scalar_out(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar value=1, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + default: 1 + dynamic_type: const at::Scalar & + is_nullable: false + name: value + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcmul_out + operator_name: _foreach_addcmul + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcmul.ScalarList_out(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Scalar[] scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: scalars + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_addcmul_out + operator_name: _foreach_addcmul + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_addcmul.Tensor_out(Tensor[] self, Tensor[] tensor1, Tensor[] tensor2, Tensor scalars, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensor2 + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: scalars + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_abs_out + operator_name: _foreach_abs + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_abs.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_acos_out + operator_name: _foreach_acos + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_acos.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_asin_out + operator_name: _foreach_asin + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_asin.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_atan_out + operator_name: _foreach_atan + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_atan.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_ceil_out + operator_name: _foreach_ceil + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_ceil.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_cos_out + operator_name: _foreach_cos + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_cos.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_cosh_out + operator_name: _foreach_cosh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_cosh.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_erf_out + operator_name: _foreach_erf + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_erf.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_erfc_out + operator_name: _foreach_erfc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_erfc.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_exp_out + operator_name: _foreach_exp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_exp.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_expm1_out + operator_name: _foreach_expm1 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_expm1.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_floor_out + operator_name: _foreach_floor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_floor.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_frac_out + operator_name: _foreach_frac + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_frac.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lerp_out + operator_name: _foreach_lerp + overload_name: List_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lerp.List_out(Tensor[] self, Tensor[] tensors1, Tensor[] weights, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weights + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: weights + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lerp_out + operator_name: _foreach_lerp + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lerp.Scalar_out(Tensor[] self, Tensor[] tensors1, Scalar weight, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: weight + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lerp_out + operator_name: _foreach_lerp + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lerp.ScalarList_out(Tensor[] self, Tensor[] tensors1, Scalar[] weight, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: weight + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: tensors1 + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: weight + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_lgamma_out + operator_name: _foreach_lgamma + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_lgamma.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log_out + operator_name: _foreach_log + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log10_out + operator_name: _foreach_log10 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log10.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log1p_out + operator_name: _foreach_log1p + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log1p.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_log2_out + operator_name: _foreach_log2 + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_log2.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_max_out + operator_name: _foreach_max + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_max.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_neg_out + operator_name: _foreach_neg + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_neg.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_norm_out + operator_name: _foreach_norm + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_norm.Scalar_out(Tensor[] self, Scalar ord=2, ScalarType? dtype=None, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &, ::std::optional, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + default: 2 + dynamic_type: const at::Scalar & + is_nullable: false + name: ord + type: const at::Scalar & + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_pow_out + operator_name: _foreach_pow + overload_name: List_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_pow.List_out(Tensor[] self, Tensor[] exponent, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exponent + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exponent + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_pow_out + operator_name: _foreach_pow + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_pow.Scalar_out(Tensor[] self, Scalar exponent, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + schema_order_cpp_signature: void (at::TensorList, const at::Scalar &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: exponent + type: const at::Scalar & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_pow_out + operator_name: _foreach_pow + overload_name: ScalarList_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_pow.ScalarList_out(Tensor[] self, Scalar[] exponent, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: exponent + type: at::ArrayRef + schema_order_cpp_signature: void (at::TensorList, at::ArrayRef, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: false + name: exponent + type: at::ArrayRef + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_reciprocal_out + operator_name: _foreach_reciprocal + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_reciprocal.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_round_out + operator_name: _foreach_round + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_round.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_rsqrt_out + operator_name: _foreach_rsqrt + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_rsqrt.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sigmoid_out + operator_name: _foreach_sigmoid + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sigmoid.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sign_out + operator_name: _foreach_sign + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sign.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sin_out + operator_name: _foreach_sin + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sin.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sinh_out + operator_name: _foreach_sinh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sinh.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_sqrt_out + operator_name: _foreach_sqrt + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_sqrt.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_tan_out + operator_name: _foreach_tan + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_tan.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_tanh_out + operator_name: _foreach_tanh + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_tanh.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_trunc_out + operator_name: _foreach_trunc + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_trunc.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_zero_out + operator_name: _foreach_zero + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_zero.out(Tensor[] self, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: void (at::TensorList, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_zero + operator_name: _foreach_zero + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_zero(Tensor[] self) -> Tensor[] self_out + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + schema_order_cpp_signature: ::std::vector (at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foreach_copy_out + operator_name: _foreach_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foreach_copy.out(Tensor[] self, Tensor[] src, bool non_blocking=False, *, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: src + type: at::TensorList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + schema_order_cpp_signature: void (at::TensorList, at::TensorList, bool, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: src + type: at::TensorList + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: non_blocking + type: bool + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: bucketize_out + operator_name: bucketize + overload_name: Scalar_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::bucketize.Scalar_out(Scalar self, Tensor boundaries, *, bool out_int32=False, bool right=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: boundaries + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Scalar &, const at::Tensor &, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: self + type: const at::Scalar & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: boundaries + type: const at::Tensor & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: out_int32 + type: bool + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: right + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: glu_jvp_out + operator_name: glu_jvp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::glu_jvp.out(Tensor glu, Tensor x, Tensor dx, int dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dx + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dx + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: glu_backward_jvp_out + operator_name: glu_backward_jvp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::glu_backward_jvp.out(Tensor grad_x, Tensor grad_glu, Tensor x, Tensor dgrad_glu, Tensor dx, int dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dgrad_glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dx + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: x + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dgrad_glu + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: dx + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: hardswish_backward_out + operator_name: hardswish_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::hardswish_backward.out(Tensor grad_output, Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rrelu_with_noise_functional + operator_name: rrelu_with_noise_functional + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::rrelu_with_noise_functional(Tensor self, Tensor noise, Scalar lower=0.125, Scalar upper=0.3333333333333333, bool training=False, Generator? generator=None) -> (Tensor, Tensor noise_out) + arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: const at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, bool, ::std::optional) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: const at::Tensor & + - annotation: null + default: 0.125 + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + default: 0.3333333333333333 + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: at::Generator + is_nullable: true + name: generator + type: ::std::optional + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: result0 + type: at::Tensor + - dynamic_type: at::Tensor + field_name: noise_out + name: noise_out + type: at::Tensor + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: rrelu_with_noise_backward_out + operator_name: rrelu_with_noise_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::rrelu_with_noise_backward.out(Tensor grad_output, Tensor self, Tensor noise, Scalar lower, Scalar upper, bool training, bool self_is_result, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: self_is_result + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Scalar &, const at::Scalar &, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: noise + type: const at::Tensor & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: lower + type: const at::Scalar & + - annotation: null + dynamic_type: const at::Scalar & + is_nullable: false + name: upper + type: const at::Scalar & + - annotation: null + dynamic_type: bool + is_nullable: false + name: training + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: self_is_result + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: mkldnn_adaptive_avg_pool2d_backward_out + operator_name: mkldnn_adaptive_avg_pool2d_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::mkldnn_adaptive_avg_pool2d_backward.out(Tensor grad_output, Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _adaptive_avg_pool2d_out + operator_name: _adaptive_avg_pool2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_adaptive_avg_pool2d.out(Tensor self, SymInt[2] output_size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _adaptive_avg_pool2d_backward_out + operator_name: _adaptive_avg_pool2d_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_adaptive_avg_pool2d_backward.out(Tensor grad_output, Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _adaptive_avg_pool3d_out + operator_name: _adaptive_avg_pool3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_adaptive_avg_pool3d.out(Tensor self, SymInt[3] output_size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: output_size + size: 3 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _adaptive_avg_pool3d_backward_out + operator_name: _adaptive_avg_pool3d_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_adaptive_avg_pool3d_backward.out(Tensor grad_output, Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_bilinear2d_out + operator_name: upsample_bilinear2d + overload_name: vec_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_bilinear2d.vec_out(Tensor input, SymInt[]? output_size, bool align_corners, float[]? scale_factors, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, bool, ::std::optional>, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: bool + is_nullable: false + name: align_corners + type: bool + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: upsample_nearest2d_out + operator_name: upsample_nearest2d + overload_name: vec_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::upsample_nearest2d.vec_out(Tensor input, SymInt[]? output_size, float[]? scale_factors, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, ::std::optional>, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: input + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: scale_factors + type: ::std::optional> + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: false + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _slow_conv2d_backward_out + operator_name: _slow_conv2d_backward + overload_name: output_mask_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_slow_conv2d_backward.output_mask_out(Tensor grad_output, Tensor self, Tensor weight, SymInt[2] kernel_size, SymInt[2] stride, SymInt[2] padding, bool[3] output_mask, *, Tensor(a!) out0, Tensor(b!) out1, Tensor(c!) out2) -> (Tensor(a!), Tensor(b!), Tensor(c!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, ::std::array, at::Tensor &, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad_output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + dynamic_type: ::std::array + is_nullable: false + name: output_mask + type: ::std::array + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - allocate: true + annotation: c! + dynamic_type: at::Tensor + is_nullable: false + name: out2 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out2 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: conv_depthwise3d_out + operator_name: conv_depthwise3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::conv_depthwise3d.out(Tensor self, Tensor weight, SymInt[3] kernel_size, Tensor? bias, SymInt[3] stride, SymInt[3] padding, SymInt[3] dilation, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: slow_conv_dilated2d_out + operator_name: slow_conv_dilated2d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv_dilated2d.out(Tensor self, Tensor weight, SymInt[2] kernel_size, Tensor? bias=None, SymInt[2] stride=1, SymInt[2] padding=0, SymInt[2] dilation=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 2 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 2 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 2 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 2 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: slow_conv_dilated3d_out + operator_name: slow_conv_dilated3d + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::slow_conv_dilated3d.out(Tensor self, Tensor weight, SymInt[3] kernel_size, Tensor? bias=None, SymInt[3] stride=1, SymInt[3] padding=0, SymInt[3] dilation=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, at::IntArrayRef, const ::std::optional &, at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: kernel_size + size: 3 + type: at::IntArrayRef + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: bias + type: const ::std::optional & + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + size: 3 + type: at::IntArrayRef + - annotation: null + default: 0 + dynamic_type: at::IntArrayRef + is_nullable: false + name: padding + size: 3 + type: at::IntArrayRef + - annotation: null + default: 1 + dynamic_type: at::IntArrayRef + is_nullable: false + name: dilation + size: 3 + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: isinf_out + operator_name: isinf + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::isinf.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: linalg_matrix_exp_out + operator_name: linalg_matrix_exp + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::linalg_matrix_exp.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_optional_intlist_out + operator_name: _test_optional_intlist + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_optional_intlist.out(Tensor values, int[]? addends, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: addends + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: addends + type: at::OptionalIntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_optional_filled_intlist_out + operator_name: _test_optional_filled_intlist + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_optional_filled_intlist.out(Tensor values, int[2]? addends, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: addends + size: 2 + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::OptionalIntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: true + name: addends + size: 2 + type: at::OptionalIntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_optional_floatlist_out + operator_name: _test_optional_floatlist + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_optional_floatlist.out(Tensor values, float[]? addends, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: addends + type: ::std::optional> + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, ::std::optional>, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: values + type: const at::Tensor & + - annotation: null + dynamic_type: at::ArrayRef + is_nullable: true + name: addends + type: ::std::optional> + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_warn_in_autograd_out + operator_name: _test_warn_in_autograd + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_warn_in_autograd.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_autograd_multiple_dispatch_out + operator_name: _test_autograd_multiple_dispatch + overload_name: fullcoverage_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_autograd_multiple_dispatch.fullcoverage_out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _test_autograd_multiple_dispatch_view_copy_out + operator_name: _test_autograd_multiple_dispatch_view_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_test_autograd_multiple_dispatch_view_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: segment_reduce_out + operator_name: segment_reduce + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::segment_reduce.out(Tensor data, str reduce, *, Tensor? lengths=None, Tensor? indices=None, Tensor? offsets=None, int axis=0, bool unsafe=False, Scalar? initial=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: lengths + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: indices + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: offsets + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: unsafe + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: initial + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, c10::string_view, const ::std::optional &, const ::std::optional &, const ::std::optional &, int64_t, bool, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: lengths + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: indices + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: offsets + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: unsafe + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: initial + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _segment_reduce_backward_out + operator_name: _segment_reduce_backward + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_segment_reduce_backward.out(Tensor grad, Tensor output, Tensor data, str reduce, *, Tensor? lengths=None, Tensor? offsets=None, int axis=0, Scalar? initial=None, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: lengths + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: offsets + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: initial + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, c10::string_view, const ::std::optional &, const ::std::optional &, int64_t, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: grad + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: output + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: data + type: const at::Tensor & + - annotation: null + dynamic_type: c10::string_view + is_nullable: false + name: reduce + type: c10::string_view + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: lengths + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: offsets + type: const ::std::optional & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + kwarg_only: true + name: axis + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: const at::Scalar & + is_nullable: true + kwarg_only: true + name: initial + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _nested_tensor_from_tensor_list_out + operator_name: _nested_tensor_from_tensor_list + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_nested_tensor_from_tensor_list.out(Tensor[] list, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: list + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: pin_memory + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (at::TensorList, ::std::optional, ::std::optional, ::std::optional, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: list + type: at::TensorList + - annotation: null + default: ::std::nullopt + dynamic_type: at::ScalarType + is_nullable: true + name: dtype + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Layout + is_nullable: true + name: layout + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: at::Device + is_nullable: true + name: device + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: bool + is_nullable: true + name: pin_memory + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fw_primal_copy_out + operator_name: _fw_primal_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fw_primal_copy.out(Tensor self, int level, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _make_dual_copy_out + operator_name: _make_dual_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_make_dual_copy.out(Tensor primal, Tensor tangent, int level, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: primal + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tangent + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: primal + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: tangent + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: level + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: view_as_real_copy_out + operator_name: view_as_real_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_as_real_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: view_as_complex_copy_out + operator_name: view_as_complex_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_as_complex_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _conj_copy_out + operator_name: _conj_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_conj_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _neg_view_copy_out + operator_name: _neg_view_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_neg_view_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: as_strided_copy_out + operator_name: as_strided_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::as_strided_copy.out(Tensor self, SymInt[] size, SymInt[] stride, SymInt? storage_offset=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: storage_offset + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _sparse_broadcast_to_copy_out + operator_name: _sparse_broadcast_to_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_sparse_broadcast_to_copy.out(Tensor self, int[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: diagonal_copy_out + operator_name: diagonal_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::diagonal_copy.out(Tensor self, int offset=0, int dim1=0, int dim2=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: offset + type: int64_t + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: dim2 + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: expand_copy_out + operator_name: expand_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::expand_copy.out(Tensor self, SymInt[] size, *, bool implicit=False, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: implicit + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + default: false + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: implicit + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: permute_copy_out + operator_name: permute_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::permute_copy.out(Tensor self, int[] dims, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dims + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _reshape_alias_copy_out + operator_name: _reshape_alias_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_reshape_alias_copy.out(Tensor self, SymInt[] size, SymInt[] stride, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: stride + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: select_copy_out + operator_name: select_copy + overload_name: int_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::select_copy.int_out(Tensor self, int dim, SymInt index, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: index + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: detach_copy_out + operator_name: detach_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::detach_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: slice_copy_out + operator_name: slice_copy + overload_name: Tensor_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::slice_copy.Tensor_out(Tensor self, int dim=0, SymInt? start=None, SymInt? end=None, SymInt step=1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, ::std::optional, ::std::optional, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: 0 + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: start + type: ::std::optional + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: end + type: ::std::optional + - annotation: null + default: 1 + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze_copy_out + operator_name: squeeze_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze_copy_out + operator_name: squeeze_copy + overload_name: dim_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze_copy.dim_out(Tensor self, int dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: squeeze_copy_out + operator_name: squeeze_copy + overload_name: dims_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::squeeze_copy.dims_out(Tensor self, int[] dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: dim + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: t_copy_out + operator_name: t_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::t_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: transpose_copy_out + operator_name: transpose_copy + overload_name: int_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::transpose_copy.int_out(Tensor self, int dim0, int dim1, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim0 + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim1 + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: unsqueeze_copy_out + operator_name: unsqueeze_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::unsqueeze_copy.out(Tensor self, int dim, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dim + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _indices_copy_out + operator_name: _indices_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_indices_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _values_copy_out + operator_name: _values_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_values_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: indices_copy_out + operator_name: indices_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::indices_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: values_copy_out + operator_name: values_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::values_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: crow_indices_copy_out + operator_name: crow_indices_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::crow_indices_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: col_indices_copy_out + operator_name: col_indices_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::col_indices_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: ccol_indices_copy_out + operator_name: ccol_indices_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::ccol_indices_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: row_indices_copy_out + operator_name: row_indices_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::row_indices_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: view_copy_out + operator_name: view_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_copy.out(Tensor self, SymInt[] size, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::IntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::IntArrayRef + is_nullable: false + name: size + type: at::IntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: view_copy_out + operator_name: view_copy + overload_name: dtype_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::view_copy.dtype_out(Tensor self, ScalarType dtype, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::ScalarType, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: at::ScalarType + is_nullable: false + name: dtype + type: at::ScalarType + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: unfold_copy_out + operator_name: unfold_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::unfold_copy.out(Tensor self, int dimension, int size, int step, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, int64_t, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: dimension + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: size + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: step + type: int64_t + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: alias_copy_out + operator_name: alias_copy + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::alias_copy.out(Tensor self, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: to_padded_tensor_out + operator_name: to_padded_tensor + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::to_padded_tensor.out(Tensor self, float padding, SymInt[]? output_size=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: padding + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, double, at::OptionalIntArrayRef, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + name: padding + type: double + - annotation: null + default: ::std::nullopt + dynamic_type: at::IntArrayRef + is_nullable: true + name: output_size + type: at::OptionalIntArrayRef + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _transformer_encoder_layer_fwd_out + operator_name: _transformer_encoder_layer_fwd + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_transformer_encoder_layer_fwd.out(Tensor src, int embed_dim, int num_heads, Tensor qkv_weight, Tensor qkv_bias, Tensor proj_weight, Tensor proj_bias, bool use_gelu, bool norm_first, float eps, Tensor norm_weight_1, Tensor norm_bias_1, Tensor norm_weight_2, Tensor norm_bias_2, Tensor ffn_weight_1, Tensor ffn_bias_1, Tensor ffn_weight_2, Tensor ffn_bias_2, Tensor? mask=None, int? mask_type=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_heads + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: use_gelu + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: norm_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_weight_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_bias_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_weight_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_bias_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_weight_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_bias_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_weight_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_bias_2 + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, int64_t, int64_t, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, bool, double, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, ::std::optional, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: src + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_heads + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + dynamic_type: bool + is_nullable: false + name: use_gelu + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + name: norm_first + type: bool + - annotation: null + dynamic_type: double + is_nullable: false + name: eps + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_weight_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_bias_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_weight_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: norm_bias_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_weight_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_bias_1 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_weight_2 + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: ffn_bias_2 + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _native_multi_head_attention_out + operator_name: _native_multi_head_attention + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_native_multi_head_attention.out(Tensor query, Tensor key, Tensor value, int embed_dim, int num_head, Tensor qkv_weight, Tensor qkv_bias, Tensor proj_weight, Tensor proj_bias, Tensor? mask=None, bool need_weights=True, bool average_attn_weights=True, int? mask_type=None, *, Tensor(a!) out0, Tensor(b!) out1) -> (Tensor(a!), Tensor(b!)) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_head + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: need_weights + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: average_attn_weights + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + schema_order_cpp_signature: ::std::tuple (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, bool, bool, ::std::optional, at::Tensor &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_head + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: need_weights + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: average_attn_weights + type: bool + - annotation: null + default: ::std::nullopt + dynamic_type: int64_t + is_nullable: true + name: mask_type + type: ::std::optional + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out0 + output: true + type: at::Tensor & + - allocate: true + annotation: b! + dynamic_type: at::Tensor + is_nullable: false + name: out1 + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out0 + type: at::Tensor & + - dynamic_type: at::Tensor + name: out1 + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _triton_scaled_dot_attention_out + operator_name: _triton_scaled_dot_attention + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_triton_scaled_dot_attention.out(Tensor q, Tensor k, Tensor v, float dropout_p=0.0, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: k + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, double, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: q + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: k + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: v + type: const at::Tensor & + - annotation: null + default: 0.0 + dynamic_type: double + is_nullable: false + name: dropout_p + type: double + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _triton_multi_head_attention_out + operator_name: _triton_multi_head_attention + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_triton_multi_head_attention.out(Tensor query, Tensor key, Tensor value, int embed_dim, int num_head, Tensor qkv_weight, Tensor qkv_bias, Tensor proj_weight, Tensor proj_bias, Tensor? mask=None, *, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_head + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, int64_t, int64_t, const at::Tensor &, const at::Tensor &, const at::Tensor &, const at::Tensor &, const ::std::optional &, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: query + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: key + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: value + type: const at::Tensor & + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: embed_dim + type: int64_t + - annotation: null + dynamic_type: int64_t + is_nullable: false + name: num_head + type: int64_t + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: qkv_bias + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_weight + type: const at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: proj_bias + type: const at::Tensor & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + name: mask + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _foobar_out + operator_name: _foobar + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_foobar.out(Tensor self, bool arg1=True, bool arg2=True, *, bool arg3=True, Tensor(a!) out) -> Tensor(a!) + arguments: + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: arg1 + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: arg2 + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: arg3 + type: bool + schema_order_cpp_signature: at::Tensor & (const at::Tensor &, bool, bool, bool, at::Tensor &) + schema_order_arguments: + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + name: self + type: const at::Tensor & + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: arg1 + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + name: arg2 + type: bool + - annotation: null + default: true + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: arg3 + type: bool + - allocate: true + annotation: a! + dynamic_type: at::Tensor + is_nullable: false + name: out + output: true + type: at::Tensor & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::Tensor + name: out + type: at::Tensor & + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adam_out + operator_name: _fused_adam + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adam.out(Tensor[] self, Tensor(b!)[] grads, Tensor(c!)[] exp_avgs, Tensor(d!)[] exp_avg_sqs, Tensor(e!)[] max_exp_avg_sqs, Tensor[] state_steps, *, float lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, double, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adam + operator_name: _fused_adam + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adam(Tensor[] self, Tensor[] grads, Tensor[] exp_avgs, Tensor[] exp_avg_sqs, Tensor[] max_exp_avg_sqs, Tensor[] state_steps, *, float lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> (Tensor[] self_out, Tensor[] grads_out, Tensor[] exp_avgs_out, Tensor[] exp_avg_sqs_out, Tensor[] max_exp_avg_sqs_out) + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple<::std::vector,::std::vector,::std::vector,::std::vector,::std::vector> (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, double, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: grads_out + name: grads_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: exp_avgs_out + name: exp_avgs_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: exp_avg_sqs_out + name: exp_avg_sqs_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: max_exp_avg_sqs_out + name: max_exp_avg_sqs_out + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adam_out + operator_name: _fused_adam + overload_name: tensor_lr_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adam.tensor_lr_out(Tensor[] self, Tensor(b!)[] grads, Tensor(c!)[] exp_avgs, Tensor(d!)[] exp_avg_sqs, Tensor(e!)[] max_exp_avg_sqs, Tensor[] state_steps, *, Tensor lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adam + operator_name: _fused_adam + overload_name: tensor_lr + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adam.tensor_lr(Tensor[] self, Tensor[] grads, Tensor[] exp_avgs, Tensor[] exp_avg_sqs, Tensor[] max_exp_avg_sqs, Tensor[] state_steps, *, Tensor lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> (Tensor[] self_out, Tensor[] grads_out, Tensor[] exp_avgs_out, Tensor[] exp_avg_sqs_out, Tensor[] max_exp_avg_sqs_out) + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple<::std::vector,::std::vector,::std::vector,::std::vector,::std::vector> (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: grads_out + name: grads_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: exp_avgs_out + name: exp_avgs_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: exp_avg_sqs_out + name: exp_avg_sqs_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: max_exp_avg_sqs_out + name: max_exp_avg_sqs_out + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adamw_out + operator_name: _fused_adamw + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adamw.out(Tensor[] self, Tensor(b!)[] grads, Tensor(c!)[] exp_avgs, Tensor(d!)[] exp_avg_sqs, Tensor(e!)[] max_exp_avg_sqs, Tensor[] state_steps, *, float lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, double, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adamw + operator_name: _fused_adamw + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adamw(Tensor[] self, Tensor[] grads, Tensor[] exp_avgs, Tensor[] exp_avg_sqs, Tensor[] max_exp_avg_sqs, Tensor[] state_steps, *, float lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> (Tensor[] self_out, Tensor[] grads_out, Tensor[] exp_avgs_out, Tensor[] exp_avg_sqs_out, Tensor[] max_exp_avg_sqs_out) + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple<::std::vector,::std::vector,::std::vector,::std::vector,::std::vector> (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, double, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: grads_out + name: grads_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: exp_avgs_out + name: exp_avgs_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: exp_avg_sqs_out + name: exp_avg_sqs_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: max_exp_avg_sqs_out + name: max_exp_avg_sqs_out + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adamw_out + operator_name: _fused_adamw + overload_name: tensor_lr_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adamw.tensor_lr_out(Tensor[] self, Tensor(b!)[] grads, Tensor(c!)[] exp_avgs, Tensor(d!)[] exp_avg_sqs, Tensor(e!)[] max_exp_avg_sqs, Tensor[] state_steps, *, Tensor lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: e! + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adamw + operator_name: _fused_adamw + overload_name: tensor_lr + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adamw.tensor_lr(Tensor[] self, Tensor[] grads, Tensor[] exp_avgs, Tensor[] exp_avg_sqs, Tensor[] max_exp_avg_sqs, Tensor[] state_steps, *, Tensor lr, float beta1, float beta2, float weight_decay, float eps, bool amsgrad, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> (Tensor[] self_out, Tensor[] grads_out, Tensor[] exp_avgs_out, Tensor[] exp_avg_sqs_out, Tensor[] max_exp_avg_sqs_out) + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple<::std::vector,::std::vector,::std::vector,::std::vector,::std::vector> (at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, double, double, double, double, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avgs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: max_exp_avg_sqs + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta1 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: beta2 + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: amsgrad + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: grads_out + name: grads_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: exp_avgs_out + name: exp_avgs_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: exp_avg_sqs_out + name: exp_avg_sqs_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: max_exp_avg_sqs_out + name: max_exp_avg_sqs_out + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_sgd_out + operator_name: _fused_sgd + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_sgd.out(Tensor[] self, Tensor(b!)[] grads, Tensor(c!)[] momentum_buffer_list, *, float weight_decay, float momentum, float lr, float dampening, bool nesterov, bool maximize, bool is_first_step, Tensor? grad_scale=None, Tensor? found_inf=None, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, double, double, double, double, bool, bool, bool, const ::std::optional &, const ::std::optional &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_sgd + operator_name: _fused_sgd + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_sgd(Tensor[] self, Tensor[] grads, Tensor[] momentum_buffer_list, *, float weight_decay, float momentum, float lr, float dampening, bool nesterov, bool maximize, bool is_first_step, Tensor? grad_scale=None, Tensor? found_inf=None) -> (Tensor[] self_out, Tensor[] grads_out, Tensor[] momentum_buffer_list_out) + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple<::std::vector,::std::vector,::std::vector> (at::TensorList, at::TensorList, at::TensorList, double, double, double, double, bool, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: grads_out + name: grads_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: momentum_buffer_list_out + name: momentum_buffer_list_out + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_sgd_out + operator_name: _fused_sgd + overload_name: tensor_lr_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_sgd.tensor_lr_out(Tensor[] self, Tensor(b!)[] grads, Tensor(c!)[] momentum_buffer_list, *, float weight_decay, float momentum, Tensor lr, float dampening, bool nesterov, bool maximize, bool is_first_step, Tensor? grad_scale=None, Tensor? found_inf=None, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, double, double, const at::Tensor &, double, bool, bool, bool, const ::std::optional &, const ::std::optional &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_sgd + operator_name: _fused_sgd + overload_name: tensor_lr + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_sgd.tensor_lr(Tensor[] self, Tensor[] grads, Tensor[] momentum_buffer_list, *, float weight_decay, float momentum, Tensor lr, float dampening, bool nesterov, bool maximize, bool is_first_step, Tensor? grad_scale=None, Tensor? found_inf=None) -> (Tensor[] self_out, Tensor[] grads_out, Tensor[] momentum_buffer_list_out) + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple<::std::vector,::std::vector,::std::vector> (at::TensorList, at::TensorList, at::TensorList, double, double, const at::Tensor &, double, bool, bool, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: momentum_buffer_list + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: momentum + type: double + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: dampening + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: nesterov + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: is_first_step + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: grads_out + name: grads_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: momentum_buffer_list_out + name: momentum_buffer_list_out + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adagrad_out + operator_name: _fused_adagrad + overload_name: out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adagrad.out(Tensor[] self, Tensor(b!)[] grads, Tensor(c!)[] state_sums, Tensor(d!)[] state_steps, *, float lr, float lr_decay, float weight_decay, float eps, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, double, double, double, double, bool, const ::std::optional &, const ::std::optional &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: d! + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adagrad + operator_name: _fused_adagrad + overload_name: '' + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adagrad(Tensor[] self, Tensor[] grads, Tensor[] state_sums, Tensor[] state_steps, *, float lr, float lr_decay, float weight_decay, float eps, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> (Tensor[] self_out, Tensor[] grads_out, Tensor[] state_sums_out, Tensor[] state_steps_out) + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple<::std::vector,::std::vector,::std::vector,::std::vector> (at::TensorList, at::TensorList, at::TensorList, at::TensorList, double, double, double, double, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: grads_out + name: grads_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: state_sums_out + name: state_sums_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: state_steps_out + name: state_steps_out + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adagrad_out + operator_name: _fused_adagrad + overload_name: tensor_lr_out + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adagrad.tensor_lr_out(Tensor[] self, Tensor(b!)[] grads, Tensor(c!)[] state_sums, Tensor[] state_steps, *, Tensor lr, float lr_decay, float weight_decay, float eps, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None, Tensor(a!)[] out) -> () + arguments: + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: void (at::TensorList, at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, double, double, double, bool, const ::std::optional &, const ::std::optional &, at::TensorList) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: b! + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: c! + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + - allocate: true + annotation: a! + dynamic_type: at::TensorList + is_nullable: false + name: out + output: true + type: at::TensorList + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: [] + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false +- name: _fused_adagrad + operator_name: _fused_adagrad + overload_name: tensor_lr + manual_kernel_registration: false + category_override: '' + schema_string: aten::_fused_adagrad.tensor_lr(Tensor[] self, Tensor[] grads, Tensor[] state_sums, Tensor[] state_steps, *, Tensor lr, float lr_decay, float weight_decay, float eps, bool maximize, Tensor? grad_scale=None, Tensor? found_inf=None) -> (Tensor[] self_out, Tensor[] grads_out, Tensor[] state_sums_out) + arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + schema_order_cpp_signature: ::std::tuple<::std::vector,::std::vector,::std::vector> (at::TensorList, at::TensorList, at::TensorList, at::TensorList, const at::Tensor &, double, double, double, bool, const ::std::optional &, const ::std::optional &) + schema_order_arguments: + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: self + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: grads + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_sums + type: at::TensorList + - annotation: null + dynamic_type: at::TensorList + is_nullable: false + name: state_steps + type: at::TensorList + - annotation: null + dynamic_type: at::Tensor + is_nullable: false + kwarg_only: true + name: lr + type: const at::Tensor & + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: lr_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: weight_decay + type: double + - annotation: null + dynamic_type: double + is_nullable: false + kwarg_only: true + name: eps + type: double + - annotation: null + dynamic_type: bool + is_nullable: false + kwarg_only: true + name: maximize + type: bool + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: grad_scale + type: const ::std::optional & + - annotation: null + default: '{}' + dynamic_type: at::Tensor + is_nullable: true + kwarg_only: true + name: found_inf + type: const ::std::optional & + method_of: + - Type + - namespace + mode: native + python_module: '' + returns: + - dynamic_type: at::TensorList + field_name: self_out + name: self_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: grads_out + name: grads_out + type: ::std::vector + - dynamic_type: at::TensorList + field_name: state_sums_out + name: state_sums_out + type: ::std::vector + inplace: false + is_factory_method: false + abstract: true + device_guard: false + with_gil: false + deprecated: false + has_math_kernel: false diff --git a/libtch/c-generated.go b/libtch/c-generated.go index b78eb9c..91b5856 100644 --- a/libtch/c-generated.go +++ b/libtch/c-generated.go @@ -177,6 +177,7 @@ cscaleBackoffFactor := *(*C.double)(unsafe.Pointer(&scaleBackoffFactor)) cgrowthInterval := *(*C.int64_t)(unsafe.Pointer(&growthInterval)) C.atg__amp_update_scale_out(ptr, out, self, growthTracker, foundInf, cscaleGrowthFactor, cscaleBackoffFactor, cgrowthInterval) } + func Atg_AutocastToFullPrecision(ptr *Ctensor, self Ctensor, cudaEnabled int32, cpuEnabled int32){ ccudaEnabled := *(*C.int)(unsafe.Pointer(&cudaEnabled)) ccpuEnabled := *(*C.int)(unsafe.Pointer(&cpuEnabled)) @@ -189,6 +190,31 @@ ccudaDtype := *(*C.int)(unsafe.Pointer(&cudaDtype)) ccpuDtype := *(*C.int)(unsafe.Pointer(&cpuDtype)) C.atg__autocast_to_reduced_precision(ptr, self, ccudaEnabled, ccpuEnabled, ccudaDtype, ccpuDtype) } +func Atg_BatchNormNoUpdate(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64){ +cmomentum := *(*C.double)(unsafe.Pointer(&momentum)) +ceps := *(*C.double)(unsafe.Pointer(&eps)) + C.atg__batch_norm_no_update(ptr, input, weight, bias, runningMean, runningVar, cmomentum, ceps) +} +func Atg_BatchNormNoUpdateOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64){ +cmomentum := *(*C.double)(unsafe.Pointer(&momentum)) +ceps := *(*C.double)(unsafe.Pointer(&eps)) + C.atg__batch_norm_no_update_out(ptr, out0, out1, out2, out3, input, weight, bias, runningMean, runningVar, cmomentum, ceps) +} +func Atg_BatchNormWithUpdate(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64){ +cmomentum := *(*C.double)(unsafe.Pointer(&momentum)) +ceps := *(*C.double)(unsafe.Pointer(&eps)) + C.atg__batch_norm_with_update(ptr, input, weight, bias, runningMean, runningVar, cmomentum, ceps) +} +func Atg_BatchNormWithUpdateFunctional(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64){ +cmomentum := *(*C.double)(unsafe.Pointer(&momentum)) +ceps := *(*C.double)(unsafe.Pointer(&eps)) + C.atg__batch_norm_with_update_functional(ptr, input, weight, bias, runningMean, runningVar, cmomentum, ceps) +} +func Atg_BatchNormWithUpdateOut(ptr *Ctensor, out Ctensor, saveMean Ctensor, saveInvstd Ctensor, reserve Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64){ +cmomentum := *(*C.double)(unsafe.Pointer(&momentum)) +ceps := *(*C.double)(unsafe.Pointer(&eps)) + C.atg__batch_norm_with_update_out(ptr, out, saveMean, saveInvstd, reserve, input, weight, bias, runningMean, runningVar, cmomentum, ceps) +} func Atg_CastByte(ptr *Ctensor, self Ctensor, nonBlocking int32){ cnonBlocking := *(*C.int)(unsafe.Pointer(&nonBlocking)) C.atg__cast_byte(ptr, self, cnonBlocking) @@ -237,6 +263,20 @@ func Atg_CholeskySolveHelperOut(ptr *Ctensor, out Ctensor, self Ctensor, a Ctens cupper := *(*C.int)(unsafe.Pointer(&upper)) C.atg__cholesky_solve_helper_out(ptr, out, self, a, cupper) } +func Atg_ChunkCat(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64, numChunks int64){ +ctensorsDataPtr := (*Ctensor)(unsafe.Pointer(&tensorsData[0])) +ctensorsLen := *(*C.int)(unsafe.Pointer(&tensorsLen)) +cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) +cnumChunks := *(*C.int64_t)(unsafe.Pointer(&numChunks)) + C.atg__chunk_cat(ptr, ctensorsDataPtr, ctensorsLen, cdim, cnumChunks) +} +func Atg_ChunkCatOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64, numChunks int64){ +ctensorsDataPtr := (*Ctensor)(unsafe.Pointer(&tensorsData[0])) +ctensorsLen := *(*C.int)(unsafe.Pointer(&tensorsLen)) +cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) +cnumChunks := *(*C.int64_t)(unsafe.Pointer(&numChunks)) + C.atg__chunk_cat_out(ptr, out, ctensorsDataPtr, ctensorsLen, cdim, cnumChunks) +} func Atg_Coalesce(ptr *Ctensor, self Ctensor){ C.atg__coalesce(ptr, self) } @@ -318,6 +358,14 @@ coutInt32 := *(*C.int)(unsafe.Pointer(&outInt32)) ctranspose := *(*C.int)(unsafe.Pointer(&transpose)) C.atg__convert_indices_from_csr_to_coo_out(ptr, out, crowIndices, colIndices, coutInt32, ctranspose) } +func Atg_ConvertWeightToInt4pack(ptr *Ctensor, self Ctensor, innerKTiles int64){ +cinnerKTiles := *(*C.int64_t)(unsafe.Pointer(&innerKTiles)) + C.atg__convert_weight_to_int4pack(ptr, self, cinnerKTiles) +} +func Atg_ConvertWeightToInt4packForCpu(ptr *Ctensor, self Ctensor, innerKTiles int64){ +cinnerKTiles := *(*C.int64_t)(unsafe.Pointer(&innerKTiles)) + C.atg__convert_weight_to_int4pack_for_cpu(ptr, self, cinnerKTiles) +} func Atg_Convolution(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, transposed int32, outputPaddingData []int64, outputPaddingLen int, groups int64, benchmark int32, deterministic int32, cudnnEnabled int32, allowTf32 int32){ cstrideDataPtr := (*C.int64_t)(unsafe.Pointer(&strideData[0])) cstrideLen := *(*C.int)(unsafe.Pointer(&strideLen)) @@ -396,9 +444,19 @@ cnonBlocking := *(*C.int)(unsafe.Pointer(&nonBlocking)) func Atg_CsltCompress(ptr *Ctensor, input Ctensor){ C.atg__cslt_compress(ptr, input) } -func Atg_CsltSparseMm(ptr *Ctensor, compressedA Ctensor, denseB Ctensor, bias Ctensor, transposeResult int32){ -ctransposeResult := *(*C.int)(unsafe.Pointer(&transposeResult)) - C.atg__cslt_sparse_mm(ptr, compressedA, denseB, bias, ctransposeResult) +func Atg_CsltSparseMm(ptr *Ctensor, compressedA Ctensor, denseB Ctensor, bias Ctensor, alpha Ctensor, outDtype int32, transposeResult int32, algId int64, splitK int64, splitKMode int64){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) +ctransposeResult := *(*C.int)(unsafe.Pointer(&transposeResult)) +calgId := *(*C.int64_t)(unsafe.Pointer(&algId)) +csplitK := *(*C.int64_t)(unsafe.Pointer(&splitK)) +csplitKMode := *(*C.int64_t)(unsafe.Pointer(&splitKMode)) + C.atg__cslt_sparse_mm(ptr, compressedA, denseB, bias, alpha, coutDtype, ctransposeResult, calgId, csplitK, csplitKMode) +} +func Atg_CsltSparseMmSearch(compressedA Ctensor, denseB Ctensor, bias Ctensor, alpha Ctensor, outDtype int32, transposeResult int32) int64{ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) +ctransposeResult := *(*C.int)(unsafe.Pointer(&transposeResult)) + cResult := C.atg__cslt_sparse_mm_search(compressedA, denseB, bias, alpha, coutDtype, ctransposeResult) + return *(*int64)(unsafe.Pointer(&cResult)) } func Atg_CtcLoss(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, inputLengthsLen int, targetLengthsData []int64, targetLengthsLen int, blank int64, zeroInfinity int32){ cinputLengthsDataPtr := (*C.int64_t)(unsafe.Pointer(&inputLengthsData[0])) @@ -451,6 +509,15 @@ cblank := *(*C.int64_t)(unsafe.Pointer(&blank)) czeroInfinity := *(*C.int)(unsafe.Pointer(&zeroInfinity)) C.atg__ctc_loss_tensor_out(ptr, out0, out1, logProbs, targets, inputLengths, targetLengths, cblank, czeroInfinity) } +func Atg_CudnnAttentionBackward(ptr *Ctensor, gradOut Ctensor, query Ctensor, key Ctensor, value Ctensor, out Ctensor, logsumexp Ctensor, philoxSeed Ctensor, philoxOffset Ctensor, attnBias Ctensor, cumSeqQ Ctensor, cumSeqK Ctensor, maxQ int64, maxK int64, dropoutP float64, isCausal int32, scaleVal float64, scaleNull int){ +cmaxQ := *(*C.int64_t)(unsafe.Pointer(&maxQ)) +cmaxK := *(*C.int64_t)(unsafe.Pointer(&maxK)) +cdropoutP := *(*C.double)(unsafe.Pointer(&dropoutP)) +cisCausal := *(*C.int)(unsafe.Pointer(&isCausal)) +cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) +cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) + C.atg__cudnn_attention_backward(ptr, gradOut, query, key, value, out, logsumexp, philoxSeed, philoxOffset, attnBias, cumSeqQ, cumSeqK, cmaxQ, cmaxK, cdropoutP, cisCausal, cscaleVal, cscaleNull) +} func Atg_CudnnCtcLoss(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, inputLengthsLen int, targetLengthsData []int64, targetLengthsLen int, blank int64, deterministic int32, zeroInfinity int32){ cinputLengthsDataPtr := (*C.int64_t)(unsafe.Pointer(&inputLengthsData[0])) cinputLengthsLen := *(*C.int)(unsafe.Pointer(&inputLengthsLen)) @@ -571,17 +638,32 @@ func Atg_DirichletGrad(ptr *Ctensor, x Ctensor, alpha Ctensor, total Ctensor){ func Atg_DirichletGradOut(ptr *Ctensor, out Ctensor, x Ctensor, alpha Ctensor, total Ctensor){ C.atg__dirichlet_grad_out(ptr, out, x, alpha, total) } -func Atg_EfficientAttentionBackward(ptr *Ctensor, gradOut_ Ctensor, query Ctensor, key Ctensor, value Ctensor, bias Ctensor, out Ctensor, cuSeqlensQ Ctensor, cuSeqlensK Ctensor, maxSeqlenK int64, maxSeqlenQ int64, logsumexp Ctensor, dropoutP float64, philoxSeed Ctensor, philoxOffset Ctensor, customMaskType int64, biasRequiresGrad int32, scaleVal float64, scaleNull int, numSplitsKeyVal int64, numSplitsKeyNull int){ -cmaxSeqlenK := *(*C.int64_t)(unsafe.Pointer(&maxSeqlenK)) +func Atg_DynQuantMatmul4bit(ptr *Ctensor, inp Ctensor, packedWeights Ctensor, blockSize int64, inFeatures int64, outFeatures int64){ +cblockSize := *(*C.int64_t)(unsafe.Pointer(&blockSize)) +cinFeatures := *(*C.int64_t)(unsafe.Pointer(&inFeatures)) +coutFeatures := *(*C.int64_t)(unsafe.Pointer(&outFeatures)) + C.atg__dyn_quant_matmul_4bit(ptr, inp, packedWeights, cblockSize, cinFeatures, coutFeatures) +} +func Atg_DynQuantPack4bitWeight(ptr *Ctensor, weights Ctensor, scalesZeros Ctensor, bias Ctensor, blockSize int64, inFeatures int64, outFeatures int64){ +cblockSize := *(*C.int64_t)(unsafe.Pointer(&blockSize)) +cinFeatures := *(*C.int64_t)(unsafe.Pointer(&inFeatures)) +coutFeatures := *(*C.int64_t)(unsafe.Pointer(&outFeatures)) + C.atg__dyn_quant_pack_4bit_weight(ptr, weights, scalesZeros, bias, cblockSize, cinFeatures, coutFeatures) +} +func Atg_EfficientAttentionBackward(ptr *Ctensor, gradOut_ Ctensor, query Ctensor, key Ctensor, value Ctensor, bias Ctensor, out Ctensor, cuSeqlensQ Ctensor, cuSeqlensK Ctensor, maxSeqlenQ int64, maxSeqlenK int64, logsumexp Ctensor, dropoutP float64, philoxSeed Ctensor, philoxOffset Ctensor, customMaskType int64, biasRequiresGrad int32, scaleVal float64, scaleNull int, numSplitsKeyVal int64, numSplitsKeyNull int, windowSizeVal int64, windowSizeNull int, sharedStorageDqdkdv int32){ cmaxSeqlenQ := *(*C.int64_t)(unsafe.Pointer(&maxSeqlenQ)) +cmaxSeqlenK := *(*C.int64_t)(unsafe.Pointer(&maxSeqlenK)) cdropoutP := *(*C.double)(unsafe.Pointer(&dropoutP)) ccustomMaskType := *(*C.int64_t)(unsafe.Pointer(&customMaskType)) cbiasRequiresGrad := *(*C.int)(unsafe.Pointer(&biasRequiresGrad)) cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) cnumSplitsKeyVal := *(*C.int64_t)(unsafe.Pointer(&numSplitsKeyVal)) -cnumSplitsKeyNull := *(*C.uint8_t)(unsafe.Pointer(&numSplitsKeyNull)) - C.atg__efficient_attention_backward(ptr, gradOut_, query, key, value, bias, out, cuSeqlensQ, cuSeqlensK, cmaxSeqlenK, cmaxSeqlenQ, logsumexp, cdropoutP, philoxSeed, philoxOffset, ccustomMaskType, cbiasRequiresGrad, cscaleVal, cscaleNull, cnumSplitsKeyVal, cnumSplitsKeyNull) +cnumSplitsKeyNull := *(*C.uint8_t)(unsafe.Pointer(&numSplitsKeyNull)) +cwindowSizeVal := *(*C.int64_t)(unsafe.Pointer(&windowSizeVal)) +cwindowSizeNull := *(*C.uint8_t)(unsafe.Pointer(&windowSizeNull)) +csharedStorageDqdkdv := *(*C.int)(unsafe.Pointer(&sharedStorageDqdkdv)) + C.atg__efficient_attention_backward(ptr, gradOut_, query, key, value, bias, out, cuSeqlensQ, cuSeqlensK, cmaxSeqlenQ, cmaxSeqlenK, logsumexp, cdropoutP, philoxSeed, philoxOffset, ccustomMaskType, cbiasRequiresGrad, cscaleVal, cscaleNull, cnumSplitsKeyVal, cnumSplitsKeyNull, cwindowSizeVal, cwindowSizeNull, csharedStorageDqdkdv) } func Atg_Efficientzerotensor(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32){ csizeDataPtr := (*C.int64_t)(unsafe.Pointer(&sizeData[0])) @@ -799,14 +881,18 @@ cseed := *(*C.int64_t)(unsafe.Pointer(&seed)) coffset := *(*C.int64_t)(unsafe.Pointer(&offset)) C.atg__fill_mem_eff_dropout_mask_(ptr, self, cdropoutP, cseed, coffset) } -func Atg_FlashAttentionBackward(ptr *Ctensor, gradOut Ctensor, query Ctensor, key Ctensor, value Ctensor, out Ctensor, logsumexp Ctensor, cumSeqQ Ctensor, cumSeqK Ctensor, maxQ int64, maxK int64, dropoutP float64, isCausal int32, philoxSeed Ctensor, philoxOffset Ctensor, scaleVal float64, scaleNull int){ +func Atg_FlashAttentionBackward(ptr *Ctensor, gradOut Ctensor, query Ctensor, key Ctensor, value Ctensor, out Ctensor, logsumexp Ctensor, cumSeqQ Ctensor, cumSeqK Ctensor, maxQ int64, maxK int64, dropoutP float64, isCausal int32, rngState Ctensor, unused Ctensor, scaleVal float64, scaleNull int, windowSizeLeftVal int64, windowSizeLeftNull int, windowSizeRightVal int64, windowSizeRightNull int){ cmaxQ := *(*C.int64_t)(unsafe.Pointer(&maxQ)) cmaxK := *(*C.int64_t)(unsafe.Pointer(&maxK)) cdropoutP := *(*C.double)(unsafe.Pointer(&dropoutP)) cisCausal := *(*C.int)(unsafe.Pointer(&isCausal)) cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) -cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) - C.atg__flash_attention_backward(ptr, gradOut, query, key, value, out, logsumexp, cumSeqQ, cumSeqK, cmaxQ, cmaxK, cdropoutP, cisCausal, philoxSeed, philoxOffset, cscaleVal, cscaleNull) +cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) +cwindowSizeLeftVal := *(*C.int64_t)(unsafe.Pointer(&windowSizeLeftVal)) +cwindowSizeLeftNull := *(*C.uint8_t)(unsafe.Pointer(&windowSizeLeftNull)) +cwindowSizeRightVal := *(*C.int64_t)(unsafe.Pointer(&windowSizeRightVal)) +cwindowSizeRightNull := *(*C.uint8_t)(unsafe.Pointer(&windowSizeRightNull)) + C.atg__flash_attention_backward(ptr, gradOut, query, key, value, out, logsumexp, cumSeqQ, cumSeqK, cmaxQ, cmaxK, cdropoutP, cisCausal, rngState, unused, cscaleVal, cscaleNull, cwindowSizeLeftVal, cwindowSizeLeftNull, cwindowSizeRightVal, cwindowSizeRightNull) } func Atg_Foobar(ptr *Ctensor, self Ctensor, arg1 int32, arg2 int32, arg3 int32){ carg1 := *(*C.int)(unsafe.Pointer(&arg1)) @@ -826,6 +912,12 @@ assertMsgLen := len(assertMsg) cassertMsgLen := *(*C.int)(unsafe.Pointer(&assertMsgLen)) C.atg__functional_assert_async(ptr, self, cassertMsg, cassertMsgLen, depToken) } +func Atg_FunctionalAssertScalar(ptr *Ctensor, selfScalar Cscalar, assertMsg string, depToken Ctensor){ +cassertMsg := C.CString(assertMsg) +assertMsgLen := len(assertMsg) +cassertMsgLen := *(*C.int)(unsafe.Pointer(&assertMsgLen)) + C.atg__functional_assert_scalar(ptr, selfScalar , cassertMsg, cassertMsgLen, depToken) +} func Atg_FunctionalSymConstrainRange(ptr *Ctensor, size Cscalar, minVal int64, minNull int, maxVal int64, maxNull int, depToken Ctensor){ cminVal := *(*C.int64_t)(unsafe.Pointer(&minVal)) cminNull := *(*C.uint8_t)(unsafe.Pointer(&minNull)) @@ -840,6 +932,10 @@ cmaxVal := *(*C.int64_t)(unsafe.Pointer(&maxVal)) cmaxNull := *(*C.uint8_t)(unsafe.Pointer(&maxNull)) C.atg__functional_sym_constrain_range_for_size(ptr, size , cminVal, cminNull, cmaxVal, cmaxNull, depToken) } + + + + func Atg_FusedDropout(ptr *Ctensor, self Ctensor, p float64){ cp := *(*C.double)(unsafe.Pointer(&p)) C.atg__fused_dropout(ptr, self, cp) @@ -875,14 +971,26 @@ cperRowFakeQuant := *(*C.int)(unsafe.Pointer(&perRowFakeQuant)) csymmetricQuant := *(*C.int)(unsafe.Pointer(&symmetricQuant)) C.atg__fused_moving_avg_obs_fq_helper_out(ptr, out0, out1, self, observerOn, fakeQuantOn, runningMin, runningMax, scale, zeroPoint, caveragingConst, cquantMin, cquantMax, cchAxis, cperRowFakeQuant, csymmetricQuant) } -func Atg_FusedSdpChoice(query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, isCausal int32, scaleVal float64, scaleNull int) int64{ +func Atg_FusedRmsNorm(ptr *Ctensor, input Ctensor, normalizedShapeData []int64, normalizedShapeLen int, weight Ctensor, epsVal float64, epsNull int){ +cnormalizedShapeDataPtr := (*C.int64_t)(unsafe.Pointer(&normalizedShapeData[0])) +cnormalizedShapeLen := *(*C.int)(unsafe.Pointer(&normalizedShapeLen)) +cepsVal := *(*C.double)(unsafe.Pointer(&epsVal)) +cepsNull := *(*C.uint8_t)(unsafe.Pointer(&epsNull)) + C.atg__fused_rms_norm(ptr, input, cnormalizedShapeDataPtr, cnormalizedShapeLen, weight, cepsVal, cepsNull) +} +func Atg_FusedSdpChoice(query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, isCausal int32, scaleVal float64, scaleNull int, enableGqa int32) int64{ cdropoutP := *(*C.double)(unsafe.Pointer(&dropoutP)) cisCausal := *(*C.int)(unsafe.Pointer(&isCausal)) cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) - cResult := C.atg__fused_sdp_choice(query, key, value, attnMask, cdropoutP, cisCausal, cscaleVal, cscaleNull) +cenableGqa := *(*C.int)(unsafe.Pointer(&enableGqa)) + cResult := C.atg__fused_sdp_choice(query, key, value, attnMask, cdropoutP, cisCausal, cscaleVal, cscaleNull, cenableGqa) return *(*int64)(unsafe.Pointer(&cResult)) } + + + + func Atg_FwPrimal(ptr *Ctensor, self Ctensor, level int64){ clevel := *(*C.int64_t)(unsafe.Pointer(&level)) C.atg__fw_primal(ptr, self, clevel) @@ -917,6 +1025,10 @@ cpaddingMode := *(*C.int64_t)(unsafe.Pointer(&paddingMode)) calignCorners := *(*C.int)(unsafe.Pointer(&alignCorners)) C.atg__grid_sampler_2d_cpu_fallback_out(ptr, out, input, grid, cinterpolationMode, cpaddingMode, calignCorners) } +func Atg_GroupedMm(ptr *Ctensor, self Ctensor, mat2 Ctensor, offs Ctensor, bias Ctensor, outDtype int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg__grouped_mm(ptr, self, mat2, offs, bias, coutDtype) +} func Atg_HasCompatibleShallowCopyType(self Ctensor, from Ctensor) bool{ cResult := C.atg__has_compatible_shallow_copy_type(self, from) cbool := (int)(cResult) @@ -949,27 +1061,6 @@ cbinsLen := *(*C.int)(unsafe.Pointer(&binsLen)) cdensity := *(*C.int)(unsafe.Pointer(&density)) C.atg__histogramdd_from_bin_tensors_out(ptr, out, self, cbinsDataPtr, cbinsLen, weight, cdensity) } -func Atg_IndexPutImpl(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32, unsafety int32){ -cindicesDataPtr := (*Ctensor)(unsafe.Pointer(&indicesData[0])) -cindicesLen := *(*C.int)(unsafe.Pointer(&indicesLen)) -caccumulate := *(*C.int)(unsafe.Pointer(&accumulate)) -cunsafety := *(*C.int)(unsafe.Pointer(&unsafety)) - C.atg__index_put_impl(ptr, self, cindicesDataPtr, cindicesLen, values, caccumulate, cunsafety) -} -func Atg_IndexPutImpl_(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32, unsafety int32){ -cindicesDataPtr := (*Ctensor)(unsafe.Pointer(&indicesData[0])) -cindicesLen := *(*C.int)(unsafe.Pointer(&indicesLen)) -caccumulate := *(*C.int)(unsafe.Pointer(&accumulate)) -cunsafety := *(*C.int)(unsafe.Pointer(&unsafety)) - C.atg__index_put_impl_(ptr, self, cindicesDataPtr, cindicesLen, values, caccumulate, cunsafety) -} -func Atg_IndexPutImplOut(ptr *Ctensor, out Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32, unsafety int32){ -cindicesDataPtr := (*Ctensor)(unsafe.Pointer(&indicesData[0])) -cindicesLen := *(*C.int)(unsafe.Pointer(&indicesLen)) -caccumulate := *(*C.int)(unsafe.Pointer(&accumulate)) -cunsafety := *(*C.int)(unsafe.Pointer(&unsafety)) - C.atg__index_put_impl_out(ptr, out, self, cindicesDataPtr, cindicesLen, values, caccumulate, cunsafety) -} func Atg_Indices(ptr *Ctensor, self Ctensor){ C.atg__indices(ptr, self) } @@ -997,6 +1088,9 @@ func Atg_IsZerotensor(self Ctensor) bool{ if cbool == 1{return true} return false } +func Atg_LazyClone(ptr *Ctensor, self Ctensor){ + C.atg__lazy_clone(ptr, self) +} func Atg_LinalgDet(ptr *Ctensor, a Ctensor){ C.atg__linalg_det(ptr, a) } @@ -1017,6 +1111,9 @@ cuPLOLen := *(*C.int)(unsafe.Pointer(&uPLOLen)) ccomputeV := *(*C.int)(unsafe.Pointer(&computeV)) C.atg__linalg_eigh_eigenvalues(ptr, eigenvalues, eigenvectors, a, cuPLO, cuPLOLen, ccomputeV) } +func Atg_LinalgEigvals(ptr *Ctensor, self Ctensor){ + C.atg__linalg_eigvals(ptr, self) +} func Atg_LinalgSlogdet(ptr *Ctensor, a Ctensor){ C.atg__linalg_slogdet(ptr, a) } @@ -1175,6 +1272,12 @@ cmaskTypeVal := *(*C.int64_t)(unsafe.Pointer(&maskTypeVal)) cmaskTypeNull := *(*C.uint8_t)(unsafe.Pointer(&maskTypeNull)) C.atg__masked_softmax_out(ptr, out, self, mask, cdimVal, cdimNull, cmaskTypeVal, cmaskTypeNull) } +func Atg_MixedDtypesLinear(ptr *Ctensor, input Ctensor, weight Ctensor, scale Ctensor, bias Ctensor, activation string){ +cactivation := C.CString(activation) +activationLen := len(activation) +cactivationLen := *(*C.int)(unsafe.Pointer(&activationLen)) + C.atg__mixed_dtypes_linear(ptr, input, weight, scale, bias, cactivation, cactivationLen) +} func Atg_MkldnnReshape(ptr *Ctensor, self Ctensor, shapeData []int64, shapeLen int){ cshapeDataPtr := (*C.int64_t)(unsafe.Pointer(&shapeData[0])) cshapeLen := *(*C.int)(unsafe.Pointer(&shapeLen)) @@ -1311,6 +1414,9 @@ func Atg_NegViewCopy(ptr *Ctensor, self Ctensor){ func Atg_NegViewCopyOut(ptr *Ctensor, out Ctensor, self Ctensor){ C.atg__neg_view_copy_out(ptr, out, self) } +func Atg_NestedComputeContiguousStridesOffsets(ptr *Ctensor, nestedSize Ctensor){ + C.atg__nested_compute_contiguous_strides_offsets(ptr, nestedSize) +} func Atg_NestedFromPadded(ptr *Ctensor, padded Ctensor, cpuNestedShapeExample Ctensor, fuseTransform0213 int32){ cfuseTransform0213 := *(*C.int)(unsafe.Pointer(&fuseTransform0213)) C.atg__nested_from_padded(ptr, padded, cpuNestedShapeExample, cfuseTransform0213) @@ -1325,6 +1431,40 @@ func Atg_NestedFromPaddedOut(ptr *Ctensor, out Ctensor, padded Ctensor, cpuNeste cfuseTransform0213 := *(*C.int)(unsafe.Pointer(&fuseTransform0213)) C.atg__nested_from_padded_out(ptr, out, padded, cpuNestedShapeExample, cfuseTransform0213) } +func Atg_NestedFromPaddedTensor(ptr *Ctensor, padded Ctensor, offsets Ctensor, dummy Ctensor, raggedIdx int64, minSeqlen Ctensor, maxSeqlen Ctensor, sumSVal int64, sumSNull int){ +craggedIdx := *(*C.int64_t)(unsafe.Pointer(&raggedIdx)) +csumSVal := *(*C.int64_t)(unsafe.Pointer(&sumSVal)) +csumSNull := *(*C.uint8_t)(unsafe.Pointer(&sumSNull)) + C.atg__nested_from_padded_tensor(ptr, padded, offsets, dummy, craggedIdx, minSeqlen, maxSeqlen, csumSVal, csumSNull) +} +func Atg_NestedGetJaggedDummy(ptr *Ctensor, any Ctensor){ + C.atg__nested_get_jagged_dummy(ptr, any) +} +func Atg_NestedGetLengths(ptr *Ctensor, self Ctensor){ + C.atg__nested_get_lengths(ptr, self) +} +func Atg_NestedGetMaxSeqlen(ptr *Ctensor, self Ctensor){ + C.atg__nested_get_max_seqlen(ptr, self) +} +func Atg_NestedGetMinSeqlen(ptr *Ctensor, self Ctensor){ + C.atg__nested_get_min_seqlen(ptr, self) +} +func Atg_NestedGetOffsets(ptr *Ctensor, self Ctensor){ + C.atg__nested_get_offsets(ptr, self) +} +func Atg_NestedGetRaggedIdx(self Ctensor) int64{ + cResult := C.atg__nested_get_ragged_idx(self) + return *(*int64)(unsafe.Pointer(&cResult)) +} +func Atg_NestedGetValues(ptr *Ctensor, self Ctensor){ + C.atg__nested_get_values(ptr, self) +} +func Atg_NestedGetValuesCopy(ptr *Ctensor, self Ctensor){ + C.atg__nested_get_values_copy(ptr, self) +} +func Atg_NestedGetValuesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor){ + C.atg__nested_get_values_copy_out(ptr, out, self) +} func Atg_NestedSelectBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, dim int64, index int64){ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) cindex := *(*C.int64_t)(unsafe.Pointer(&index)) @@ -1345,6 +1485,18 @@ func Atg_NestedViewFromBufferCopy(ptr *Ctensor, self Ctensor, nestedSize Ctensor func Atg_NestedViewFromBufferCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, nestedSize Ctensor, nestedStrides Ctensor, offsets Ctensor){ C.atg__nested_view_from_buffer_copy_out(ptr, out, self, nestedSize, nestedStrides, offsets) } +func Atg_NestedViewFromJagged(ptr *Ctensor, self Ctensor, offsets Ctensor, dummy Ctensor, lengths Ctensor, raggedIdx int64, minSeqlen Ctensor, maxSeqlen Ctensor){ +craggedIdx := *(*C.int64_t)(unsafe.Pointer(&raggedIdx)) + C.atg__nested_view_from_jagged(ptr, self, offsets, dummy, lengths, craggedIdx, minSeqlen, maxSeqlen) +} +func Atg_NestedViewFromJaggedCopy(ptr *Ctensor, self Ctensor, offsets Ctensor, dummy Ctensor, lengths Ctensor, raggedIdx int64, minSeqlen Ctensor, maxSeqlen Ctensor){ +craggedIdx := *(*C.int64_t)(unsafe.Pointer(&raggedIdx)) + C.atg__nested_view_from_jagged_copy(ptr, self, offsets, dummy, lengths, craggedIdx, minSeqlen, maxSeqlen) +} +func Atg_NestedViewFromJaggedCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, offsets Ctensor, dummy Ctensor, lengths Ctensor, raggedIdx int64, minSeqlen Ctensor, maxSeqlen Ctensor){ +craggedIdx := *(*C.int64_t)(unsafe.Pointer(&raggedIdx)) + C.atg__nested_view_from_jagged_copy_out(ptr, out, self, offsets, dummy, lengths, craggedIdx, minSeqlen, maxSeqlen) +} func Atg_NewZerosWithSameFeatureMeta(ptr *Ctensor, self Ctensor, other Ctensor, selfNumBatchDims int64){ cselfNumBatchDims := *(*C.int64_t)(unsafe.Pointer(&selfNumBatchDims)) C.atg__new_zeros_with_same_feature_meta(ptr, self, other, cselfNumBatchDims) @@ -1432,6 +1584,7 @@ func Atg_PreluKernelBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, wei C.atg__prelu_kernel_backward(ptr, gradOutput, self, weight) } + func Atg_RemoveBatchDim(ptr *Ctensor, self Ctensor, level int64, batchSize int64, outDim int64){ clevel := *(*C.int64_t)(unsafe.Pointer(&level)) cbatchSize := *(*C.int64_t)(unsafe.Pointer(&batchSize)) @@ -1489,6 +1642,11 @@ func Atg_RowwisePrune(ptr *Ctensor, weight Ctensor, mask Ctensor, compressedIndi ccompressedIndicesDtype := *(*C.int)(unsafe.Pointer(&compressedIndicesDtype)) C.atg__rowwise_prune(ptr, weight, mask, ccompressedIndicesDtype) } +func Atg_SafeSoftmax(ptr *Ctensor, self Ctensor, dim int64, dtype int32){ +cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) +cdtype := *(*C.int)(unsafe.Pointer(&dtype)) + C.atg__safe_softmax(ptr, self, cdim, cdtype) +} func Atg_SampleDirichlet(ptr *Ctensor, self Ctensor){ C.atg__sample_dirichlet(ptr, self) } @@ -1498,12 +1656,29 @@ func Atg_SampleDirichletOut(ptr *Ctensor, out Ctensor, self Ctensor){ func Atg_SaturateWeightToFp16(ptr *Ctensor, weight Ctensor){ C.atg__saturate_weight_to_fp16(ptr, weight) } -func Atg_ScaledDotProductAttentionMath(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, isCausal int32, dropoutMask Ctensor, scaleVal float64, scaleNull int){ +func Atg_ScaledDotProductAttentionMath(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, isCausal int32, dropoutMask Ctensor, scaleVal float64, scaleNull int, enableGqa int32){ +cdropoutP := *(*C.double)(unsafe.Pointer(&dropoutP)) +cisCausal := *(*C.int)(unsafe.Pointer(&isCausal)) +cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) +cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) +cenableGqa := *(*C.int)(unsafe.Pointer(&enableGqa)) + C.atg__scaled_dot_product_attention_math(ptr, query, key, value, attnMask, cdropoutP, cisCausal, dropoutMask, cscaleVal, cscaleNull, cenableGqa) +} +func Atg_ScaledDotProductAttentionMathForMps(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, isCausal int32, dropoutMask Ctensor, scaleVal float64, scaleNull int){ +cdropoutP := *(*C.double)(unsafe.Pointer(&dropoutP)) +cisCausal := *(*C.int)(unsafe.Pointer(&isCausal)) +cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) +cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) + C.atg__scaled_dot_product_attention_math_for_mps(ptr, query, key, value, attnMask, cdropoutP, cisCausal, dropoutMask, cscaleVal, cscaleNull) +} +func Atg_ScaledDotProductCudnnAttentionBackward(ptr *Ctensor, gradOut Ctensor, query Ctensor, key Ctensor, value Ctensor, out Ctensor, logsumexp Ctensor, philoxSeed Ctensor, philoxOffset Ctensor, attnBias Ctensor, cumSeqQ Ctensor, cumSeqK Ctensor, maxQ int64, maxK int64, dropoutP float64, isCausal int32, scaleVal float64, scaleNull int){ +cmaxQ := *(*C.int64_t)(unsafe.Pointer(&maxQ)) +cmaxK := *(*C.int64_t)(unsafe.Pointer(&maxK)) cdropoutP := *(*C.double)(unsafe.Pointer(&dropoutP)) cisCausal := *(*C.int)(unsafe.Pointer(&isCausal)) cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) - C.atg__scaled_dot_product_attention_math(ptr, query, key, value, attnMask, cdropoutP, cisCausal, dropoutMask, cscaleVal, cscaleNull) + C.atg__scaled_dot_product_cudnn_attention_backward(ptr, gradOut, query, key, value, out, logsumexp, philoxSeed, philoxOffset, attnBias, cumSeqQ, cumSeqK, cmaxQ, cmaxK, cdropoutP, cisCausal, cscaleVal, cscaleNull) } func Atg_ScaledDotProductEfficientAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnBias Ctensor, computeLogSumexp int32, dropoutP float64, isCausal int32, scaleVal float64, scaleNull int){ ccomputeLogSumexp := *(*C.int)(unsafe.Pointer(&computeLogSumexp)) @@ -1522,13 +1697,91 @@ cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) C.atg__scaled_dot_product_flash_attention_backward(ptr, gradOut, query, key, value, out, logsumexp, cumSeqQ, cumSeqK, cmaxQ, cmaxK, cdropoutP, cisCausal, philoxSeed, philoxOffset, cscaleVal, cscaleNull) } -func Atg_ScaledMm(ptr *Ctensor, self Ctensor, mat2 Ctensor, bias Ctensor, outDtype int32, scaleA Ctensor, scaleB Ctensor, scaleResult Ctensor){ -coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) - C.atg__scaled_mm(ptr, self, mat2, bias, coutDtype, scaleA, scaleB, scaleResult) +func Atg_ScaledDotProductFlashAttentionForCpu(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, dropoutP float64, isCausal int32, attnMask Ctensor, scaleVal float64, scaleNull int){ +cdropoutP := *(*C.double)(unsafe.Pointer(&dropoutP)) +cisCausal := *(*C.int)(unsafe.Pointer(&isCausal)) +cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) +cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) + C.atg__scaled_dot_product_flash_attention_for_cpu(ptr, query, key, value, cdropoutP, cisCausal, attnMask, cscaleVal, cscaleNull) } -func Atg_ScaledMmOut(ptr *Ctensor, out Ctensor, outAmax Ctensor, self Ctensor, mat2 Ctensor, bias Ctensor, outDtype int32, scaleA Ctensor, scaleB Ctensor, scaleResult Ctensor){ -coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) - C.atg__scaled_mm_out(ptr, out, outAmax, self, mat2, bias, coutDtype, scaleA, scaleB, scaleResult) +func Atg_ScaledDotProductFlashAttentionForCpuBackward(ptr *Ctensor, gradOut Ctensor, query Ctensor, key Ctensor, value Ctensor, out Ctensor, logsumexp Ctensor, dropoutP float64, isCausal int32, attnMask Ctensor, scaleVal float64, scaleNull int){ +cdropoutP := *(*C.double)(unsafe.Pointer(&dropoutP)) +cisCausal := *(*C.int)(unsafe.Pointer(&isCausal)) +cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) +cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) + C.atg__scaled_dot_product_flash_attention_for_cpu_backward(ptr, gradOut, query, key, value, out, logsumexp, cdropoutP, cisCausal, attnMask, cscaleVal, cscaleNull) +} +func Atg_ScaledGroupedMm(ptr *Ctensor, self Ctensor, mat2 Ctensor, scaleA Ctensor, scaleB Ctensor, offs Ctensor, bias Ctensor, scaleResult Ctensor, outDtype int32, useFastAccum int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) +cuseFastAccum := *(*C.int)(unsafe.Pointer(&useFastAccum)) + C.atg__scaled_grouped_mm(ptr, self, mat2, scaleA, scaleB, offs, bias, scaleResult, coutDtype, cuseFastAccum) +} +func Atg_ScaledGroupedMmV2(ptr *Ctensor, self Ctensor, mat2 Ctensor, scaleAData []Ctensor, scaleALen int, recipeAData []int64, recipeALen int, swizzleAData []int64, swizzleALen int, scaleBData []Ctensor, scaleBLen int, recipeBData []int64, recipeBLen int, swizzleBData []int64, swizzleBLen int, offs Ctensor, bias Ctensor, outDtype int32, contractionDimData []int64, contractionDimLen int, useFastAccum int32){ +cscaleADataPtr := (*Ctensor)(unsafe.Pointer(&scaleAData[0])) +cscaleALen := *(*C.int)(unsafe.Pointer(&scaleALen)) +crecipeADataPtr := (*C.int64_t)(unsafe.Pointer(&recipeAData[0])) +crecipeALen := *(*C.int)(unsafe.Pointer(&recipeALen)) +cswizzleADataPtr := (*C.int64_t)(unsafe.Pointer(&swizzleAData[0])) +cswizzleALen := *(*C.int)(unsafe.Pointer(&swizzleALen)) +cscaleBDataPtr := (*Ctensor)(unsafe.Pointer(&scaleBData[0])) +cscaleBLen := *(*C.int)(unsafe.Pointer(&scaleBLen)) +crecipeBDataPtr := (*C.int64_t)(unsafe.Pointer(&recipeBData[0])) +crecipeBLen := *(*C.int)(unsafe.Pointer(&recipeBLen)) +cswizzleBDataPtr := (*C.int64_t)(unsafe.Pointer(&swizzleBData[0])) +cswizzleBLen := *(*C.int)(unsafe.Pointer(&swizzleBLen)) +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) +ccontractionDimDataPtr := (*C.int64_t)(unsafe.Pointer(&contractionDimData[0])) +ccontractionDimLen := *(*C.int)(unsafe.Pointer(&contractionDimLen)) +cuseFastAccum := *(*C.int)(unsafe.Pointer(&useFastAccum)) + C.atg__scaled_grouped_mm_v2(ptr, self, mat2, cscaleADataPtr, cscaleALen, crecipeADataPtr, crecipeALen, cswizzleADataPtr, cswizzleALen, cscaleBDataPtr, cscaleBLen, crecipeBDataPtr, crecipeBLen, cswizzleBDataPtr, cswizzleBLen, offs, bias, coutDtype, ccontractionDimDataPtr, ccontractionDimLen, cuseFastAccum) +} +func Atg_ScaledMm(ptr *Ctensor, self Ctensor, mat2 Ctensor, scaleA Ctensor, scaleB Ctensor, bias Ctensor, scaleResult Ctensor, outDtype int32, useFastAccum int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) +cuseFastAccum := *(*C.int)(unsafe.Pointer(&useFastAccum)) + C.atg__scaled_mm(ptr, self, mat2, scaleA, scaleB, bias, scaleResult, coutDtype, cuseFastAccum) +} +func Atg_ScaledMmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor, scaleA Ctensor, scaleB Ctensor, bias Ctensor, scaleResult Ctensor, outDtype int32, useFastAccum int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) +cuseFastAccum := *(*C.int)(unsafe.Pointer(&useFastAccum)) + C.atg__scaled_mm_out(ptr, out, self, mat2, scaleA, scaleB, bias, scaleResult, coutDtype, cuseFastAccum) +} +func Atg_ScaledMmV2(ptr *Ctensor, self Ctensor, mat2 Ctensor, scaleAData []Ctensor, scaleALen int, recipeAData []int64, recipeALen int, swizzleAData []int64, swizzleALen int, scaleBData []Ctensor, scaleBLen int, recipeBData []int64, recipeBLen int, swizzleBData []int64, swizzleBLen int, bias Ctensor, outDtype int32, contractionDimData []int64, contractionDimLen int, useFastAccum int32){ +cscaleADataPtr := (*Ctensor)(unsafe.Pointer(&scaleAData[0])) +cscaleALen := *(*C.int)(unsafe.Pointer(&scaleALen)) +crecipeADataPtr := (*C.int64_t)(unsafe.Pointer(&recipeAData[0])) +crecipeALen := *(*C.int)(unsafe.Pointer(&recipeALen)) +cswizzleADataPtr := (*C.int64_t)(unsafe.Pointer(&swizzleAData[0])) +cswizzleALen := *(*C.int)(unsafe.Pointer(&swizzleALen)) +cscaleBDataPtr := (*Ctensor)(unsafe.Pointer(&scaleBData[0])) +cscaleBLen := *(*C.int)(unsafe.Pointer(&scaleBLen)) +crecipeBDataPtr := (*C.int64_t)(unsafe.Pointer(&recipeBData[0])) +crecipeBLen := *(*C.int)(unsafe.Pointer(&recipeBLen)) +cswizzleBDataPtr := (*C.int64_t)(unsafe.Pointer(&swizzleBData[0])) +cswizzleBLen := *(*C.int)(unsafe.Pointer(&swizzleBLen)) +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) +ccontractionDimDataPtr := (*C.int64_t)(unsafe.Pointer(&contractionDimData[0])) +ccontractionDimLen := *(*C.int)(unsafe.Pointer(&contractionDimLen)) +cuseFastAccum := *(*C.int)(unsafe.Pointer(&useFastAccum)) + C.atg__scaled_mm_v2(ptr, self, mat2, cscaleADataPtr, cscaleALen, crecipeADataPtr, crecipeALen, cswizzleADataPtr, cswizzleALen, cscaleBDataPtr, cscaleBLen, crecipeBDataPtr, crecipeBLen, cswizzleBDataPtr, cswizzleBLen, bias, coutDtype, ccontractionDimDataPtr, ccontractionDimLen, cuseFastAccum) +} +func Atg_ScaledMmV2Out(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor, scaleAData []Ctensor, scaleALen int, recipeAData []int64, recipeALen int, swizzleAData []int64, swizzleALen int, scaleBData []Ctensor, scaleBLen int, recipeBData []int64, recipeBLen int, swizzleBData []int64, swizzleBLen int, bias Ctensor, outDtype int32, contractionDimData []int64, contractionDimLen int, useFastAccum int32){ +cscaleADataPtr := (*Ctensor)(unsafe.Pointer(&scaleAData[0])) +cscaleALen := *(*C.int)(unsafe.Pointer(&scaleALen)) +crecipeADataPtr := (*C.int64_t)(unsafe.Pointer(&recipeAData[0])) +crecipeALen := *(*C.int)(unsafe.Pointer(&recipeALen)) +cswizzleADataPtr := (*C.int64_t)(unsafe.Pointer(&swizzleAData[0])) +cswizzleALen := *(*C.int)(unsafe.Pointer(&swizzleALen)) +cscaleBDataPtr := (*Ctensor)(unsafe.Pointer(&scaleBData[0])) +cscaleBLen := *(*C.int)(unsafe.Pointer(&scaleBLen)) +crecipeBDataPtr := (*C.int64_t)(unsafe.Pointer(&recipeBData[0])) +crecipeBLen := *(*C.int)(unsafe.Pointer(&recipeBLen)) +cswizzleBDataPtr := (*C.int64_t)(unsafe.Pointer(&swizzleBData[0])) +cswizzleBLen := *(*C.int)(unsafe.Pointer(&swizzleBLen)) +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) +ccontractionDimDataPtr := (*C.int64_t)(unsafe.Pointer(&contractionDimData[0])) +ccontractionDimLen := *(*C.int)(unsafe.Pointer(&contractionDimLen)) +cuseFastAccum := *(*C.int)(unsafe.Pointer(&useFastAccum)) + C.atg__scaled_mm_v2_out(ptr, out, self, mat2, cscaleADataPtr, cscaleALen, crecipeADataPtr, crecipeALen, cswizzleADataPtr, cswizzleALen, cscaleBDataPtr, cscaleBLen, crecipeBDataPtr, crecipeBLen, cswizzleBDataPtr, cswizzleBLen, bias, coutDtype, ccontractionDimDataPtr, ccontractionDimLen, cuseFastAccum) } func Atg_ScatterReduce(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, reduce string, includeSelf int32){ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) @@ -1663,6 +1916,18 @@ coptionsKind := *(*C.int)(unsafe.Pointer(&optionsKind)) coptionsDevice := *(*C.int)(unsafe.Pointer(&optionsDevice)) C.atg__sparse_compressed_tensor_unsafe(ptr, compressedIndices, plainIndices, values, csizeDataPtr, csizeLen, coptionsKind, coptionsDevice) } +func Atg_SparseCompressedTensorWithDims(ptr *Ctensor, nnz int64, denseDim int64, sizeData []int64, sizeLen int, blocksizeData []int64, blocksizeLen int, indexDtype int32, optionsKind int32, optionsDevice int32){ +cnnz := *(*C.int64_t)(unsafe.Pointer(&nnz)) +cdenseDim := *(*C.int64_t)(unsafe.Pointer(&denseDim)) +csizeDataPtr := (*C.int64_t)(unsafe.Pointer(&sizeData[0])) +csizeLen := *(*C.int)(unsafe.Pointer(&sizeLen)) +cblocksizeDataPtr := (*C.int64_t)(unsafe.Pointer(&blocksizeData[0])) +cblocksizeLen := *(*C.int)(unsafe.Pointer(&blocksizeLen)) +cindexDtype := *(*C.int)(unsafe.Pointer(&indexDtype)) +coptionsKind := *(*C.int)(unsafe.Pointer(&optionsKind)) +coptionsDevice := *(*C.int)(unsafe.Pointer(&optionsDevice)) + C.atg__sparse_compressed_tensor_with_dims(ptr, cnnz, cdenseDim, csizeDataPtr, csizeLen, cblocksizeDataPtr, cblocksizeLen, cindexDtype, coptionsKind, coptionsDevice) +} func Atg_SparseCooTensorUnsafe(ptr *Ctensor, indices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32, isCoalesced int32){ csizeDataPtr := (*C.int64_t)(unsafe.Pointer(&sizeData[0])) csizeLen := *(*C.int)(unsafe.Pointer(&sizeLen)) @@ -1793,11 +2058,29 @@ reduceLen := len(reduce) creduceLen := *(*C.int)(unsafe.Pointer(&reduceLen)) C.atg__sparse_mm_reduce_impl(ptr, self, other, creduce, creduceLen) } -func Atg_SparseSemiStructuredLinear(ptr *Ctensor, input Ctensor, weight Ctensor, meta Ctensor, bias Ctensor, activation string){ +func Atg_SparseSemiStructuredApply(ptr *Ctensor, input Ctensor, threadMasks Ctensor){ + C.atg__sparse_semi_structured_apply(ptr, input, threadMasks) +} +func Atg_SparseSemiStructuredApplyDense(ptr *Ctensor, input Ctensor, threadMasks Ctensor){ + C.atg__sparse_semi_structured_apply_dense(ptr, input, threadMasks) +} +func Atg_SparseSemiStructuredLinear(ptr *Ctensor, input Ctensor, weight Ctensor, meta Ctensor, bias Ctensor, activation string, outDtype int32){ cactivation := C.CString(activation) activationLen := len(activation) -cactivationLen := *(*C.int)(unsafe.Pointer(&activationLen)) - C.atg__sparse_semi_structured_linear(ptr, input, weight, meta, bias, cactivation, cactivationLen) +cactivationLen := *(*C.int)(unsafe.Pointer(&activationLen)) +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg__sparse_semi_structured_linear(ptr, input, weight, meta, bias, cactivation, cactivationLen, coutDtype) +} +func Atg_SparseSemiStructuredMm(ptr *Ctensor, mat1 Ctensor, mat1Meta Ctensor, mat2 Ctensor, outDtype int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg__sparse_semi_structured_mm(ptr, mat1, mat1Meta, mat2, coutDtype) +} +func Atg_SparseSemiStructuredTile(ptr *Ctensor, input Ctensor, algorithm string, useCutlass int32){ +calgorithm := C.CString(algorithm) +algorithmLen := len(algorithm) +calgorithmLen := *(*C.int)(unsafe.Pointer(&algorithmLen)) +cuseCutlass := *(*C.int)(unsafe.Pointer(&useCutlass)) + C.atg__sparse_semi_structured_tile(ptr, input, calgorithm, calgorithmLen, cuseCutlass) } func Atg_SparseSoftmax(ptr *Ctensor, self Ctensor, dim int64, halfToFloat int32){ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) @@ -1873,6 +2156,10 @@ cshapeLen := *(*C.int)(unsafe.Pointer(&shapeLen)) clayout := *(*C.int8_t)(unsafe.Pointer(&layout)) C.atg__spdiags_out(ptr, out, diagonals, offsets, cshapeDataPtr, cshapeLen, clayout) } +func Atg_Spsolve(ptr *Ctensor, a Ctensor, b Ctensor, left int32){ +cleft := *(*C.int)(unsafe.Pointer(&left)) + C.atg__spsolve(ptr, a, b, cleft) +} func Atg_Stack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64){ ctensorsDataPtr := (*Ctensor)(unsafe.Pointer(&tensorsData[0])) ctensorsLen := *(*C.int)(unsafe.Pointer(&tensorsLen)) @@ -1967,6 +2254,11 @@ caddendsDataPtr := (*C.int64_t)(unsafe.Pointer(&addendsData[0])) caddendsLen := *(*C.int)(unsafe.Pointer(&addendsLen)) C.atg__test_optional_intlist_out(ptr, out, values, caddendsDataPtr, caddendsLen) } +func Atg_TestParallelMaterialize(ptr *Ctensor, self Ctensor, numParallel int64, skipFirst int32){ +cnumParallel := *(*C.int64_t)(unsafe.Pointer(&numParallel)) +cskipFirst := *(*C.int)(unsafe.Pointer(&skipFirst)) + C.atg__test_parallel_materialize(ptr, self, cnumParallel, cskipFirst) +} func Atg_TestSerializationSubcmul(ptr *Ctensor, self Ctensor, other Ctensor){ C.atg__test_serialization_subcmul(ptr, self, other) } @@ -2177,17 +2469,6 @@ func Atg_UnpackDual(ptr *Ctensor, dual Ctensor, level int64){ clevel := *(*C.int64_t)(unsafe.Pointer(&level)) C.atg__unpack_dual(ptr, dual, clevel) } -func Atg_UnsafeIndex(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int){ -cindicesDataPtr := (*Ctensor)(unsafe.Pointer(&indicesData[0])) -cindicesLen := *(*C.int)(unsafe.Pointer(&indicesLen)) - C.atg__unsafe_index(ptr, self, cindicesDataPtr, cindicesLen) -} -func Atg_UnsafeIndexPut(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32){ -cindicesDataPtr := (*Ctensor)(unsafe.Pointer(&indicesData[0])) -cindicesLen := *(*C.int)(unsafe.Pointer(&indicesLen)) -caccumulate := *(*C.int)(unsafe.Pointer(&accumulate)) - C.atg__unsafe_index_put(ptr, self, cindicesDataPtr, cindicesLen, values, caccumulate) -} func Atg_UnsafeView(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int){ csizeDataPtr := (*C.int64_t)(unsafe.Pointer(&sizeData[0])) csizeLen := *(*C.int)(unsafe.Pointer(&sizeLen)) @@ -2480,6 +2761,21 @@ func Atg_Version(self Ctensor) int64{ cResult := C.atg__version(self) return *(*int64)(unsafe.Pointer(&cResult)) } +func Atg_WeightInt4packMm(ptr *Ctensor, self Ctensor, mat2 Ctensor, qGroupSize int64, qScaleAndZeros Ctensor){ +cqGroupSize := *(*C.int64_t)(unsafe.Pointer(&qGroupSize)) + C.atg__weight_int4pack_mm(ptr, self, mat2, cqGroupSize, qScaleAndZeros) +} +func Atg_WeightInt4packMmForCpu(ptr *Ctensor, self Ctensor, mat2 Ctensor, qGroupSize int64, qScaleAndZeros Ctensor){ +cqGroupSize := *(*C.int64_t)(unsafe.Pointer(&qGroupSize)) + C.atg__weight_int4pack_mm_for_cpu(ptr, self, mat2, cqGroupSize, qScaleAndZeros) +} +func Atg_WeightInt4packMmWithScalesAndZeros(ptr *Ctensor, self Ctensor, mat2 Ctensor, qGroupSize int64, qScale Ctensor, qZeros Ctensor){ +cqGroupSize := *(*C.int64_t)(unsafe.Pointer(&qGroupSize)) + C.atg__weight_int4pack_mm_with_scales_and_zeros(ptr, self, mat2, cqGroupSize, qScale, qZeros) +} +func Atg_WeightInt8packMm(ptr *Ctensor, self Ctensor, mat2 Ctensor, scales Ctensor){ + C.atg__weight_int8pack_mm(ptr, self, mat2, scales) +} func Atg_WeightNorm(ptr *Ctensor, v Ctensor, g Ctensor, dim int64){ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) C.atg__weight_norm(ptr, v, g, cdim) @@ -2504,6 +2800,13 @@ func Atg_WeightNormInterfaceOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, v Cten cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) C.atg__weight_norm_interface_out(ptr, out0, out1, v, g, cdim) } +func Atg_WrappedLinearPrepack(ptr *Ctensor, weight Ctensor, weightScale Ctensor, weightZeroPoint Ctensor, bias Ctensor){ + C.atg__wrapped_linear_prepack(ptr, weight, weightScale, weightZeroPoint, bias) +} +func Atg_WrappedQuantizedLinearPrepacked(ptr *Ctensor, input Ctensor, inputScale Ctensor, inputZeroPoint Ctensor, packedWeight Ctensor, outputScale Ctensor, outputZeroPoint Ctensor, outChannel int64){ +coutChannel := *(*C.int64_t)(unsafe.Pointer(&outChannel)) + C.atg__wrapped_quantized_linear_prepacked(ptr, input, inputScale, inputZeroPoint, packedWeight, outputScale, outputZeroPoint, coutChannel) +} func AtgAbs(ptr *Ctensor, self Ctensor){ C.atg_abs(ptr, self) } @@ -2545,6 +2848,11 @@ coutputSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&outputSizeData[0])) coutputSizeLen := *(*C.int)(unsafe.Pointer(&outputSizeLen)) C.atg_adaptive_avg_pool1d(ptr, self, coutputSizeDataPtr, coutputSizeLen) } +func AtgAdaptiveAvgPool1dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int){ +coutputSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&outputSizeData[0])) +coutputSizeLen := *(*C.int)(unsafe.Pointer(&outputSizeLen)) + C.atg_adaptive_avg_pool1d_out(ptr, out, self, coutputSizeDataPtr, coutputSizeLen) +} func AtgAdaptiveAvgPool2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int){ coutputSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&outputSizeData[0])) coutputSizeLen := *(*C.int)(unsafe.Pointer(&outputSizeLen)) @@ -2656,6 +2964,14 @@ func AtgAddmm(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor){ func AtgAddmm_(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor){ C.atg_addmm_(ptr, self, mat1, mat2) } +func AtgAddmmDtype(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor, outDtype int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg_addmm_dtype(ptr, self, mat1, mat2, coutDtype) +} +func AtgAddmmDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor, outDtype int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg_addmm_dtype_out(ptr, out, self, mat1, mat2, coutDtype) +} func AtgAddmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor){ C.atg_addmm_out(ptr, out, self, mat1, mat2) } @@ -2722,6 +3038,18 @@ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) C.atg_all_dim(ptr, self, cdim, ckeepdim) } +func AtgAllDims(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32){ +cdimDataPtr := (*C.int64_t)(unsafe.Pointer(&dimData[0])) +cdimLen := *(*C.int)(unsafe.Pointer(&dimLen)) +ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) + C.atg_all_dims(ptr, self, cdimDataPtr, cdimLen, ckeepdim) +} +func AtgAllDimsOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32){ +cdimDataPtr := (*C.int64_t)(unsafe.Pointer(&dimData[0])) +cdimLen := *(*C.int)(unsafe.Pointer(&dimLen)) +ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) + C.atg_all_dims_out(ptr, out, self, cdimDataPtr, cdimLen, ckeepdim) +} func AtgAllOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, keepdim int32){ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) @@ -2799,6 +3127,18 @@ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) C.atg_any_dim(ptr, self, cdim, ckeepdim) } +func AtgAnyDims(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32){ +cdimDataPtr := (*C.int64_t)(unsafe.Pointer(&dimData[0])) +cdimLen := *(*C.int)(unsafe.Pointer(&dimLen)) +ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) + C.atg_any_dims(ptr, self, cdimDataPtr, cdimLen, ckeepdim) +} +func AtgAnyDimsOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32){ +cdimDataPtr := (*C.int64_t)(unsafe.Pointer(&dimData[0])) +cdimLen := *(*C.int)(unsafe.Pointer(&dimLen)) +ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) + C.atg_any_dims_out(ptr, out, self, cdimDataPtr, cdimLen, ckeepdim) +} func AtgAnyOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, keepdim int32){ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) @@ -3048,6 +3388,17 @@ cceilMode := *(*C.int)(unsafe.Pointer(&ceilMode)) ccountIncludePad := *(*C.int)(unsafe.Pointer(&countIncludePad)) C.atg_avg_pool1d(ptr, self, ckernelSizeDataPtr, ckernelSizeLen, cstrideDataPtr, cstrideLen, cpaddingDataPtr, cpaddingLen, cceilMode, ccountIncludePad) } +func AtgAvgPool1dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32){ +ckernelSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&kernelSizeData[0])) +ckernelSizeLen := *(*C.int)(unsafe.Pointer(&kernelSizeLen)) +cstrideDataPtr := (*C.int64_t)(unsafe.Pointer(&strideData[0])) +cstrideLen := *(*C.int)(unsafe.Pointer(&strideLen)) +cpaddingDataPtr := (*C.int64_t)(unsafe.Pointer(&paddingData[0])) +cpaddingLen := *(*C.int)(unsafe.Pointer(&paddingLen)) +cceilMode := *(*C.int)(unsafe.Pointer(&ceilMode)) +ccountIncludePad := *(*C.int)(unsafe.Pointer(&countIncludePad)) + C.atg_avg_pool1d_out(ptr, out, self, ckernelSizeDataPtr, ckernelSizeLen, cstrideDataPtr, cstrideLen, cpaddingDataPtr, cpaddingLen, cceilMode, ccountIncludePad) +} func AtgAvgPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32, divisorOverrideVal int64, divisorOverrideNull int){ ckernelSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&kernelSizeData[0])) ckernelSizeLen := *(*C.int)(unsafe.Pointer(&kernelSizeLen)) @@ -3158,6 +3509,14 @@ func AtgBaddbmm(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor, beta func AtgBaddbmm_(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor){ C.atg_baddbmm_(ptr, self, batch1, batch2) } +func AtgBaddbmmDtype(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor, outDtype int32, beta Cscalar, alpha Cscalar){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg_baddbmm_dtype(ptr, self, batch1, batch2, coutDtype, beta , alpha ) +} +func AtgBaddbmmDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor, outDtype int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg_baddbmm_dtype_out(ptr, out, self, batch1, batch2, coutDtype) +} func AtgBaddbmmOut(ptr *Ctensor, out Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor){ C.atg_baddbmm_out(ptr, out, self, batch1, batch2) } @@ -3476,6 +3835,14 @@ ctensorsLen := *(*C.int)(unsafe.Pointer(&tensorsLen)) func AtgBmm(ptr *Ctensor, self Ctensor, mat2 Ctensor){ C.atg_bmm(ptr, self, mat2) } +func AtgBmmDtype(ptr *Ctensor, self Ctensor, mat2 Ctensor, outDtype int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg_bmm_dtype(ptr, self, mat2, coutDtype) +} +func AtgBmmDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor, outDtype int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg_bmm_dtype_out(ptr, out, self, mat2, coutDtype) +} func AtgBmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor){ C.atg_bmm_out(ptr, out, self, mat2) } @@ -3505,10 +3872,10 @@ coutInt32 := *(*C.int)(unsafe.Pointer(&outInt32)) cright := *(*C.int)(unsafe.Pointer(&right)) C.atg_bucketize_tensor_out(ptr, out, self, boundaries, coutInt32, cright) } -func AtgCanCast(from int32, to int32) bool{ -cfrom := *(*C.int)(unsafe.Pointer(&from)) +func AtgCanCast(from_ int32, to int32) bool{ +cfrom_ := *(*C.int)(unsafe.Pointer(&from_)) cto := *(*C.int)(unsafe.Pointer(&to)) - cResult := C.atg_can_cast(cfrom, cto) + cResult := C.atg_can_cast(cfrom_, cto) cbool := (int)(cResult) if cbool == 1{return true} return false @@ -4960,6 +5327,12 @@ func AtgFbgemmLinearFp16Weight(ptr *Ctensor, input Ctensor, packedWeight Ctensor func AtgFbgemmLinearFp16WeightFp32Activation(ptr *Ctensor, input Ctensor, packedWeight Ctensor, bias Ctensor){ C.atg_fbgemm_linear_fp16_weight_fp32_activation(ptr, input, packedWeight, bias) } +func AtgFbgemmLinearFp16WeightFp32ActivationOut(ptr *Ctensor, input Ctensor, packedWeight Ctensor, bias Ctensor, output Ctensor){ + C.atg_fbgemm_linear_fp16_weight_fp32_activation_out(ptr, input, packedWeight, bias, output) +} +func AtgFbgemmLinearFp16WeightOut(ptr *Ctensor, input Ctensor, packedWeight Ctensor, bias Ctensor, output Ctensor){ + C.atg_fbgemm_linear_fp16_weight_out(ptr, input, packedWeight, bias, output) +} func AtgFbgemmLinearInt8Weight(ptr *Ctensor, input Ctensor, weight Ctensor, packed Ctensor, colOffsets Ctensor, weightScale Cscalar, weightZeroPoint Cscalar, bias Ctensor){ C.atg_fbgemm_linear_int8_weight(ptr, input, weight, packed, colOffsets, weightScale , weightZeroPoint , bias) } @@ -5481,6 +5854,9 @@ func AtgFloorDivideScalar(ptr *Ctensor, self Ctensor, other Cscalar){ func AtgFloorDivideScalar_(ptr *Ctensor, self Ctensor, other Cscalar){ C.atg_floor_divide_scalar_(ptr, self, other ) } +func AtgFloorDivideScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar){ + C.atg_floor_divide_scalar_out(ptr, out, self, other ) +} func AtgFloorOut(ptr *Ctensor, out Ctensor, self Ctensor){ C.atg_floor_out(ptr, out, self) } @@ -6020,6 +6396,20 @@ func AtgHardtanhBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ct func AtgHardtanhOut(ptr *Ctensor, out Ctensor, self Ctensor){ C.atg_hardtanh_out(ptr, out, self) } +func AtgHashTensor(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, mode int64){ +cdimDataPtr := (*C.int64_t)(unsafe.Pointer(&dimData[0])) +cdimLen := *(*C.int)(unsafe.Pointer(&dimLen)) +ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) +cmode := *(*C.int64_t)(unsafe.Pointer(&mode)) + C.atg_hash_tensor(ptr, self, cdimDataPtr, cdimLen, ckeepdim, cmode) +} +func AtgHashTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, mode int64){ +cdimDataPtr := (*C.int64_t)(unsafe.Pointer(&dimData[0])) +cdimLen := *(*C.int)(unsafe.Pointer(&dimLen)) +ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) +cmode := *(*C.int64_t)(unsafe.Pointer(&mode)) + C.atg_hash_tensor_out(ptr, out, self, cdimDataPtr, cdimLen, ckeepdim, cmode) +} func AtgHeaviside(ptr *Ctensor, self Ctensor, values Ctensor){ C.atg_heaviside(ptr, self, values) } @@ -6141,11 +6531,6 @@ cstrideLen := *(*C.int)(unsafe.Pointer(&strideLen)) func AtgImag(ptr *Ctensor, self Ctensor){ C.atg_imag(ptr, self) } -func AtgIndex(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int){ -cindicesDataPtr := (*Ctensor)(unsafe.Pointer(&indicesData[0])) -cindicesLen := *(*C.int)(unsafe.Pointer(&indicesLen)) - C.atg_index(ptr, self, cindicesDataPtr, cindicesLen) -} func AtgIndexAdd(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor){ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) C.atg_index_add(ptr, self, cdim, index, source) @@ -6194,24 +6579,6 @@ func AtgIndexFillIntTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64 cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) C.atg_index_fill_int_tensor_out(ptr, out, self, cdim, index, value) } -func AtgIndexPut(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32){ -cindicesDataPtr := (*Ctensor)(unsafe.Pointer(&indicesData[0])) -cindicesLen := *(*C.int)(unsafe.Pointer(&indicesLen)) -caccumulate := *(*C.int)(unsafe.Pointer(&accumulate)) - C.atg_index_put(ptr, self, cindicesDataPtr, cindicesLen, values, caccumulate) -} -func AtgIndexPut_(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32){ -cindicesDataPtr := (*Ctensor)(unsafe.Pointer(&indicesData[0])) -cindicesLen := *(*C.int)(unsafe.Pointer(&indicesLen)) -caccumulate := *(*C.int)(unsafe.Pointer(&accumulate)) - C.atg_index_put_(ptr, self, cindicesDataPtr, cindicesLen, values, caccumulate) -} -func AtgIndexPutOut(ptr *Ctensor, out Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32){ -cindicesDataPtr := (*Ctensor)(unsafe.Pointer(&indicesData[0])) -cindicesLen := *(*C.int)(unsafe.Pointer(&indicesLen)) -caccumulate := *(*C.int)(unsafe.Pointer(&accumulate)) - C.atg_index_put_out(ptr, out, self, cindicesDataPtr, cindicesLen, values, caccumulate) -} func AtgIndexReduce(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor, reduce string, includeSelf int32){ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) creduce := C.CString(reduce) @@ -6250,11 +6617,6 @@ func AtgIndexSelectOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) C.atg_index_select_out(ptr, out, self, cdim, index) } -func AtgIndexTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int){ -cindicesDataPtr := (*Ctensor)(unsafe.Pointer(&indicesData[0])) -cindicesLen := *(*C.int)(unsafe.Pointer(&indicesLen)) - C.atg_index_tensor_out(ptr, out, self, cindicesDataPtr, cindicesLen) -} func AtgIndices(ptr *Ctensor, self Ctensor){ C.atg_indices(ptr, self) } @@ -7117,6 +7479,36 @@ func AtgLinspaceOut(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar, steps csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) C.atg_linspace_out(ptr, out, start , end , csteps) } +func AtgLinspaceScalarTensor(ptr *Ctensor, start Cscalar, end Ctensor, steps int64, optionsKind int32, optionsDevice int32){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) +coptionsKind := *(*C.int)(unsafe.Pointer(&optionsKind)) +coptionsDevice := *(*C.int)(unsafe.Pointer(&optionsDevice)) + C.atg_linspace_scalar_tensor(ptr, start , end, csteps, coptionsKind, coptionsDevice) +} +func AtgLinspaceScalarTensorOut(ptr *Ctensor, out Ctensor, start Cscalar, end Ctensor, steps int64){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) + C.atg_linspace_scalar_tensor_out(ptr, out, start , end, csteps) +} +func AtgLinspaceTensorScalar(ptr *Ctensor, start Ctensor, end Cscalar, steps int64, optionsKind int32, optionsDevice int32){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) +coptionsKind := *(*C.int)(unsafe.Pointer(&optionsKind)) +coptionsDevice := *(*C.int)(unsafe.Pointer(&optionsDevice)) + C.atg_linspace_tensor_scalar(ptr, start, end , csteps, coptionsKind, coptionsDevice) +} +func AtgLinspaceTensorScalarOut(ptr *Ctensor, out Ctensor, start Ctensor, end Cscalar, steps int64){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) + C.atg_linspace_tensor_scalar_out(ptr, out, start, end , csteps) +} +func AtgLinspaceTensorTensor(ptr *Ctensor, start Ctensor, end Ctensor, steps int64, optionsKind int32, optionsDevice int32){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) +coptionsKind := *(*C.int)(unsafe.Pointer(&optionsKind)) +coptionsDevice := *(*C.int)(unsafe.Pointer(&optionsDevice)) + C.atg_linspace_tensor_tensor(ptr, start, end, csteps, coptionsKind, coptionsDevice) +} +func AtgLinspaceTensorTensorOut(ptr *Ctensor, out Ctensor, start Ctensor, end Ctensor, steps int64){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) + C.atg_linspace_tensor_tensor_out(ptr, out, start, end, csteps) +} func AtgLog(ptr *Ctensor, self Ctensor){ C.atg_log(ptr, self) } @@ -7286,6 +7678,42 @@ csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) cbase := *(*C.double)(unsafe.Pointer(&base)) C.atg_logspace_out(ptr, out, start , end , csteps, cbase) } +func AtgLogspaceScalarTensor(ptr *Ctensor, start Cscalar, end Ctensor, steps int64, base float64, optionsKind int32, optionsDevice int32){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) +cbase := *(*C.double)(unsafe.Pointer(&base)) +coptionsKind := *(*C.int)(unsafe.Pointer(&optionsKind)) +coptionsDevice := *(*C.int)(unsafe.Pointer(&optionsDevice)) + C.atg_logspace_scalar_tensor(ptr, start , end, csteps, cbase, coptionsKind, coptionsDevice) +} +func AtgLogspaceScalarTensorOut(ptr *Ctensor, out Ctensor, start Cscalar, end Ctensor, steps int64, base float64){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) +cbase := *(*C.double)(unsafe.Pointer(&base)) + C.atg_logspace_scalar_tensor_out(ptr, out, start , end, csteps, cbase) +} +func AtgLogspaceTensorScalar(ptr *Ctensor, start Ctensor, end Cscalar, steps int64, base float64, optionsKind int32, optionsDevice int32){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) +cbase := *(*C.double)(unsafe.Pointer(&base)) +coptionsKind := *(*C.int)(unsafe.Pointer(&optionsKind)) +coptionsDevice := *(*C.int)(unsafe.Pointer(&optionsDevice)) + C.atg_logspace_tensor_scalar(ptr, start, end , csteps, cbase, coptionsKind, coptionsDevice) +} +func AtgLogspaceTensorScalarOut(ptr *Ctensor, out Ctensor, start Ctensor, end Cscalar, steps int64, base float64){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) +cbase := *(*C.double)(unsafe.Pointer(&base)) + C.atg_logspace_tensor_scalar_out(ptr, out, start, end , csteps, cbase) +} +func AtgLogspaceTensorTensor(ptr *Ctensor, start Ctensor, end Ctensor, steps int64, base float64, optionsKind int32, optionsDevice int32){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) +cbase := *(*C.double)(unsafe.Pointer(&base)) +coptionsKind := *(*C.int)(unsafe.Pointer(&optionsKind)) +coptionsDevice := *(*C.int)(unsafe.Pointer(&optionsDevice)) + C.atg_logspace_tensor_tensor(ptr, start, end, csteps, cbase, coptionsKind, coptionsDevice) +} +func AtgLogspaceTensorTensorOut(ptr *Ctensor, out Ctensor, start Ctensor, end Ctensor, steps int64, base float64){ +csteps := *(*C.int64_t)(unsafe.Pointer(&steps)) +cbase := *(*C.double)(unsafe.Pointer(&base)) + C.atg_logspace_tensor_tensor_out(ptr, out, start, end, csteps, cbase) +} func AtgLogsumexp(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32){ cdimDataPtr := (*C.int64_t)(unsafe.Pointer(&dimData[0])) cdimLen := *(*C.int)(unsafe.Pointer(&dimLen)) @@ -7391,6 +7819,11 @@ func AtgMaskedScatter(ptr *Ctensor, self Ctensor, mask Ctensor, source Ctensor){ func AtgMaskedScatter_(ptr *Ctensor, self Ctensor, mask Ctensor, source Ctensor){ C.atg_masked_scatter_(ptr, self, mask, source) } +func AtgMaskedScatterBackward(ptr *Ctensor, gradOutput Ctensor, mask Ctensor, sizesData []int64, sizesLen int){ +csizesDataPtr := (*C.int64_t)(unsafe.Pointer(&sizesData[0])) +csizesLen := *(*C.int)(unsafe.Pointer(&sizesLen)) + C.atg_masked_scatter_backward(ptr, gradOutput, mask, csizesDataPtr, csizesLen) +} func AtgMaskedScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, source Ctensor){ C.atg_masked_scatter_out(ptr, out, self, mask, source) } @@ -7661,6 +8094,10 @@ ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) cdtype := *(*C.int)(unsafe.Pointer(&dtype)) C.atg_mean_dim(ptr, self, cdimDataPtr, cdimLen, ckeepdim, cdtype) } +func AtgMeanDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32){ +cdtype := *(*C.int)(unsafe.Pointer(&dtype)) + C.atg_mean_dtype_out(ptr, out, self, cdtype) +} func AtgMeanOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32){ cdimDataPtr := (*C.int64_t)(unsafe.Pointer(&dimData[0])) cdimLen := *(*C.int)(unsafe.Pointer(&dimLen)) @@ -8055,25 +8492,29 @@ cinputSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&inputSizeData[0])) cinputSizeLen := *(*C.int)(unsafe.Pointer(&inputSizeLen)) C.atg_mkldnn_reorder_conv2d_weight_out(ptr, out, self, cpaddingDataPtr, cpaddingLen, cstrideDataPtr, cstrideLen, cdilationDataPtr, cdilationLen, cgroups, cinputSizeDataPtr, cinputSizeLen) } -func AtgMkldnnReorderConv3dWeight(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64){ +func AtgMkldnnReorderConv3dWeight(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, inputSizeData []int64, inputSizeLen int){ cpaddingDataPtr := (*C.int64_t)(unsafe.Pointer(&paddingData[0])) cpaddingLen := *(*C.int)(unsafe.Pointer(&paddingLen)) cstrideDataPtr := (*C.int64_t)(unsafe.Pointer(&strideData[0])) cstrideLen := *(*C.int)(unsafe.Pointer(&strideLen)) cdilationDataPtr := (*C.int64_t)(unsafe.Pointer(&dilationData[0])) cdilationLen := *(*C.int)(unsafe.Pointer(&dilationLen)) -cgroups := *(*C.int64_t)(unsafe.Pointer(&groups)) - C.atg_mkldnn_reorder_conv3d_weight(ptr, self, cpaddingDataPtr, cpaddingLen, cstrideDataPtr, cstrideLen, cdilationDataPtr, cdilationLen, cgroups) +cgroups := *(*C.int64_t)(unsafe.Pointer(&groups)) +cinputSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&inputSizeData[0])) +cinputSizeLen := *(*C.int)(unsafe.Pointer(&inputSizeLen)) + C.atg_mkldnn_reorder_conv3d_weight(ptr, self, cpaddingDataPtr, cpaddingLen, cstrideDataPtr, cstrideLen, cdilationDataPtr, cdilationLen, cgroups, cinputSizeDataPtr, cinputSizeLen) } -func AtgMkldnnReorderConv3dWeightOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64){ +func AtgMkldnnReorderConv3dWeightOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, inputSizeData []int64, inputSizeLen int){ cpaddingDataPtr := (*C.int64_t)(unsafe.Pointer(&paddingData[0])) cpaddingLen := *(*C.int)(unsafe.Pointer(&paddingLen)) cstrideDataPtr := (*C.int64_t)(unsafe.Pointer(&strideData[0])) cstrideLen := *(*C.int)(unsafe.Pointer(&strideLen)) cdilationDataPtr := (*C.int64_t)(unsafe.Pointer(&dilationData[0])) cdilationLen := *(*C.int)(unsafe.Pointer(&dilationLen)) -cgroups := *(*C.int64_t)(unsafe.Pointer(&groups)) - C.atg_mkldnn_reorder_conv3d_weight_out(ptr, out, self, cpaddingDataPtr, cpaddingLen, cstrideDataPtr, cstrideLen, cdilationDataPtr, cdilationLen, cgroups) +cgroups := *(*C.int64_t)(unsafe.Pointer(&groups)) +cinputSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&inputSizeData[0])) +cinputSizeLen := *(*C.int)(unsafe.Pointer(&inputSizeLen)) + C.atg_mkldnn_reorder_conv3d_weight_out(ptr, out, self, cpaddingDataPtr, cpaddingLen, cstrideDataPtr, cstrideLen, cdilationDataPtr, cdilationLen, cgroups, cinputSizeDataPtr, cinputSizeLen) } func AtgMkldnnRnnLayer(ptr *Ctensor, input Ctensor, weight0 Ctensor, weight1 Ctensor, weight2 Ctensor, weight3 Ctensor, hx_ Ctensor, cx_ Ctensor, reverse int32, batchSizesData []int64, batchSizesLen int, mode int64, hiddenSize int64, numLayers int64, hasBiases int32, bidirectional int32, batchFirst int32, train int32){ creverse := *(*C.int)(unsafe.Pointer(&reverse)) @@ -8130,6 +8571,14 @@ ctrain := *(*C.int)(unsafe.Pointer(&train)) func AtgMm(ptr *Ctensor, self Ctensor, mat2 Ctensor){ C.atg_mm(ptr, self, mat2) } +func AtgMmDtype(ptr *Ctensor, self Ctensor, mat2 Ctensor, outDtype int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg_mm_dtype(ptr, self, mat2, coutDtype) +} +func AtgMmDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor, outDtype int32){ +coutDtype := *(*C.int)(unsafe.Pointer(&outDtype)) + C.atg_mm_dtype_out(ptr, out, self, mat2, coutDtype) +} func AtgMmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor){ C.atg_mm_out(ptr, out, self, mat2) } @@ -8822,12 +9271,15 @@ cvalueVal := *(*C.double)(unsafe.Pointer(&valueVal)) cvalueNull := *(*C.uint8_t)(unsafe.Pointer(&valueNull)) C.atg_pad(ptr, self, cpadDataPtr, cpadLen, cmode, cmodeLen, cvalueVal, cvalueNull) } -func AtgPadSequence(ptr *Ctensor, sequencesData []Ctensor, sequencesLen int, batchFirst int32, paddingValue float64){ +func AtgPadSequence(ptr *Ctensor, sequencesData []Ctensor, sequencesLen int, batchFirst int32, paddingValue float64, paddingSide string){ csequencesDataPtr := (*Ctensor)(unsafe.Pointer(&sequencesData[0])) csequencesLen := *(*C.int)(unsafe.Pointer(&sequencesLen)) cbatchFirst := *(*C.int)(unsafe.Pointer(&batchFirst)) -cpaddingValue := *(*C.double)(unsafe.Pointer(&paddingValue)) - C.atg_pad_sequence(ptr, csequencesDataPtr, csequencesLen, cbatchFirst, cpaddingValue) +cpaddingValue := *(*C.double)(unsafe.Pointer(&paddingValue)) +cpaddingSide := C.CString(paddingSide) +paddingSideLen := len(paddingSide) +cpaddingSideLen := *(*C.int)(unsafe.Pointer(&paddingSideLen)) + C.atg_pad_sequence(ptr, csequencesDataPtr, csequencesLen, cbatchFirst, cpaddingValue, cpaddingSide, cpaddingSideLen) } func AtgPairwiseDistance(ptr *Ctensor, x1 Ctensor, x2 Ctensor, p float64, eps float64, keepdim int32){ cp := *(*C.double)(unsafe.Pointer(&p)) @@ -9223,6 +9675,12 @@ func AtgRandintLikeOut(ptr *Ctensor, out Ctensor, self Ctensor, high int64){ chigh := *(*C.int64_t)(unsafe.Pointer(&high)) C.atg_randint_like_out(ptr, out, self, chigh) } +func AtgRandintLikeTensor(ptr *Ctensor, self Ctensor, high Ctensor){ + C.atg_randint_like_tensor(ptr, self, high) +} +func AtgRandintLikeTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, high Ctensor){ + C.atg_randint_like_tensor_out(ptr, out, self, high) +} func AtgRandintLow(ptr *Ctensor, low int64, high int64, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32){ clow := *(*C.int64_t)(unsafe.Pointer(&low)) chigh := *(*C.int64_t)(unsafe.Pointer(&high)) @@ -9606,6 +10064,13 @@ func AtgRetainsGrad(self Ctensor) bool{ if cbool == 1{return true} return false } +func AtgRmsNorm(ptr *Ctensor, input Ctensor, normalizedShapeData []int64, normalizedShapeLen int, weight Ctensor, epsVal float64, epsNull int){ +cnormalizedShapeDataPtr := (*C.int64_t)(unsafe.Pointer(&normalizedShapeData[0])) +cnormalizedShapeLen := *(*C.int)(unsafe.Pointer(&normalizedShapeLen)) +cepsVal := *(*C.double)(unsafe.Pointer(&epsVal)) +cepsNull := *(*C.uint8_t)(unsafe.Pointer(&epsNull)) + C.atg_rms_norm(ptr, input, cnormalizedShapeDataPtr, cnormalizedShapeLen, weight, cepsVal, cepsNull) +} func AtgRnnRelu(ptr *Ctensor, input Ctensor, hx Ctensor, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32, batchFirst int32){ cparamsDataPtr := (*Ctensor)(unsafe.Pointer(¶msData[0])) cparamsLen := *(*C.int)(unsafe.Pointer(¶msLen)) @@ -9746,6 +10211,10 @@ ctraining := *(*C.int)(unsafe.Pointer(&training)) cselfIsResult := *(*C.int)(unsafe.Pointer(&selfIsResult)) C.atg_rrelu_with_noise_backward_out(ptr, out, gradOutput, self, noise, lower , upper , ctraining, cselfIsResult) } +func AtgRreluWithNoiseFunctional(ptr *Ctensor, self Ctensor, noise Ctensor, training int32){ +ctraining := *(*C.int)(unsafe.Pointer(&training)) + C.atg_rrelu_with_noise_functional(ptr, self, noise, ctraining) +} func AtgRreluWithNoiseOut(ptr *Ctensor, out Ctensor, self Ctensor, noise Ctensor, training int32){ ctraining := *(*C.int)(unsafe.Pointer(&training)) C.atg_rrelu_with_noise_out(ptr, out, self, noise, ctraining) @@ -9779,12 +10248,13 @@ coptionsDevice := *(*C.int)(unsafe.Pointer(&optionsDevice)) func AtgScalarTensorOut(ptr *Ctensor, out Ctensor, s Cscalar){ C.atg_scalar_tensor_out(ptr, out, s ) } -func AtgScaledDotProductAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, isCausal int32, scaleVal float64, scaleNull int){ +func AtgScaledDotProductAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, isCausal int32, scaleVal float64, scaleNull int, enableGqa int32){ cdropoutP := *(*C.double)(unsafe.Pointer(&dropoutP)) cisCausal := *(*C.int)(unsafe.Pointer(&isCausal)) cscaleVal := *(*C.double)(unsafe.Pointer(&scaleVal)) -cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) - C.atg_scaled_dot_product_attention(ptr, query, key, value, attnMask, cdropoutP, cisCausal, cscaleVal, cscaleNull) +cscaleNull := *(*C.uint8_t)(unsafe.Pointer(&scaleNull)) +cenableGqa := *(*C.int)(unsafe.Pointer(&enableGqa)) + C.atg_scaled_dot_product_attention(ptr, query, key, value, attnMask, cdropoutP, cisCausal, cscaleVal, cscaleNull, cenableGqa) } func AtgScatter(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor){ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) @@ -10113,6 +10583,15 @@ cendNull := *(*C.uint8_t)(unsafe.Pointer(&endNull)) cstep := *(*C.int64_t)(unsafe.Pointer(&step)) C.atg_slice_copy_tensor_out(ptr, out, self, cdim, cstartVal, cstartNull, cendVal, cendNull, cstep) } +func AtgSliceInverse(ptr *Ctensor, self Ctensor, src Ctensor, dim int64, startVal int64, startNull int, endVal int64, endNull int, step int64){ +cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) +cstartVal := *(*C.int64_t)(unsafe.Pointer(&startVal)) +cstartNull := *(*C.uint8_t)(unsafe.Pointer(&startNull)) +cendVal := *(*C.int64_t)(unsafe.Pointer(&endVal)) +cendNull := *(*C.uint8_t)(unsafe.Pointer(&endNull)) +cstep := *(*C.int64_t)(unsafe.Pointer(&step)) + C.atg_slice_inverse(ptr, self, src, cdim, cstartVal, cstartNull, cendVal, cendNull, cstep) +} func AtgSliceScatter(ptr *Ctensor, self Ctensor, src Ctensor, dim int64, startVal int64, startNull int, endVal int64, endNull int, step int64){ cdim := *(*C.int64_t)(unsafe.Pointer(&dim)) cstartVal := *(*C.int64_t)(unsafe.Pointer(&startVal)) @@ -11162,7 +11641,7 @@ cunbiased := *(*C.int)(unsafe.Pointer(&unbiased)) ckeepdim := *(*C.int)(unsafe.Pointer(&keepdim)) C.atg_std_out(ptr, out, self, cdimDataPtr, cdimLen, cunbiased, ckeepdim) } -func AtgStft(ptr *Ctensor, self Ctensor, nFft int64, hopLengthVal int64, hopLengthNull int, winLengthVal int64, winLengthNull int, window Ctensor, normalized int32, onesided int32, returnComplex int32){ +func AtgStft(ptr *Ctensor, self Ctensor, nFft int64, hopLengthVal int64, hopLengthNull int, winLengthVal int64, winLengthNull int, window Ctensor, normalized int32, onesided int32, returnComplex int32, alignToWindow int32){ cnFft := *(*C.int64_t)(unsafe.Pointer(&nFft)) chopLengthVal := *(*C.int64_t)(unsafe.Pointer(&hopLengthVal)) chopLengthNull := *(*C.uint8_t)(unsafe.Pointer(&hopLengthNull)) @@ -11170,10 +11649,11 @@ cwinLengthVal := *(*C.int64_t)(unsafe.Pointer(&winLengthVal)) cwinLengthNull := *(*C.uint8_t)(unsafe.Pointer(&winLengthNull)) cnormalized := *(*C.int)(unsafe.Pointer(&normalized)) conesided := *(*C.int)(unsafe.Pointer(&onesided)) -creturnComplex := *(*C.int)(unsafe.Pointer(&returnComplex)) - C.atg_stft(ptr, self, cnFft, chopLengthVal, chopLengthNull, cwinLengthVal, cwinLengthNull, window, cnormalized, conesided, creturnComplex) +creturnComplex := *(*C.int)(unsafe.Pointer(&returnComplex)) +calignToWindow := *(*C.int)(unsafe.Pointer(&alignToWindow)) + C.atg_stft(ptr, self, cnFft, chopLengthVal, chopLengthNull, cwinLengthVal, cwinLengthNull, window, cnormalized, conesided, creturnComplex, calignToWindow) } -func AtgStftCenter(ptr *Ctensor, self Ctensor, nFft int64, hopLengthVal int64, hopLengthNull int, winLengthVal int64, winLengthNull int, window Ctensor, center int32, padMode string, normalized int32, onesided int32, returnComplex int32){ +func AtgStftCenter(ptr *Ctensor, self Ctensor, nFft int64, hopLengthVal int64, hopLengthNull int, winLengthVal int64, winLengthNull int, window Ctensor, center int32, padMode string, normalized int32, onesided int32, returnComplex int32, alignToWindow int32){ cnFft := *(*C.int64_t)(unsafe.Pointer(&nFft)) chopLengthVal := *(*C.int64_t)(unsafe.Pointer(&hopLengthVal)) chopLengthNull := *(*C.uint8_t)(unsafe.Pointer(&hopLengthNull)) @@ -11185,8 +11665,9 @@ padModeLen := len(padMode) cpadModeLen := *(*C.int)(unsafe.Pointer(&padModeLen)) cnormalized := *(*C.int)(unsafe.Pointer(&normalized)) conesided := *(*C.int)(unsafe.Pointer(&onesided)) -creturnComplex := *(*C.int)(unsafe.Pointer(&returnComplex)) - C.atg_stft_center(ptr, self, cnFft, chopLengthVal, chopLengthNull, cwinLengthVal, cwinLengthNull, window, ccenter, cpadMode, cpadModeLen, cnormalized, conesided, creturnComplex) +creturnComplex := *(*C.int)(unsafe.Pointer(&returnComplex)) +calignToWindow := *(*C.int)(unsafe.Pointer(&alignToWindow)) + C.atg_stft_center(ptr, self, cnFft, chopLengthVal, chopLengthNull, cwinLengthVal, cwinLengthNull, window, ccenter, cpadMode, cpadModeLen, cnormalized, conesided, creturnComplex, calignToWindow) } func AtgSub(ptr *Ctensor, self Ctensor, other Ctensor){ C.atg_sub(ptr, self, other) @@ -11852,6 +12333,14 @@ cscaleFactorsDataPtr := (*C.double)(unsafe.Pointer(&scaleFactorsData[0])) cscaleFactorsLen := *(*C.int)(unsafe.Pointer(&scaleFactorsLen)) C.atg_upsample_bilinear2d_vec(ptr, input, coutputSizeDataPtr, coutputSizeLen, calignCorners, cscaleFactorsDataPtr, cscaleFactorsLen) } +func AtgUpsampleBilinear2dVecOut(ptr *Ctensor, out Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scaleFactorsData []float64, scaleFactorsLen int){ +coutputSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&outputSizeData[0])) +coutputSizeLen := *(*C.int)(unsafe.Pointer(&outputSizeLen)) +calignCorners := *(*C.int)(unsafe.Pointer(&alignCorners)) +cscaleFactorsDataPtr := (*C.double)(unsafe.Pointer(&scaleFactorsData[0])) +cscaleFactorsLen := *(*C.int)(unsafe.Pointer(&scaleFactorsLen)) + C.atg_upsample_bilinear2d_vec_out(ptr, out, input, coutputSizeDataPtr, coutputSizeLen, calignCorners, cscaleFactorsDataPtr, cscaleFactorsLen) +} func AtgUpsampleLinear1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesVal float64, scalesNull int){ coutputSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&outputSizeData[0])) coutputSizeLen := *(*C.int)(unsafe.Pointer(&outputSizeLen)) @@ -11982,6 +12471,13 @@ cscaleFactorsDataPtr := (*C.double)(unsafe.Pointer(&scaleFactorsData[0])) cscaleFactorsLen := *(*C.int)(unsafe.Pointer(&scaleFactorsLen)) C.atg_upsample_nearest2d_vec(ptr, input, coutputSizeDataPtr, coutputSizeLen, cscaleFactorsDataPtr, cscaleFactorsLen) } +func AtgUpsampleNearest2dVecOut(ptr *Ctensor, out Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, scaleFactorsData []float64, scaleFactorsLen int){ +coutputSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&outputSizeData[0])) +coutputSizeLen := *(*C.int)(unsafe.Pointer(&outputSizeLen)) +cscaleFactorsDataPtr := (*C.double)(unsafe.Pointer(&scaleFactorsData[0])) +cscaleFactorsLen := *(*C.int)(unsafe.Pointer(&scaleFactorsLen)) + C.atg_upsample_nearest2d_vec_out(ptr, out, input, coutputSizeDataPtr, coutputSizeLen, cscaleFactorsDataPtr, cscaleFactorsLen) +} func AtgUpsampleNearest3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int){ coutputSizeDataPtr := (*C.int64_t)(unsafe.Pointer(&outputSizeData[0])) coutputSizeLen := *(*C.int)(unsafe.Pointer(&outputSizeLen)) diff --git a/libtch/lib.go b/libtch/lib.go index b67e375..81e19aa 100644 --- a/libtch/lib.go +++ b/libtch/lib.go @@ -1,9 +1,9 @@ package libtch // #cgo CFLAGS: -I${SRCDIR} -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function -// #cgo CFLAGS: -I/usr/local/cuda/include -// #cgo LDFLAGS: -lstdc++ -ltorch -ltorch_cpu -ltorch_cuda -lc10 -// #cgo LDFLAGS: -L/usr/local/cuda/lib64 -lcuda -// #cgo CFLAGS: -D_GLIBCXX_USE_CXX11_ABI=1 +// #cgo CFLAGS: -I/Users/tlc/src/gotch/libtorch-2.10.0-macos/include/torch/csrc/api/include +// #cgo CFLAGS: -I/Users/tlc/src/gotch/libtorch-2.10.0-macos/include +// #cgo LDFLAGS: -L/Users/tlc/src/gotch/libtorch-2.10.0-macos/lib -ltorch -ltorch_cpu -lc10 +// #cgo LDFLAGS: -Wl,-rpath,/Users/tlc/src/gotch/libtorch-2.10.0-macos/lib // #cgo CXXFLAGS: -std=c++17 import "C" diff --git a/libtch/tensor.go b/libtch/tensor.go index c238b63..2945abe 100644 --- a/libtch/tensor.go +++ b/libtch/tensor.go @@ -152,6 +152,12 @@ func AtcCudaIsAvailable() bool { return *(*bool)(unsafe.Pointer(&result)) } +// int atc_mps_is_available(); +func AtcMpsIsAvailable() bool { + result := C.atc_mps_is_available() + return *(*bool)(unsafe.Pointer(&result)) +} + // int atc_cudnn_is_available(); func AtcCudnnIsAvailable() bool { result := C.atc_cudnn_is_available() diff --git a/libtch/torch_api.cpp b/libtch/torch_api.cpp index 7800d12..b5fe25b 100644 --- a/libtch/torch_api.cpp +++ b/libtch/torch_api.cpp @@ -49,7 +49,7 @@ c10::List> of_carray_tensor_opt(torch::Tensor **vs, at::Device device_of_int(int d) { if (d == -3) return at::Device(at::kVulkan); - // if (d == -2) return at::Device(at::kMPS); + if (d == -2) return at::Device(at::kMPS); if (d < 0) return at::Device(at::kCPU); return at::Device(at::kCUDA, /*index=*/d); @@ -757,6 +757,11 @@ int atc_cuda_is_available() { return -1; } +int atc_mps_is_available() { + PROTECT(return torch::mps::is_available();) + return -1; +} + int atc_cudnn_is_available() { PROTECT(return torch::cuda::cudnn_is_available();) return -1; diff --git a/libtch/torch_api.h b/libtch/torch_api.h index 8ac7cb5..8465917 100644 --- a/libtch/torch_api.h +++ b/libtch/torch_api.h @@ -155,6 +155,7 @@ void ats_free(scalar); int atc_cuda_device_count(); int atc_cuda_is_available(); +int atc_mps_is_available(); int atc_cudnn_is_available(); void atc_set_benchmark_cudnn(int b); void atc_synchronize(int64_t device_index); diff --git a/libtch/torch_api_generated.cpp.h b/libtch/torch_api_generated.cpp.h index f74a993..3ac3160 100644 --- a/libtch/torch_api_generated.cpp.h +++ b/libtch/torch_api_generated.cpp.h @@ -341,6 +341,12 @@ void atg__amp_update_scale_out(tensor *out__, tensor out, tensor self, tensor gr ) } +void atg__assert_scalar(scalar self_scalar, char* assert_msg_ptr, int assert_msg_len) { + PROTECT( + torch::_assert_scalar(*self_scalar, std::string(assert_msg_ptr, assert_msg_len)); + ) +} + void atg__autocast_to_full_precision(tensor *out__, tensor self, int cuda_enabled, int cpu_enabled) { PROTECT( auto outputs__ = self->_autocast_to_full_precision((bool)cuda_enabled, (bool)cpu_enabled); @@ -355,6 +361,58 @@ void atg__autocast_to_reduced_precision(tensor *out__, tensor self, int cuda_ena ) } +void atg__batch_norm_no_update(tensor *out__, tensor input, tensor weight, tensor bias, tensor running_mean, tensor running_var, double momentum, double eps) { + PROTECT( + auto outputs__ = torch::_batch_norm_no_update(*input, (weight ? *weight : torch::Tensor()), (bias ? *bias : torch::Tensor()), (running_mean ? *running_mean : torch::Tensor()), (running_var ? *running_var : torch::Tensor()), momentum, eps); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + out__[2] = new torch::Tensor(std::get<2>(outputs__)); + out__[3] = new torch::Tensor(std::get<3>(outputs__)); + ) +} + +void atg__batch_norm_no_update_out(tensor *out__, tensor out0, tensor out1, tensor out2, tensor out3, tensor input, tensor weight, tensor bias, tensor running_mean, tensor running_var, double momentum, double eps) { + PROTECT( + auto outputs__ = torch::_batch_norm_no_update_out(*out0, *out1, *out2, *out3, *input, (weight ? *weight : torch::Tensor()), (bias ? *bias : torch::Tensor()), (running_mean ? *running_mean : torch::Tensor()), (running_var ? *running_var : torch::Tensor()), momentum, eps); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + out__[2] = new torch::Tensor(std::get<2>(outputs__)); + out__[3] = new torch::Tensor(std::get<3>(outputs__)); + ) +} + +void atg__batch_norm_with_update(tensor *out__, tensor input, tensor weight, tensor bias, tensor running_mean, tensor running_var, double momentum, double eps) { + PROTECT( + auto outputs__ = torch::_batch_norm_with_update(*input, (weight ? *weight : torch::Tensor()), (bias ? *bias : torch::Tensor()), *running_mean, *running_var, momentum, eps); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + out__[2] = new torch::Tensor(std::get<2>(outputs__)); + out__[3] = new torch::Tensor(std::get<3>(outputs__)); + ) +} + +void atg__batch_norm_with_update_functional(tensor *out__, tensor input, tensor weight, tensor bias, tensor running_mean, tensor running_var, double momentum, double eps) { + PROTECT( + auto outputs__ = torch::_batch_norm_with_update_functional(*input, (weight ? *weight : torch::Tensor()), (bias ? *bias : torch::Tensor()), *running_mean, *running_var, momentum, eps); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + out__[2] = new torch::Tensor(std::get<2>(outputs__)); + out__[3] = new torch::Tensor(std::get<3>(outputs__)); + out__[4] = new torch::Tensor(std::get<4>(outputs__)); + out__[5] = new torch::Tensor(std::get<5>(outputs__)); + ) +} + +void atg__batch_norm_with_update_out(tensor *out__, tensor out, tensor save_mean, tensor save_invstd, tensor reserve, tensor input, tensor weight, tensor bias, tensor running_mean, tensor running_var, double momentum, double eps) { + PROTECT( + auto outputs__ = torch::_batch_norm_with_update_out(*out, *save_mean, *save_invstd, *reserve, *input, (weight ? *weight : torch::Tensor()), (bias ? *bias : torch::Tensor()), *running_mean, *running_var, momentum, eps); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + out__[2] = new torch::Tensor(std::get<2>(outputs__)); + out__[3] = new torch::Tensor(std::get<3>(outputs__)); + ) +} + void atg__cast_byte(tensor *out__, tensor self, int non_blocking) { PROTECT( auto outputs__ = torch::_cast_Byte(*self, (bool)non_blocking); @@ -439,6 +497,20 @@ void atg__cholesky_solve_helper_out(tensor *out__, tensor out, tensor self, tens ) } +void atg__chunk_cat(tensor *out__, tensor *tensors_data, int tensors_len, int64_t dim, int64_t num_chunks) { + PROTECT( + auto outputs__ = torch::_chunk_cat(of_carray_tensor(tensors_data, tensors_len), dim, num_chunks); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__chunk_cat_out(tensor *out__, tensor out, tensor *tensors_data, int tensors_len, int64_t dim, int64_t num_chunks) { + PROTECT( + auto outputs__ = torch::_chunk_cat_out(*out, of_carray_tensor(tensors_data, tensors_len), dim, num_chunks); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__coalesce(tensor *out__, tensor self) { PROTECT( auto outputs__ = torch::_coalesce(*self); @@ -565,6 +637,20 @@ void atg__convert_indices_from_csr_to_coo_out(tensor *out__, tensor out, tensor ) } +void atg__convert_weight_to_int4pack(tensor *out__, tensor self, int64_t innerKTiles) { + PROTECT( + auto outputs__ = torch::_convert_weight_to_int4pack(*self, innerKTiles); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__convert_weight_to_int4pack_for_cpu(tensor *out__, tensor self, int64_t innerKTiles) { + PROTECT( + auto outputs__ = torch::_convert_weight_to_int4pack_for_cpu(*self, innerKTiles); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__convolution(tensor *out__, tensor input, tensor weight, tensor bias, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int64_t *dilation_data, int dilation_len, int transposed, int64_t *output_padding_data, int output_padding_len, int64_t groups, int benchmark, int deterministic, int cudnn_enabled, int allow_tf32) { PROTECT( auto outputs__ = torch::_convolution(*input, *weight, (bias ? *bias : torch::Tensor()), torch::IntArrayRef(stride_data, stride_len), torch::IntArrayRef(padding_data, padding_len), torch::IntArrayRef(dilation_data, dilation_len), (bool)transposed, torch::IntArrayRef(output_padding_data, output_padding_len), groups, (bool)benchmark, (bool)deterministic, (bool)cudnn_enabled, (bool)allow_tf32); @@ -628,13 +714,20 @@ void atg__cslt_compress(tensor *out__, tensor input) { ) } -void atg__cslt_sparse_mm(tensor *out__, tensor compressed_A, tensor dense_B, tensor bias, int transpose_result) { +void atg__cslt_sparse_mm(tensor *out__, tensor compressed_A, tensor dense_B, tensor bias, tensor alpha, int out_dtype, int transpose_result, int64_t alg_id, int64_t split_k, int64_t split_k_mode) { PROTECT( - auto outputs__ = torch::_cslt_sparse_mm(*compressed_A, *dense_B, (bias ? *bias : torch::Tensor()), (bool)transpose_result); + auto outputs__ = torch::_cslt_sparse_mm(*compressed_A, *dense_B, (bias ? *bias : torch::Tensor()), (alpha ? *alpha : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype)), (bool)transpose_result, alg_id, split_k, split_k_mode); out__[0] = new torch::Tensor(outputs__); ) } +int64_t atg__cslt_sparse_mm_search(tensor compressed_A, tensor dense_B, tensor bias, tensor alpha, int out_dtype, int transpose_result) { + PROTECT( + return torch::_cslt_sparse_mm_search(*compressed_A, *dense_B, (bias ? *bias : torch::Tensor()), (alpha ? *alpha : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype)), (bool)transpose_result); + ) + return 0; +} + void atg__ctc_loss(tensor *out__, tensor log_probs, tensor targets, int64_t *input_lengths_data, int input_lengths_len, int64_t *target_lengths_data, int target_lengths_len, int64_t blank, int zero_infinity) { PROTECT( auto outputs__ = torch::_ctc_loss(*log_probs, *targets, torch::IntArrayRef(input_lengths_data, input_lengths_len), torch::IntArrayRef(target_lengths_data, target_lengths_len), blank, (bool)zero_infinity); @@ -688,6 +781,15 @@ void atg__ctc_loss_tensor_out(tensor *out__, tensor out0, tensor out1, tensor lo ) } +void atg__cudnn_attention_backward(tensor *out__, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, tensor philox_seed, tensor philox_offset, tensor attn_bias, tensor cum_seq_q, tensor cum_seq_k, int64_t max_q, int64_t max_k, double dropout_p, int is_causal, double scale_v, uint8_t scale_null) { + PROTECT( + auto outputs__ = torch::_cudnn_attention_backward(*grad_out, *query, *key, *value, *out, *logsumexp, *philox_seed, *philox_offset, *attn_bias, *cum_seq_q, *cum_seq_k, max_q, max_k, dropout_p, (bool)is_causal, scale_null ? c10::nullopt : c10::optional(scale_v)); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + out__[2] = new torch::Tensor(std::get<2>(outputs__)); + ) +} + void atg__cudnn_ctc_loss(tensor *out__, tensor log_probs, tensor targets, int64_t *input_lengths_data, int input_lengths_len, int64_t *target_lengths_data, int target_lengths_len, int64_t blank, int deterministic, int zero_infinity) { PROTECT( auto outputs__ = torch::_cudnn_ctc_loss(*log_probs, *targets, torch::IntArrayRef(input_lengths_data, input_lengths_len), torch::IntArrayRef(target_lengths_data, target_lengths_len), blank, (bool)deterministic, (bool)zero_infinity); @@ -804,9 +906,23 @@ void atg__dirichlet_grad_out(tensor *out__, tensor out, tensor x, tensor alpha, ) } -void atg__efficient_attention_backward(tensor *out__, tensor grad_out_, tensor query, tensor key, tensor value, tensor bias, tensor out, tensor cu_seqlens_q, tensor cu_seqlens_k, int64_t max_seqlen_k, int64_t max_seqlen_q, tensor logsumexp, double dropout_p, tensor philox_seed, tensor philox_offset, int64_t custom_mask_type, int bias_requires_grad, double scale_v, uint8_t scale_null, int64_t num_splits_key_v, uint8_t num_splits_key_null) { +void atg__dyn_quant_matmul_4bit(tensor *out__, tensor inp, tensor packed_weights, int64_t block_size, int64_t in_features, int64_t out_features) { PROTECT( - auto outputs__ = torch::_efficient_attention_backward(*grad_out_, *query, *key, *value, (bias ? *bias : torch::Tensor()), *out, (cu_seqlens_q ? *cu_seqlens_q : torch::Tensor()), (cu_seqlens_k ? *cu_seqlens_k : torch::Tensor()), max_seqlen_k, max_seqlen_q, *logsumexp, dropout_p, *philox_seed, *philox_offset, custom_mask_type, (bool)bias_requires_grad, scale_null ? c10::nullopt : c10::optional(scale_v), num_splits_key_null ? c10::nullopt : c10::optional(num_splits_key_v)); + auto outputs__ = torch::_dyn_quant_matmul_4bit(*inp, *packed_weights, block_size, in_features, out_features); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__dyn_quant_pack_4bit_weight(tensor *out__, tensor weights, tensor scales_zeros, tensor bias, int64_t block_size, int64_t in_features, int64_t out_features) { + PROTECT( + auto outputs__ = torch::_dyn_quant_pack_4bit_weight(*weights, *scales_zeros, (bias ? *bias : torch::Tensor()), block_size, in_features, out_features); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__efficient_attention_backward(tensor *out__, tensor grad_out_, tensor query, tensor key, tensor value, tensor bias, tensor out, tensor cu_seqlens_q, tensor cu_seqlens_k, int64_t max_seqlen_q, int64_t max_seqlen_k, tensor logsumexp, double dropout_p, tensor philox_seed, tensor philox_offset, int64_t custom_mask_type, int bias_requires_grad, double scale_v, uint8_t scale_null, int64_t num_splits_key_v, uint8_t num_splits_key_null, int64_t window_size_v, uint8_t window_size_null, int shared_storage_dqdkdv) { + PROTECT( + auto outputs__ = torch::_efficient_attention_backward(*grad_out_, *query, *key, *value, (bias ? *bias : torch::Tensor()), *out, (cu_seqlens_q ? *cu_seqlens_q : torch::Tensor()), (cu_seqlens_k ? *cu_seqlens_k : torch::Tensor()), max_seqlen_q, max_seqlen_k, *logsumexp, dropout_p, *philox_seed, *philox_offset, custom_mask_type, (bool)bias_requires_grad, scale_null ? c10::nullopt : c10::optional(scale_v), num_splits_key_null ? c10::nullopt : c10::optional(num_splits_key_v), window_size_null ? c10::nullopt : c10::optional(window_size_v), (bool)shared_storage_dqdkdv); out__[0] = new torch::Tensor(std::get<0>(outputs__)); out__[1] = new torch::Tensor(std::get<1>(outputs__)); out__[2] = new torch::Tensor(std::get<2>(outputs__)); @@ -1063,9 +1179,9 @@ void atg__fill_mem_eff_dropout_mask_(tensor *out__, tensor self, double dropout_ ) } -void atg__flash_attention_backward(tensor *out__, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, tensor cum_seq_q, tensor cum_seq_k, int64_t max_q, int64_t max_k, double dropout_p, int is_causal, tensor philox_seed, tensor philox_offset, double scale_v, uint8_t scale_null) { +void atg__flash_attention_backward(tensor *out__, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, tensor cum_seq_q, tensor cum_seq_k, int64_t max_q, int64_t max_k, double dropout_p, int is_causal, tensor rng_state, tensor unused, double scale_v, uint8_t scale_null, int64_t window_size_left_v, uint8_t window_size_left_null, int64_t window_size_right_v, uint8_t window_size_right_null) { PROTECT( - auto outputs__ = torch::_flash_attention_backward(*grad_out, *query, *key, *value, *out, *logsumexp, *cum_seq_q, *cum_seq_k, max_q, max_k, dropout_p, (bool)is_causal, *philox_seed, *philox_offset, scale_null ? c10::nullopt : c10::optional(scale_v)); + auto outputs__ = torch::_flash_attention_backward(*grad_out, *query, *key, *value, *out, *logsumexp, *cum_seq_q, *cum_seq_k, max_q, max_k, dropout_p, (bool)is_causal, *rng_state, *unused, scale_null ? c10::nullopt : c10::optional(scale_v), window_size_left_null ? c10::nullopt : c10::optional(window_size_left_v), window_size_right_null ? c10::nullopt : c10::optional(window_size_right_v)); out__[0] = new torch::Tensor(std::get<0>(outputs__)); out__[1] = new torch::Tensor(std::get<1>(outputs__)); out__[2] = new torch::Tensor(std::get<2>(outputs__)); @@ -1093,6 +1209,13 @@ void atg__functional_assert_async(tensor *out__, tensor self, char* assert_msg_p ) } +void atg__functional_assert_scalar(tensor *out__, scalar self_scalar, char* assert_msg_ptr, int assert_msg_len, tensor dep_token) { + PROTECT( + auto outputs__ = torch::_functional_assert_scalar(*self_scalar, std::string(assert_msg_ptr, assert_msg_len), *dep_token); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__functional_sym_constrain_range(tensor *out__, scalar size, int64_t min_v, uint8_t min_null, int64_t max_v, uint8_t max_null, tensor dep_token) { PROTECT( auto outputs__ = torch::_functional_sym_constrain_range(*size, min_null ? c10::nullopt : c10::optional(min_v), max_null ? c10::nullopt : c10::optional(max_v), *dep_token); @@ -1107,6 +1230,30 @@ void atg__functional_sym_constrain_range_for_size(tensor *out__, scalar size, in ) } +void atg__fused_adagrad(tensor *out_data, int out_len, tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *state_sums_data, int state_sums_len, tensor *state_steps_data, int state_steps_len, double lr, double lr_decay, double weight_decay, double eps, int maximize, tensor grad_scale, tensor found_inf) { + PROTECT( + torch::_fused_adagrad_out(of_carray_tensor(out_data, out_len), of_carray_tensor(self_data, self_len), of_carray_tensor(grads_data, grads_len), of_carray_tensor(state_sums_data, state_sums_len), of_carray_tensor(state_steps_data, state_steps_len), lr, lr_decay, weight_decay, eps, (bool)maximize, (grad_scale ? *grad_scale : torch::Tensor()), (found_inf ? *found_inf : torch::Tensor())); + ) +} + +void atg__fused_adagrad_(tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *state_sums_data, int state_sums_len, tensor *state_steps_data, int state_steps_len, double lr, double lr_decay, double weight_decay, double eps, int maximize, tensor grad_scale, tensor found_inf) { + PROTECT( + torch::_fused_adagrad_(of_carray_tensor(self_data, self_len), of_carray_tensor(grads_data, grads_len), of_carray_tensor(state_sums_data, state_sums_len), of_carray_tensor(state_steps_data, state_steps_len), lr, lr_decay, weight_decay, eps, (bool)maximize, (grad_scale ? *grad_scale : torch::Tensor()), (found_inf ? *found_inf : torch::Tensor())); + ) +} + +void atg__fused_adagrad_tensor_lr_(tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *state_sums_data, int state_sums_len, tensor *state_steps_data, int state_steps_len, tensor lr, double lr_decay, double weight_decay, double eps, int maximize, tensor grad_scale, tensor found_inf) { + PROTECT( + torch::_fused_adagrad_(of_carray_tensor(self_data, self_len), of_carray_tensor(grads_data, grads_len), of_carray_tensor(state_sums_data, state_sums_len), of_carray_tensor(state_steps_data, state_steps_len), *lr, lr_decay, weight_decay, eps, (bool)maximize, (grad_scale ? *grad_scale : torch::Tensor()), (found_inf ? *found_inf : torch::Tensor())); + ) +} + +void atg__fused_adagrad_tensor_lr_out(tensor *out_data, int out_len, tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *state_sums_data, int state_sums_len, tensor *state_steps_data, int state_steps_len, tensor lr, double lr_decay, double weight_decay, double eps, int maximize, tensor grad_scale, tensor found_inf) { + PROTECT( + torch::_fused_adagrad_out(of_carray_tensor(out_data, out_len), of_carray_tensor(self_data, self_len), of_carray_tensor(grads_data, grads_len), of_carray_tensor(state_sums_data, state_sums_len), of_carray_tensor(state_steps_data, state_steps_len), *lr, lr_decay, weight_decay, eps, (bool)maximize, (grad_scale ? *grad_scale : torch::Tensor()), (found_inf ? *found_inf : torch::Tensor())); + ) +} + void atg__fused_dropout(tensor *out__, tensor self, double p) { PROTECT( auto outputs__ = torch::_fused_dropout(*self, p); @@ -1151,13 +1298,45 @@ void atg__fused_moving_avg_obs_fq_helper_out(tensor *out__, tensor out0, tensor ) } -int64_t atg__fused_sdp_choice(tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, double scale_v, uint8_t scale_null) { +void atg__fused_rms_norm(tensor *out__, tensor input, int64_t *normalized_shape_data, int normalized_shape_len, tensor weight, double eps_v, uint8_t eps_null) { + PROTECT( + auto outputs__ = torch::_fused_rms_norm(*input, torch::IntArrayRef(normalized_shape_data, normalized_shape_len), (weight ? *weight : torch::Tensor()), eps_null ? c10::nullopt : c10::optional(eps_v)); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + ) +} + +int64_t atg__fused_sdp_choice(tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, double scale_v, uint8_t scale_null, int enable_gqa) { PROTECT( - return torch::_fused_sdp_choice(*query, *key, *value, (attn_mask ? *attn_mask : torch::Tensor()), dropout_p, (bool)is_causal, scale_null ? c10::nullopt : c10::optional(scale_v)); + return torch::_fused_sdp_choice(*query, *key, *value, (attn_mask ? *attn_mask : torch::Tensor()), dropout_p, (bool)is_causal, scale_null ? c10::nullopt : c10::optional(scale_v), (bool)enable_gqa); ) return 0; } +void atg__fused_sgd(tensor *out_data, int out_len, tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *momentum_buffer_list_data, int momentum_buffer_list_len, double weight_decay, double momentum, double lr, double dampening, int nesterov, int maximize, int is_first_step, tensor grad_scale, tensor found_inf) { + PROTECT( + torch::_fused_sgd_out(of_carray_tensor(out_data, out_len), of_carray_tensor(self_data, self_len), of_carray_tensor(grads_data, grads_len), of_carray_tensor(momentum_buffer_list_data, momentum_buffer_list_len), weight_decay, momentum, lr, dampening, (bool)nesterov, (bool)maximize, (bool)is_first_step, (grad_scale ? *grad_scale : torch::Tensor()), (found_inf ? *found_inf : torch::Tensor())); + ) +} + +void atg__fused_sgd_(tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *momentum_buffer_list_data, int momentum_buffer_list_len, double weight_decay, double momentum, double lr, double dampening, int nesterov, int maximize, int is_first_step, tensor grad_scale, tensor found_inf) { + PROTECT( + torch::_fused_sgd_(of_carray_tensor(self_data, self_len), of_carray_tensor(grads_data, grads_len), of_carray_tensor(momentum_buffer_list_data, momentum_buffer_list_len), weight_decay, momentum, lr, dampening, (bool)nesterov, (bool)maximize, (bool)is_first_step, (grad_scale ? *grad_scale : torch::Tensor()), (found_inf ? *found_inf : torch::Tensor())); + ) +} + +void atg__fused_sgd_tensor_lr_(tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *momentum_buffer_list_data, int momentum_buffer_list_len, double weight_decay, double momentum, tensor lr, double dampening, int nesterov, int maximize, int is_first_step, tensor grad_scale, tensor found_inf) { + PROTECT( + torch::_fused_sgd_(of_carray_tensor(self_data, self_len), of_carray_tensor(grads_data, grads_len), of_carray_tensor(momentum_buffer_list_data, momentum_buffer_list_len), weight_decay, momentum, *lr, dampening, (bool)nesterov, (bool)maximize, (bool)is_first_step, (grad_scale ? *grad_scale : torch::Tensor()), (found_inf ? *found_inf : torch::Tensor())); + ) +} + +void atg__fused_sgd_tensor_lr_out(tensor *out_data, int out_len, tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *momentum_buffer_list_data, int momentum_buffer_list_len, double weight_decay, double momentum, tensor lr, double dampening, int nesterov, int maximize, int is_first_step, tensor grad_scale, tensor found_inf) { + PROTECT( + torch::_fused_sgd_out(of_carray_tensor(out_data, out_len), of_carray_tensor(self_data, self_len), of_carray_tensor(grads_data, grads_len), of_carray_tensor(momentum_buffer_list_data, momentum_buffer_list_len), weight_decay, momentum, *lr, dampening, (bool)nesterov, (bool)maximize, (bool)is_first_step, (grad_scale ? *grad_scale : torch::Tensor()), (found_inf ? *found_inf : torch::Tensor())); + ) +} + void atg__fw_primal(tensor *out__, tensor self, int64_t level) { PROTECT( auto outputs__ = self->_fw_primal(level); @@ -1208,6 +1387,13 @@ void atg__grid_sampler_2d_cpu_fallback_out(tensor *out__, tensor out, tensor inp ) } +void atg__grouped_mm(tensor *out__, tensor self, tensor mat2, tensor offs, tensor bias, int out_dtype) { + PROTECT( + auto outputs__ = torch::_grouped_mm(*self, *mat2, (offs ? *offs : torch::Tensor()), (bias ? *bias : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype))); + out__[0] = new torch::Tensor(outputs__); + ) +} + int atg__has_compatible_shallow_copy_type(tensor self, tensor from) { PROTECT( return torch::_has_compatible_shallow_copy_type(*self, *from); @@ -1243,27 +1429,6 @@ void atg__histogramdd_from_bin_tensors_out(tensor *out__, tensor out, tensor sel ) } -void atg__index_put_impl(tensor *out__, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate, int unsafe) { - PROTECT( - auto outputs__ = torch::_index_put_impl(*self, of_carray_tensor_opt(indices_data, indices_len), *values, (bool)accumulate, (bool)unsafe); - out__[0] = new torch::Tensor(outputs__); - ) -} - -void atg__index_put_impl_(tensor *out__, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate, int unsafe) { - PROTECT( - auto outputs__ = torch::_index_put_impl_(*self, of_carray_tensor_opt(indices_data, indices_len), *values, (bool)accumulate, (bool)unsafe); - out__[0] = new torch::Tensor(outputs__); - ) -} - -void atg__index_put_impl_out(tensor *out__, tensor out, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate, int unsafe) { - PROTECT( - auto outputs__ = torch::_index_put_impl_out(*out, *self, of_carray_tensor_opt(indices_data, indices_len), *values, (bool)accumulate, (bool)unsafe); - out__[0] = new torch::Tensor(outputs__); - ) -} - void atg__indices(tensor *out__, tensor self) { PROTECT( auto outputs__ = self->_indices(); @@ -1320,6 +1485,13 @@ int atg__is_zerotensor(tensor self) { return 0; } +void atg__lazy_clone(tensor *out__, tensor self) { + PROTECT( + auto outputs__ = torch::_lazy_clone(*self); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__linalg_det(tensor *out__, tensor A) { PROTECT( auto outputs__ = torch::_linalg_det(*A); @@ -1354,6 +1526,13 @@ void atg__linalg_eigh_eigenvalues(tensor *out__, tensor eigenvalues, tensor eige ) } +void atg__linalg_eigvals(tensor *out__, tensor self) { + PROTECT( + auto outputs__ = torch::_linalg_eigvals(*self); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__linalg_slogdet(tensor *out__, tensor A) { PROTECT( auto outputs__ = torch::_linalg_slogdet(*A); @@ -1585,6 +1764,13 @@ void atg__masked_softmax_out(tensor *out__, tensor out, tensor self, tensor mask ) } +void atg__mixed_dtypes_linear(tensor *out__, tensor input, tensor weight, tensor scale, tensor bias, char* activation_ptr, int activation_len) { + PROTECT( + auto outputs__ = torch::_mixed_dtypes_linear(*input, *weight, *scale, (bias ? *bias : torch::Tensor()), std::string(activation_ptr, activation_len)); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__mkldnn_reshape(tensor *out__, tensor self, int64_t *shape_data, int shape_len) { PROTECT( auto outputs__ = torch::_mkldnn_reshape(*self, torch::IntArrayRef(shape_data, shape_len)); @@ -1750,6 +1936,14 @@ void atg__neg_view_copy_out(tensor *out__, tensor out, tensor self) { ) } +void atg__nested_compute_contiguous_strides_offsets(tensor *out__, tensor nested_size) { + PROTECT( + auto outputs__ = torch::_nested_compute_contiguous_strides_offsets(*nested_size); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + ) +} + void atg__nested_from_padded(tensor *out__, tensor padded, tensor cpu_nested_shape_example, int fuse_transform_0213) { PROTECT( auto outputs__ = torch::_nested_from_padded(*padded, *cpu_nested_shape_example, (bool)fuse_transform_0213); @@ -1778,6 +1972,76 @@ void atg__nested_from_padded_out(tensor *out__, tensor out, tensor padded, tenso ) } +void atg__nested_from_padded_tensor(tensor *out__, tensor padded, tensor offsets, tensor dummy, int64_t ragged_idx, tensor min_seqlen, tensor max_seqlen, int64_t sum_S_v, uint8_t sum_S_null) { + PROTECT( + auto outputs__ = torch::_nested_from_padded_tensor(*padded, *offsets, *dummy, ragged_idx, (min_seqlen ? *min_seqlen : torch::Tensor()), (max_seqlen ? *max_seqlen : torch::Tensor()), sum_S_null ? c10::nullopt : c10::optional(sum_S_v)); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__nested_get_jagged_dummy(tensor *out__, tensor any) { + PROTECT( + auto outputs__ = torch::_nested_get_jagged_dummy(*any); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__nested_get_lengths(tensor *out__, tensor self) { + PROTECT( + auto outputs__ = torch::_nested_get_lengths(*self); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__nested_get_max_seqlen(tensor *out__, tensor self) { + PROTECT( + auto outputs__ = torch::_nested_get_max_seqlen(*self); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__nested_get_min_seqlen(tensor *out__, tensor self) { + PROTECT( + auto outputs__ = torch::_nested_get_min_seqlen(*self); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__nested_get_offsets(tensor *out__, tensor self) { + PROTECT( + auto outputs__ = torch::_nested_get_offsets(*self); + out__[0] = new torch::Tensor(outputs__); + ) +} + +int64_t atg__nested_get_ragged_idx(tensor self) { + PROTECT( + return torch::_nested_get_ragged_idx(*self); + ) + return 0; +} + +void atg__nested_get_values(tensor *out__, tensor self) { + PROTECT( + auto outputs__ = torch::_nested_get_values(*self); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__nested_get_values_copy(tensor *out__, tensor self) { + PROTECT( + auto outputs__ = torch::_nested_get_values_copy(*self); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__nested_get_values_copy_out(tensor *out__, tensor out, tensor self) { + PROTECT( + auto outputs__ = torch::_nested_get_values_copy_out(*out, *self); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__nested_select_backward(tensor *out__, tensor grad_output, tensor self, int64_t dim, int64_t index) { PROTECT( auto outputs__ = torch::_nested_select_backward(*grad_output, *self, dim, index); @@ -1813,6 +2077,27 @@ void atg__nested_view_from_buffer_copy_out(tensor *out__, tensor out, tensor sel ) } +void atg__nested_view_from_jagged(tensor *out__, tensor self, tensor offsets, tensor dummy, tensor lengths, int64_t ragged_idx, tensor min_seqlen, tensor max_seqlen) { + PROTECT( + auto outputs__ = torch::_nested_view_from_jagged(*self, *offsets, *dummy, (lengths ? *lengths : torch::Tensor()), ragged_idx, (min_seqlen ? *min_seqlen : torch::Tensor()), (max_seqlen ? *max_seqlen : torch::Tensor())); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__nested_view_from_jagged_copy(tensor *out__, tensor self, tensor offsets, tensor dummy, tensor lengths, int64_t ragged_idx, tensor min_seqlen, tensor max_seqlen) { + PROTECT( + auto outputs__ = torch::_nested_view_from_jagged_copy(*self, *offsets, *dummy, (lengths ? *lengths : torch::Tensor()), ragged_idx, (min_seqlen ? *min_seqlen : torch::Tensor()), (max_seqlen ? *max_seqlen : torch::Tensor())); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__nested_view_from_jagged_copy_out(tensor *out__, tensor out, tensor self, tensor offsets, tensor dummy, tensor lengths, int64_t ragged_idx, tensor min_seqlen, tensor max_seqlen) { + PROTECT( + auto outputs__ = torch::_nested_view_from_jagged_copy_out(*out, *self, *offsets, *dummy, (lengths ? *lengths : torch::Tensor()), ragged_idx, (min_seqlen ? *min_seqlen : torch::Tensor()), (max_seqlen ? *max_seqlen : torch::Tensor())); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__new_zeros_with_same_feature_meta(tensor *out__, tensor self, tensor other, int64_t self_num_batch_dims) { PROTECT( auto outputs__ = torch::_new_zeros_with_same_feature_meta(*self, *other, self_num_batch_dims); @@ -1943,6 +2228,12 @@ void atg__prelu_kernel_backward(tensor *out__, tensor grad_output, tensor self, ) } +void atg__print(char* s_ptr, int s_len) { + PROTECT( + torch::_print(std::string(s_ptr, s_len)); + ) +} + void atg__propagate_xla_data(tensor input, tensor output) { PROTECT( torch::_propagate_xla_data(*input, *output); @@ -2020,6 +2311,13 @@ void atg__rowwise_prune(tensor *out__, tensor weight, tensor mask, int compresse ) } +void atg__safe_softmax(tensor *out__, tensor self, int64_t dim, int dtype) { + PROTECT( + auto outputs__ = torch::_safe_softmax(*self, dim, dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(dtype))); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__sample_dirichlet(tensor *out__, tensor self) { PROTECT( auto outputs__ = torch::_sample_dirichlet(*self); @@ -2041,14 +2339,31 @@ void atg__saturate_weight_to_fp16(tensor *out__, tensor weight) { ) } -void atg__scaled_dot_product_attention_math(tensor *out__, tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, tensor dropout_mask, double scale_v, uint8_t scale_null) { +void atg__scaled_dot_product_attention_math(tensor *out__, tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, tensor dropout_mask, double scale_v, uint8_t scale_null, int enable_gqa) { + PROTECT( + auto outputs__ = torch::_scaled_dot_product_attention_math(*query, *key, *value, (attn_mask ? *attn_mask : torch::Tensor()), dropout_p, (bool)is_causal, (dropout_mask ? *dropout_mask : torch::Tensor()), scale_null ? c10::nullopt : c10::optional(scale_v), (bool)enable_gqa); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + ) +} + +void atg__scaled_dot_product_attention_math_for_mps(tensor *out__, tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, tensor dropout_mask, double scale_v, uint8_t scale_null) { PROTECT( - auto outputs__ = torch::_scaled_dot_product_attention_math(*query, *key, *value, (attn_mask ? *attn_mask : torch::Tensor()), dropout_p, (bool)is_causal, (dropout_mask ? *dropout_mask : torch::Tensor()), scale_null ? c10::nullopt : c10::optional(scale_v)); + auto outputs__ = torch::_scaled_dot_product_attention_math_for_mps(*query, *key, *value, (attn_mask ? *attn_mask : torch::Tensor()), dropout_p, (bool)is_causal, (dropout_mask ? *dropout_mask : torch::Tensor()), scale_null ? c10::nullopt : c10::optional(scale_v)); out__[0] = new torch::Tensor(std::get<0>(outputs__)); out__[1] = new torch::Tensor(std::get<1>(outputs__)); ) } +void atg__scaled_dot_product_cudnn_attention_backward(tensor *out__, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, tensor philox_seed, tensor philox_offset, tensor attn_bias, tensor cum_seq_q, tensor cum_seq_k, int64_t max_q, int64_t max_k, double dropout_p, int is_causal, double scale_v, uint8_t scale_null) { + PROTECT( + auto outputs__ = torch::_scaled_dot_product_cudnn_attention_backward(*grad_out, *query, *key, *value, *out, *logsumexp, *philox_seed, *philox_offset, *attn_bias, *cum_seq_q, *cum_seq_k, max_q, max_k, dropout_p, (bool)is_causal, scale_null ? c10::nullopt : c10::optional(scale_v)); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + out__[2] = new torch::Tensor(std::get<2>(outputs__)); + ) +} + void atg__scaled_dot_product_efficient_attention(tensor *out__, tensor query, tensor key, tensor value, tensor attn_bias, int compute_log_sumexp, double dropout_p, int is_causal, double scale_v, uint8_t scale_null) { PROTECT( auto outputs__ = torch::_scaled_dot_product_efficient_attention(*query, *key, *value, (attn_bias ? *attn_bias : torch::Tensor()), (bool)compute_log_sumexp, dropout_p, (bool)is_causal, scale_null ? c10::nullopt : c10::optional(scale_v)); @@ -2068,19 +2383,62 @@ void atg__scaled_dot_product_flash_attention_backward(tensor *out__, tensor grad ) } -void atg__scaled_mm(tensor *out__, tensor self, tensor mat2, tensor bias, int out_dtype, tensor scale_a, tensor scale_b, tensor scale_result) { +void atg__scaled_dot_product_flash_attention_for_cpu(tensor *out__, tensor query, tensor key, tensor value, double dropout_p, int is_causal, tensor attn_mask, double scale_v, uint8_t scale_null) { PROTECT( - auto outputs__ = torch::_scaled_mm(*self, *mat2, (bias ? *bias : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype)), (scale_a ? *scale_a : torch::Tensor()), (scale_b ? *scale_b : torch::Tensor()), (scale_result ? *scale_result : torch::Tensor())); + auto outputs__ = torch::_scaled_dot_product_flash_attention_for_cpu(*query, *key, *value, dropout_p, (bool)is_causal, (attn_mask ? *attn_mask : torch::Tensor()), scale_null ? c10::nullopt : c10::optional(scale_v)); out__[0] = new torch::Tensor(std::get<0>(outputs__)); out__[1] = new torch::Tensor(std::get<1>(outputs__)); ) } -void atg__scaled_mm_out(tensor *out__, tensor out, tensor out_amax, tensor self, tensor mat2, tensor bias, int out_dtype, tensor scale_a, tensor scale_b, tensor scale_result) { +void atg__scaled_dot_product_flash_attention_for_cpu_backward(tensor *out__, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, double dropout_p, int is_causal, tensor attn_mask, double scale_v, uint8_t scale_null) { PROTECT( - auto outputs__ = torch::_scaled_mm_out(*out, *out_amax, *self, *mat2, (bias ? *bias : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype)), (scale_a ? *scale_a : torch::Tensor()), (scale_b ? *scale_b : torch::Tensor()), (scale_result ? *scale_result : torch::Tensor())); + auto outputs__ = torch::_scaled_dot_product_flash_attention_for_cpu_backward(*grad_out, *query, *key, *value, *out, *logsumexp, dropout_p, (bool)is_causal, (attn_mask ? *attn_mask : torch::Tensor()), scale_null ? c10::nullopt : c10::optional(scale_v)); out__[0] = new torch::Tensor(std::get<0>(outputs__)); out__[1] = new torch::Tensor(std::get<1>(outputs__)); + out__[2] = new torch::Tensor(std::get<2>(outputs__)); + ) +} + +void atg__scaled_grouped_mm(tensor *out__, tensor self, tensor mat2, tensor scale_a, tensor scale_b, tensor offs, tensor bias, tensor scale_result, int out_dtype, int use_fast_accum) { + PROTECT( + auto outputs__ = torch::_scaled_grouped_mm(*self, *mat2, *scale_a, *scale_b, (offs ? *offs : torch::Tensor()), (bias ? *bias : torch::Tensor()), (scale_result ? *scale_result : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype)), (bool)use_fast_accum); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__scaled_grouped_mm_v2(tensor *out__, tensor self, tensor mat2, tensor *scale_a_data, int scale_a_len, int64_t *recipe_a_data, int recipe_a_len, int64_t *swizzle_a_data, int swizzle_a_len, tensor *scale_b_data, int scale_b_len, int64_t *recipe_b_data, int recipe_b_len, int64_t *swizzle_b_data, int swizzle_b_len, tensor offs, tensor bias, int out_dtype, int64_t *contraction_dim_data, int contraction_dim_len, int use_fast_accum) { + PROTECT( + auto outputs__ = torch::_scaled_grouped_mm_v2(*self, *mat2, of_carray_tensor(scale_a_data, scale_a_len), torch::IntArrayRef(recipe_a_data, recipe_a_len), torch::IntArrayRef(swizzle_a_data, swizzle_a_len), of_carray_tensor(scale_b_data, scale_b_len), torch::IntArrayRef(recipe_b_data, recipe_b_len), torch::IntArrayRef(swizzle_b_data, swizzle_b_len), (offs ? *offs : torch::Tensor()), (bias ? *bias : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype)), torch::IntArrayRef(contraction_dim_data, contraction_dim_len), (bool)use_fast_accum); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__scaled_mm(tensor *out__, tensor self, tensor mat2, tensor scale_a, tensor scale_b, tensor bias, tensor scale_result, int out_dtype, int use_fast_accum) { + PROTECT( + auto outputs__ = torch::_scaled_mm(*self, *mat2, *scale_a, *scale_b, (bias ? *bias : torch::Tensor()), (scale_result ? *scale_result : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype)), (bool)use_fast_accum); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__scaled_mm_out(tensor *out__, tensor out, tensor self, tensor mat2, tensor scale_a, tensor scale_b, tensor bias, tensor scale_result, int out_dtype, int use_fast_accum) { + PROTECT( + auto outputs__ = torch::_scaled_mm_out(*out, *self, *mat2, *scale_a, *scale_b, (bias ? *bias : torch::Tensor()), (scale_result ? *scale_result : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype)), (bool)use_fast_accum); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__scaled_mm_v2(tensor *out__, tensor self, tensor mat2, tensor *scale_a_data, int scale_a_len, int64_t *recipe_a_data, int recipe_a_len, int64_t *swizzle_a_data, int swizzle_a_len, tensor *scale_b_data, int scale_b_len, int64_t *recipe_b_data, int recipe_b_len, int64_t *swizzle_b_data, int swizzle_b_len, tensor bias, int out_dtype, int64_t *contraction_dim_data, int contraction_dim_len, int use_fast_accum) { + PROTECT( + auto outputs__ = torch::_scaled_mm_v2(*self, *mat2, of_carray_tensor(scale_a_data, scale_a_len), torch::IntArrayRef(recipe_a_data, recipe_a_len), torch::IntArrayRef(swizzle_a_data, swizzle_a_len), of_carray_tensor(scale_b_data, scale_b_len), torch::IntArrayRef(recipe_b_data, recipe_b_len), torch::IntArrayRef(swizzle_b_data, swizzle_b_len), (bias ? *bias : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype)), torch::IntArrayRef(contraction_dim_data, contraction_dim_len), (bool)use_fast_accum); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__scaled_mm_v2_out(tensor *out__, tensor out, tensor self, tensor mat2, tensor *scale_a_data, int scale_a_len, int64_t *recipe_a_data, int recipe_a_len, int64_t *swizzle_a_data, int swizzle_a_len, tensor *scale_b_data, int scale_b_len, int64_t *recipe_b_data, int recipe_b_len, int64_t *swizzle_b_data, int swizzle_b_len, tensor bias, int out_dtype, int64_t *contraction_dim_data, int contraction_dim_len, int use_fast_accum) { + PROTECT( + auto outputs__ = torch::_scaled_mm_v2_out(*out, *self, *mat2, of_carray_tensor(scale_a_data, scale_a_len), torch::IntArrayRef(recipe_a_data, recipe_a_len), torch::IntArrayRef(swizzle_a_data, swizzle_a_len), of_carray_tensor(scale_b_data, scale_b_len), torch::IntArrayRef(recipe_b_data, recipe_b_len), torch::IntArrayRef(swizzle_b_data, swizzle_b_len), (bias ? *bias : torch::Tensor()), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype)), torch::IntArrayRef(contraction_dim_data, contraction_dim_len), (bool)use_fast_accum); + out__[0] = new torch::Tensor(outputs__); ) } @@ -2248,6 +2606,13 @@ void atg__sparse_compressed_tensor_unsafe(tensor *out__, tensor compressed_indic ) } +void atg__sparse_compressed_tensor_with_dims(tensor *out__, int64_t nnz, int64_t dense_dim, int64_t *size_data, int size_len, int64_t *blocksize_data, int blocksize_len, int index_dtype, int options_kind, int options_device) { + PROTECT( + auto outputs__ = torch::_sparse_compressed_tensor_with_dims(nnz, dense_dim, torch::IntArrayRef(size_data, size_len), torch::IntArrayRef(blocksize_data, blocksize_len), at::ScalarType(index_dtype), at::device(device_of_int(options_device)).dtype(at::ScalarType(options_kind))); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__sparse_coo_tensor_unsafe(tensor *out__, tensor indices, tensor values, int64_t *size_data, int size_len, int options_kind, int options_device, int is_coalesced) { PROTECT( auto outputs__ = torch::_sparse_coo_tensor_unsafe(*indices, *values, torch::IntArrayRef(size_data, size_len), at::device(device_of_int(options_device)).dtype(at::ScalarType(options_kind)), (bool)is_coalesced); @@ -2396,13 +2761,46 @@ void atg__sparse_mm_reduce_impl(tensor *out__, tensor self, tensor other, char* ) } -void atg__sparse_semi_structured_linear(tensor *out__, tensor input, tensor weight, tensor meta, tensor bias, char* activation_ptr, int activation_len) { +void atg__sparse_semi_structured_apply(tensor *out__, tensor input, tensor thread_masks) { + PROTECT( + auto outputs__ = torch::_sparse_semi_structured_apply(*input, *thread_masks); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + ) +} + +void atg__sparse_semi_structured_apply_dense(tensor *out__, tensor input, tensor thread_masks) { + PROTECT( + auto outputs__ = torch::_sparse_semi_structured_apply_dense(*input, *thread_masks); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__sparse_semi_structured_linear(tensor *out__, tensor input, tensor weight, tensor meta, tensor bias, char* activation_ptr, int activation_len, int out_dtype) { PROTECT( - auto outputs__ = torch::_sparse_semi_structured_linear(*input, *weight, *meta, (bias ? *bias : torch::Tensor()), std::string(activation_ptr, activation_len)); + auto outputs__ = torch::_sparse_semi_structured_linear(*input, *weight, *meta, (bias ? *bias : torch::Tensor()), std::string(activation_ptr, activation_len), out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype))); out__[0] = new torch::Tensor(outputs__); ) } +void atg__sparse_semi_structured_mm(tensor *out__, tensor mat1, tensor mat1_meta, tensor mat2, int out_dtype) { + PROTECT( + auto outputs__ = torch::_sparse_semi_structured_mm(*mat1, *mat1_meta, *mat2, out_dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(out_dtype))); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__sparse_semi_structured_tile(tensor *out__, tensor input, char* algorithm_ptr, int algorithm_len, int use_cutlass) { + PROTECT( + auto outputs__ = torch::_sparse_semi_structured_tile(*input, std::string(algorithm_ptr, algorithm_len), (bool)use_cutlass); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + out__[2] = new torch::Tensor(std::get<2>(outputs__)); + out__[3] = new torch::Tensor(std::get<3>(outputs__)); + out__[4] = new torch::Tensor(std::get<4>(outputs__)); + ) +} + void atg__sparse_softmax(tensor *out__, tensor self, int64_t dim, int half_to_float) { PROTECT( auto outputs__ = torch::_sparse_softmax(*self, dim, (bool)half_to_float); @@ -2515,6 +2913,13 @@ void atg__spdiags_out(tensor *out__, tensor out, tensor diagonals, tensor offset ) } +void atg__spsolve(tensor *out__, tensor A, tensor B, int left) { + PROTECT( + auto outputs__ = torch::_spsolve(*A, *B, (bool)left); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__stack(tensor *out__, tensor *tensors_data, int tensors_len, int64_t dim) { PROTECT( auto outputs__ = torch::_stack(of_carray_tensor(tensors_data, tensors_len), dim); @@ -2676,6 +3081,13 @@ void atg__test_optional_intlist_out(tensor *out__, tensor out, tensor values, in ) } +void atg__test_parallel_materialize(tensor *out__, tensor self, int64_t num_parallel, int skip_first) { + PROTECT( + auto outputs__ = torch::_test_parallel_materialize(*self, num_parallel, (bool)skip_first); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg__test_serialization_subcmul(tensor *out__, tensor self, tensor other) { PROTECT( auto outputs__ = torch::_test_serialization_subcmul(*self, *other); @@ -2953,20 +3365,6 @@ void atg__unpack_dual(tensor *out__, tensor dual, int64_t level) { ) } -void atg__unsafe_index(tensor *out__, tensor self, tensor *indices_data, int indices_len) { - PROTECT( - auto outputs__ = torch::_unsafe_index(*self, of_carray_tensor_opt(indices_data, indices_len)); - out__[0] = new torch::Tensor(outputs__); - ) -} - -void atg__unsafe_index_put(tensor *out__, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate) { - PROTECT( - auto outputs__ = torch::_unsafe_index_put(*self, of_carray_tensor_opt(indices_data, indices_len), *values, (bool)accumulate); - out__[0] = new torch::Tensor(outputs__); - ) -} - void atg__unsafe_view(tensor *out__, tensor self, int64_t *size_data, int size_len) { PROTECT( auto outputs__ = torch::_unsafe_view(*self, torch::IntArrayRef(size_data, size_len)); @@ -3186,23 +3584,51 @@ void atg__values(tensor *out__, tensor self) { void atg__values_copy(tensor *out__, tensor self) { PROTECT( - auto outputs__ = torch::_values_copy(*self); + auto outputs__ = torch::_values_copy(*self); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__values_copy_out(tensor *out__, tensor out, tensor self) { + PROTECT( + auto outputs__ = torch::_values_copy_out(*out, *self); + out__[0] = new torch::Tensor(outputs__); + ) +} + +int64_t atg__version(tensor self) { + PROTECT( + return self->_version(); + ) + return 0; +} + +void atg__weight_int4pack_mm(tensor *out__, tensor self, tensor mat2, int64_t qGroupSize, tensor qScaleAndZeros) { + PROTECT( + auto outputs__ = torch::_weight_int4pack_mm(*self, *mat2, qGroupSize, *qScaleAndZeros); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__weight_int4pack_mm_for_cpu(tensor *out__, tensor self, tensor mat2, int64_t qGroupSize, tensor qScaleAndZeros) { + PROTECT( + auto outputs__ = torch::_weight_int4pack_mm_for_cpu(*self, *mat2, qGroupSize, *qScaleAndZeros); out__[0] = new torch::Tensor(outputs__); ) } -void atg__values_copy_out(tensor *out__, tensor out, tensor self) { +void atg__weight_int4pack_mm_with_scales_and_zeros(tensor *out__, tensor self, tensor mat2, int64_t qGroupSize, tensor qScale, tensor qZeros) { PROTECT( - auto outputs__ = torch::_values_copy_out(*out, *self); + auto outputs__ = torch::_weight_int4pack_mm_with_scales_and_zeros(*self, *mat2, qGroupSize, *qScale, *qZeros); out__[0] = new torch::Tensor(outputs__); ) } -int64_t atg__version(tensor self) { +void atg__weight_int8pack_mm(tensor *out__, tensor self, tensor mat2, tensor scales) { PROTECT( - return self->_version(); + auto outputs__ = torch::_weight_int8pack_mm(*self, *mat2, *scales); + out__[0] = new torch::Tensor(outputs__); ) - return 0; } void atg__weight_norm(tensor *out__, tensor v, tensor g, int64_t dim) { @@ -3252,6 +3678,20 @@ void atg__weight_norm_interface_out(tensor *out__, tensor out0, tensor out1, ten ) } +void atg__wrapped_linear_prepack(tensor *out__, tensor weight, tensor weight_scale, tensor weight_zero_point, tensor bias) { + PROTECT( + auto outputs__ = torch::_wrapped_linear_prepack(*weight, *weight_scale, *weight_zero_point, *bias); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg__wrapped_quantized_linear_prepacked(tensor *out__, tensor input, tensor input_scale, tensor input_zero_point, tensor packed_weight, tensor output_scale, tensor output_zero_point, int64_t out_channel) { + PROTECT( + auto outputs__ = torch::_wrapped_quantized_linear_prepacked(*input, *input_scale, *input_zero_point, *packed_weight, *output_scale, *output_zero_point, out_channel); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_abs(tensor *out__, tensor self) { PROTECT( auto outputs__ = torch::abs(*self); @@ -3343,6 +3783,13 @@ void atg_adaptive_avg_pool1d(tensor *out__, tensor self, int64_t *output_size_da ) } +void atg_adaptive_avg_pool1d_out(tensor *out__, tensor out, tensor self, int64_t *output_size_data, int output_size_len) { + PROTECT( + auto outputs__ = torch::adaptive_avg_pool1d_out(*out, *self, torch::IntArrayRef(output_size_data, output_size_len)); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_adaptive_avg_pool2d(tensor *out__, tensor self, int64_t *output_size_data, int output_size_len) { PROTECT( auto outputs__ = torch::adaptive_avg_pool2d(*self, torch::IntArrayRef(output_size_data, output_size_len)); @@ -3565,6 +4012,20 @@ void atg_addmm_(tensor *out__, tensor self, tensor mat1, tensor mat2) { ) } +void atg_addmm_dtype(tensor *out__, tensor self, tensor mat1, tensor mat2, int out_dtype) { + PROTECT( + auto outputs__ = torch::addmm(*self, *mat1, *mat2, at::ScalarType(out_dtype)); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_addmm_dtype_out(tensor *out__, tensor out, tensor self, tensor mat1, tensor mat2, int out_dtype) { + PROTECT( + auto outputs__ = torch::addmm_out(*out, *self, *mat1, *mat2, at::ScalarType(out_dtype)); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_addmm_out(tensor *out__, tensor out, tensor self, tensor mat1, tensor mat2) { PROTECT( auto outputs__ = torch::addmm_out(*out, *self, *mat1, *mat2); @@ -3704,6 +4165,20 @@ void atg_all_dim(tensor *out__, tensor self, int64_t dim, int keepdim) { ) } +void atg_all_dims(tensor *out__, tensor self, int64_t *dim_data, int dim_len, int keepdim) { + PROTECT( + auto outputs__ = torch::all(*self, dim_data == nullptr ? c10::nullopt : c10::optional(torch::IntArrayRef(dim_data, dim_len)), (bool)keepdim); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_all_dims_out(tensor *out__, tensor out, tensor self, int64_t *dim_data, int dim_len, int keepdim) { + PROTECT( + auto outputs__ = torch::all_out(*out, *self, dim_data == nullptr ? c10::nullopt : c10::optional(torch::IntArrayRef(dim_data, dim_len)), (bool)keepdim); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_all_out(tensor *out__, tensor out, tensor self, int64_t dim, int keepdim) { PROTECT( auto outputs__ = torch::all_out(*out, *self, dim, (bool)keepdim); @@ -3811,6 +4286,20 @@ void atg_any_dim(tensor *out__, tensor self, int64_t dim, int keepdim) { ) } +void atg_any_dims(tensor *out__, tensor self, int64_t *dim_data, int dim_len, int keepdim) { + PROTECT( + auto outputs__ = torch::any(*self, dim_data == nullptr ? c10::nullopt : c10::optional(torch::IntArrayRef(dim_data, dim_len)), (bool)keepdim); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_any_dims_out(tensor *out__, tensor out, tensor self, int64_t *dim_data, int dim_len, int keepdim) { + PROTECT( + auto outputs__ = torch::any_out(*out, *self, dim_data == nullptr ? c10::nullopt : c10::optional(torch::IntArrayRef(dim_data, dim_len)), (bool)keepdim); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_any_out(tensor *out__, tensor out, tensor self, int64_t dim, int keepdim) { PROTECT( auto outputs__ = torch::any_out(*out, *self, dim, (bool)keepdim); @@ -4256,6 +4745,13 @@ void atg_avg_pool1d(tensor *out__, tensor self, int64_t *kernel_size_data, int k ) } +void atg_avg_pool1d_out(tensor *out__, tensor out, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int ceil_mode, int count_include_pad) { + PROTECT( + auto outputs__ = torch::avg_pool1d_out(*out, *self, torch::IntArrayRef(kernel_size_data, kernel_size_len), torch::IntArrayRef(stride_data, stride_len), torch::IntArrayRef(padding_data, padding_len), (bool)ceil_mode, (bool)count_include_pad); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_avg_pool2d(tensor *out__, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int ceil_mode, int count_include_pad, int64_t divisor_override_v, uint8_t divisor_override_null) { PROTECT( auto outputs__ = torch::avg_pool2d(*self, torch::IntArrayRef(kernel_size_data, kernel_size_len), torch::IntArrayRef(stride_data, stride_len), torch::IntArrayRef(padding_data, padding_len), (bool)ceil_mode, (bool)count_include_pad, divisor_override_null ? c10::nullopt : c10::optional(divisor_override_v)); @@ -4326,6 +4822,20 @@ void atg_baddbmm_(tensor *out__, tensor self, tensor batch1, tensor batch2) { ) } +void atg_baddbmm_dtype(tensor *out__, tensor self, tensor batch1, tensor batch2, int out_dtype, scalar beta, scalar alpha) { + PROTECT( + auto outputs__ = torch::baddbmm(*self, *batch1, *batch2, at::ScalarType(out_dtype), *beta, *alpha); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_baddbmm_dtype_out(tensor *out__, tensor out, tensor self, tensor batch1, tensor batch2, int out_dtype) { + PROTECT( + auto outputs__ = torch::baddbmm_out(*out, *self, *batch1, *batch2, at::ScalarType(out_dtype)); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_baddbmm_out(tensor *out__, tensor out, tensor self, tensor batch1, tensor batch2) { PROTECT( auto outputs__ = torch::baddbmm_out(*out, *self, *batch1, *batch2); @@ -4942,6 +5452,20 @@ void atg_bmm(tensor *out__, tensor self, tensor mat2) { ) } +void atg_bmm_dtype(tensor *out__, tensor self, tensor mat2, int out_dtype) { + PROTECT( + auto outputs__ = torch::bmm(*self, *mat2, at::ScalarType(out_dtype)); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_bmm_dtype_out(tensor *out__, tensor out, tensor self, tensor mat2, int out_dtype) { + PROTECT( + auto outputs__ = torch::bmm_out(*out, *self, *mat2, at::ScalarType(out_dtype)); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_bmm_out(tensor *out__, tensor out, tensor self, tensor mat2) { PROTECT( auto outputs__ = torch::bmm_out(*out, *self, *mat2); @@ -4997,9 +5521,9 @@ void atg_bucketize_tensor_out(tensor *out__, tensor out, tensor self, tensor bou ) } -int atg_can_cast(int from, int to) { +int atg_can_cast(int from_, int to) { PROTECT( - return torch::can_cast(at::ScalarType(from), at::ScalarType(to)); + return torch::can_cast(at::ScalarType(from_), at::ScalarType(to)); ) return 0; } @@ -7131,7 +7655,21 @@ void atg_fbgemm_linear_fp16_weight(tensor *out__, tensor input, tensor packed_we void atg_fbgemm_linear_fp16_weight_fp32_activation(tensor *out__, tensor input, tensor packed_weight, tensor bias) { PROTECT( - auto outputs__ = torch::fbgemm_linear_fp16_weight_fp32_activation(*input, *packed_weight, *bias); + auto outputs__ = torch::fbgemm_linear_fp16_weight_fp32_activation(*input, *packed_weight, (bias ? *bias : torch::Tensor())); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_fbgemm_linear_fp16_weight_fp32_activation_out(tensor *out__, tensor input, tensor packed_weight, tensor bias, tensor output) { + PROTECT( + auto outputs__ = torch::fbgemm_linear_fp16_weight_fp32_activation(*input, *packed_weight, (bias ? *bias : torch::Tensor()), *output); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_fbgemm_linear_fp16_weight_out(tensor *out__, tensor input, tensor packed_weight, tensor bias, tensor output) { + PROTECT( + auto outputs__ = torch::fbgemm_linear_fp16_weight(*input, *packed_weight, *bias, *output); out__[0] = new torch::Tensor(outputs__); ) } @@ -7710,6 +8248,13 @@ void atg_floor_divide_scalar_(tensor *out__, tensor self, scalar other) { ) } +void atg_floor_divide_scalar_out(tensor *out__, tensor out, tensor self, scalar other) { + PROTECT( + auto outputs__ = torch::floor_divide_out(*out, *self, *other); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_floor_out(tensor *out__, tensor out, tensor self) { PROTECT( auto outputs__ = torch::floor_out(*out, *self); @@ -8588,6 +9133,20 @@ void atg_hardtanh_out(tensor *out__, tensor out, tensor self) { ) } +void atg_hash_tensor(tensor *out__, tensor self, int64_t *dim_data, int dim_len, int keepdim, int64_t mode) { + PROTECT( + auto outputs__ = torch::hash_tensor(*self, torch::IntArrayRef(dim_data, dim_len), (bool)keepdim, mode); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_hash_tensor_out(tensor *out__, tensor out, tensor self, int64_t *dim_data, int dim_len, int keepdim, int64_t mode) { + PROTECT( + auto outputs__ = torch::hash_tensor_out(*out, *self, torch::IntArrayRef(dim_data, dim_len), (bool)keepdim, mode); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_heaviside(tensor *out__, tensor self, tensor values) { PROTECT( auto outputs__ = torch::heaviside(*self, *values); @@ -8817,13 +9376,6 @@ void atg_imag(tensor *out__, tensor self) { ) } -void atg_index(tensor *out__, tensor self, tensor *indices_data, int indices_len) { - PROTECT( - auto outputs__ = torch::index(*self, of_carray_tensor_opt(indices_data, indices_len)); - out__[0] = new torch::Tensor(outputs__); - ) -} - void atg_index_add(tensor *out__, tensor self, int64_t dim, tensor index, tensor source) { PROTECT( auto outputs__ = torch::index_add(*self, dim, *index, *source); @@ -8908,27 +9460,6 @@ void atg_index_fill_int_tensor_out(tensor *out__, tensor out, tensor self, int64 ) } -void atg_index_put(tensor *out__, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate) { - PROTECT( - auto outputs__ = torch::index_put(*self, of_carray_tensor_opt(indices_data, indices_len), *values, (bool)accumulate); - out__[0] = new torch::Tensor(outputs__); - ) -} - -void atg_index_put_(tensor *out__, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate) { - PROTECT( - auto outputs__ = torch::index_put_(*self, of_carray_tensor_opt(indices_data, indices_len), *values, (bool)accumulate); - out__[0] = new torch::Tensor(outputs__); - ) -} - -void atg_index_put_out(tensor *out__, tensor out, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate) { - PROTECT( - auto outputs__ = torch::index_put_out(*out, *self, of_carray_tensor_opt(indices_data, indices_len), *values, (bool)accumulate); - out__[0] = new torch::Tensor(outputs__); - ) -} - void atg_index_reduce(tensor *out__, tensor self, int64_t dim, tensor index, tensor source, char* reduce_ptr, int reduce_len, int include_self) { PROTECT( auto outputs__ = torch::index_reduce(*self, dim, *index, *source, std::string(reduce_ptr, reduce_len), (bool)include_self); @@ -8971,13 +9502,6 @@ void atg_index_select_out(tensor *out__, tensor out, tensor self, int64_t dim, t ) } -void atg_index_tensor_out(tensor *out__, tensor out, tensor self, tensor *indices_data, int indices_len) { - PROTECT( - auto outputs__ = torch::index_out(*out, *self, of_carray_tensor_opt(indices_data, indices_len)); - out__[0] = new torch::Tensor(outputs__); - ) -} - void atg_indices(tensor *out__, tensor self) { PROTECT( auto outputs__ = self->indices(); @@ -10385,6 +10909,48 @@ void atg_linspace_out(tensor *out__, tensor out, scalar start, scalar end, int64 ) } +void atg_linspace_scalar_tensor(tensor *out__, scalar start, tensor end, int64_t steps, int options_kind, int options_device) { + PROTECT( + auto outputs__ = torch::linspace(*start, *end, steps, at::device(device_of_int(options_device)).dtype(at::ScalarType(options_kind))); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_linspace_scalar_tensor_out(tensor *out__, tensor out, scalar start, tensor end, int64_t steps) { + PROTECT( + auto outputs__ = torch::linspace_out(*out, *start, *end, steps); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_linspace_tensor_scalar(tensor *out__, tensor start, scalar end, int64_t steps, int options_kind, int options_device) { + PROTECT( + auto outputs__ = torch::linspace(*start, *end, steps, at::device(device_of_int(options_device)).dtype(at::ScalarType(options_kind))); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_linspace_tensor_scalar_out(tensor *out__, tensor out, tensor start, scalar end, int64_t steps) { + PROTECT( + auto outputs__ = torch::linspace_out(*out, *start, *end, steps); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_linspace_tensor_tensor(tensor *out__, tensor start, tensor end, int64_t steps, int options_kind, int options_device) { + PROTECT( + auto outputs__ = torch::linspace(*start, *end, steps, at::device(device_of_int(options_device)).dtype(at::ScalarType(options_kind))); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_linspace_tensor_tensor_out(tensor *out__, tensor out, tensor start, tensor end, int64_t steps) { + PROTECT( + auto outputs__ = torch::linspace_out(*out, *start, *end, steps); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_log(tensor *out__, tensor self) { PROTECT( auto outputs__ = torch::log(*self); @@ -10714,6 +11280,48 @@ void atg_logspace_out(tensor *out__, tensor out, scalar start, scalar end, int64 ) } +void atg_logspace_scalar_tensor(tensor *out__, scalar start, tensor end, int64_t steps, double base, int options_kind, int options_device) { + PROTECT( + auto outputs__ = torch::logspace(*start, *end, steps, base, at::device(device_of_int(options_device)).dtype(at::ScalarType(options_kind))); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_logspace_scalar_tensor_out(tensor *out__, tensor out, scalar start, tensor end, int64_t steps, double base) { + PROTECT( + auto outputs__ = torch::logspace_out(*out, *start, *end, steps, base); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_logspace_tensor_scalar(tensor *out__, tensor start, scalar end, int64_t steps, double base, int options_kind, int options_device) { + PROTECT( + auto outputs__ = torch::logspace(*start, *end, steps, base, at::device(device_of_int(options_device)).dtype(at::ScalarType(options_kind))); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_logspace_tensor_scalar_out(tensor *out__, tensor out, tensor start, scalar end, int64_t steps, double base) { + PROTECT( + auto outputs__ = torch::logspace_out(*out, *start, *end, steps, base); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_logspace_tensor_tensor(tensor *out__, tensor start, tensor end, int64_t steps, double base, int options_kind, int options_device) { + PROTECT( + auto outputs__ = torch::logspace(*start, *end, steps, base, at::device(device_of_int(options_device)).dtype(at::ScalarType(options_kind))); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_logspace_tensor_tensor_out(tensor *out__, tensor out, tensor start, tensor end, int64_t steps, double base) { + PROTECT( + auto outputs__ = torch::logspace_out(*out, *start, *end, steps, base); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_logsumexp(tensor *out__, tensor self, int64_t *dim_data, int dim_len, int keepdim) { PROTECT( auto outputs__ = torch::logsumexp(*self, torch::IntArrayRef(dim_data, dim_len), (bool)keepdim); @@ -10891,6 +11499,13 @@ void atg_masked_scatter_(tensor *out__, tensor self, tensor mask, tensor source) ) } +void atg_masked_scatter_backward(tensor *out__, tensor grad_output, tensor mask, int64_t *sizes_data, int sizes_len) { + PROTECT( + auto outputs__ = torch::masked_scatter_backward(*grad_output, *mask, torch::IntArrayRef(sizes_data, sizes_len)); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_masked_scatter_out(tensor *out__, tensor out, tensor self, tensor mask, tensor source) { PROTECT( auto outputs__ = torch::masked_scatter_out(*out, *self, *mask, *source); @@ -11171,6 +11786,13 @@ void atg_mean_dim(tensor *out__, tensor self, int64_t *dim_data, int dim_len, in ) } +void atg_mean_dtype_out(tensor *out__, tensor out, tensor self, int dtype) { + PROTECT( + auto outputs__ = torch::mean_out(*out, *self, dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(dtype))); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_mean_out(tensor *out__, tensor out, tensor self, int64_t *dim_data, int dim_len, int keepdim, int dtype) { PROTECT( auto outputs__ = torch::mean_out(*out, *self, dim_data == nullptr ? c10::nullopt : c10::optional(torch::IntArrayRef(dim_data, dim_len)), (bool)keepdim, dtype < 0 ? c10::nullopt : c10::optional(at::ScalarType(dtype))); @@ -11597,16 +12219,16 @@ void atg_mkldnn_reorder_conv2d_weight_out(tensor *out__, tensor out, tensor self ) } -void atg_mkldnn_reorder_conv3d_weight(tensor *out__, tensor self, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len, int64_t *dilation_data, int dilation_len, int64_t groups) { +void atg_mkldnn_reorder_conv3d_weight(tensor *out__, tensor self, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len, int64_t *dilation_data, int dilation_len, int64_t groups, int64_t *input_size_data, int input_size_len) { PROTECT( - auto outputs__ = torch::mkldnn_reorder_conv3d_weight(*self, torch::IntArrayRef(padding_data, padding_len), torch::IntArrayRef(stride_data, stride_len), torch::IntArrayRef(dilation_data, dilation_len), groups); + auto outputs__ = torch::mkldnn_reorder_conv3d_weight(*self, torch::IntArrayRef(padding_data, padding_len), torch::IntArrayRef(stride_data, stride_len), torch::IntArrayRef(dilation_data, dilation_len), groups, input_size_data == nullptr ? c10::nullopt : c10::optional(torch::IntArrayRef(input_size_data, input_size_len))); out__[0] = new torch::Tensor(outputs__); ) } -void atg_mkldnn_reorder_conv3d_weight_out(tensor *out__, tensor out, tensor self, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len, int64_t *dilation_data, int dilation_len, int64_t groups) { +void atg_mkldnn_reorder_conv3d_weight_out(tensor *out__, tensor out, tensor self, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len, int64_t *dilation_data, int dilation_len, int64_t groups, int64_t *input_size_data, int input_size_len) { PROTECT( - auto outputs__ = torch::mkldnn_reorder_conv3d_weight_out(*out, *self, torch::IntArrayRef(padding_data, padding_len), torch::IntArrayRef(stride_data, stride_len), torch::IntArrayRef(dilation_data, dilation_len), groups); + auto outputs__ = torch::mkldnn_reorder_conv3d_weight_out(*out, *self, torch::IntArrayRef(padding_data, padding_len), torch::IntArrayRef(stride_data, stride_len), torch::IntArrayRef(dilation_data, dilation_len), groups, input_size_data == nullptr ? c10::nullopt : c10::optional(torch::IntArrayRef(input_size_data, input_size_len))); out__[0] = new torch::Tensor(outputs__); ) } @@ -11664,6 +12286,20 @@ void atg_mm(tensor *out__, tensor self, tensor mat2) { ) } +void atg_mm_dtype(tensor *out__, tensor self, tensor mat2, int out_dtype) { + PROTECT( + auto outputs__ = torch::mm(*self, *mat2, at::ScalarType(out_dtype)); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_mm_dtype_out(tensor *out__, tensor out, tensor self, tensor mat2, int out_dtype) { + PROTECT( + auto outputs__ = torch::mm_out(*out, *self, *mat2, at::ScalarType(out_dtype)); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_mm_out(tensor *out__, tensor out, tensor self, tensor mat2) { PROTECT( auto outputs__ = torch::mm_out(*out, *self, *mat2); @@ -12717,9 +13353,9 @@ void atg_pad(tensor *out__, tensor self, int64_t *pad_data, int pad_len, char* m ) } -void atg_pad_sequence(tensor *out__, tensor *sequences_data, int sequences_len, int batch_first, double padding_value) { +void atg_pad_sequence(tensor *out__, tensor *sequences_data, int sequences_len, int batch_first, double padding_value, char* padding_side_ptr, int padding_side_len) { PROTECT( - auto outputs__ = torch::pad_sequence(of_carray_tensor(sequences_data, sequences_len), (bool)batch_first, padding_value); + auto outputs__ = torch::pad_sequence(of_carray_tensor(sequences_data, sequences_len), (bool)batch_first, padding_value, std::string(padding_side_ptr, padding_side_len)); out__[0] = new torch::Tensor(outputs__); ) } @@ -13293,6 +13929,20 @@ void atg_randint_like_out(tensor *out__, tensor out, tensor self, int64_t high) ) } +void atg_randint_like_tensor(tensor *out__, tensor self, tensor high) { + PROTECT( + auto outputs__ = torch::randint_like(*self, *high); + out__[0] = new torch::Tensor(outputs__); + ) +} + +void atg_randint_like_tensor_out(tensor *out__, tensor out, tensor self, tensor high) { + PROTECT( + auto outputs__ = torch::randint_like_out(*out, *self, *high); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_randint_low(tensor *out__, int64_t low, int64_t high, int64_t *size_data, int size_len, int options_kind, int options_device) { PROTECT( auto outputs__ = torch::randint(low, high, torch::IntArrayRef(size_data, size_len), at::device(device_of_int(options_device)).dtype(at::ScalarType(options_kind))); @@ -13909,6 +14559,13 @@ int atg_retains_grad(tensor self) { return 0; } +void atg_rms_norm(tensor *out__, tensor input, int64_t *normalized_shape_data, int normalized_shape_len, tensor weight, double eps_v, uint8_t eps_null) { + PROTECT( + auto outputs__ = torch::rms_norm(*input, torch::IntArrayRef(normalized_shape_data, normalized_shape_len), (weight ? *weight : torch::Tensor()), eps_null ? c10::nullopt : c10::optional(eps_v)); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_rnn_relu(tensor *out__, tensor input, tensor hx, tensor *params_data, int params_len, int has_biases, int64_t num_layers, double dropout, int train, int bidirectional, int batch_first) { PROTECT( auto outputs__ = torch::rnn_relu(*input, *hx, of_carray_tensor(params_data, params_len), (bool)has_biases, num_layers, dropout, (bool)train, (bool)bidirectional, (bool)batch_first); @@ -14102,6 +14759,14 @@ void atg_rrelu_with_noise_backward_out(tensor *out__, tensor out, tensor grad_ou ) } +void atg_rrelu_with_noise_functional(tensor *out__, tensor self, tensor noise, int training) { + PROTECT( + auto outputs__ = torch::rrelu_with_noise_functional(*self, *noise, (bool)training); + out__[0] = new torch::Tensor(std::get<0>(outputs__)); + out__[1] = new torch::Tensor(std::get<1>(outputs__)); + ) +} + void atg_rrelu_with_noise_out(tensor *out__, tensor out, tensor self, tensor noise, int training) { PROTECT( auto outputs__ = torch::rrelu_with_noise_out(*out, *self, *noise, (bool)training); @@ -14172,9 +14837,9 @@ void atg_scalar_tensor_out(tensor *out__, tensor out, scalar s) { ) } -void atg_scaled_dot_product_attention(tensor *out__, tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, double scale_v, uint8_t scale_null) { +void atg_scaled_dot_product_attention(tensor *out__, tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, double scale_v, uint8_t scale_null, int enable_gqa) { PROTECT( - auto outputs__ = torch::scaled_dot_product_attention(*query, *key, *value, (attn_mask ? *attn_mask : torch::Tensor()), dropout_p, (bool)is_causal, scale_null ? c10::nullopt : c10::optional(scale_v)); + auto outputs__ = torch::scaled_dot_product_attention(*query, *key, *value, (attn_mask ? *attn_mask : torch::Tensor()), dropout_p, (bool)is_causal, scale_null ? c10::nullopt : c10::optional(scale_v), (bool)enable_gqa); out__[0] = new torch::Tensor(outputs__); ) } @@ -14669,6 +15334,13 @@ void atg_slice_copy_tensor_out(tensor *out__, tensor out, tensor self, int64_t d ) } +void atg_slice_inverse(tensor *out__, tensor self, tensor src, int64_t dim, int64_t start_v, uint8_t start_null, int64_t end_v, uint8_t end_null, int64_t step) { + PROTECT( + auto outputs__ = torch::slice_inverse(*self, *src, dim, start_null ? c10::nullopt : c10::optional(start_v), end_null ? c10::nullopt : c10::optional(end_v), step); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_slice_scatter(tensor *out__, tensor self, tensor src, int64_t dim, int64_t start_v, uint8_t start_null, int64_t end_v, uint8_t end_null, int64_t step) { PROTECT( auto outputs__ = torch::slice_scatter(*self, *src, dim, start_null ? c10::nullopt : c10::optional(start_v), end_null ? c10::nullopt : c10::optional(end_v), step); @@ -16585,16 +17257,16 @@ void atg_std_out(tensor *out__, tensor out, tensor self, int64_t *dim_data, int ) } -void atg_stft(tensor *out__, tensor self, int64_t n_fft, int64_t hop_length_v, uint8_t hop_length_null, int64_t win_length_v, uint8_t win_length_null, tensor window, int normalized, int onesided, int return_complex) { +void atg_stft(tensor *out__, tensor self, int64_t n_fft, int64_t hop_length_v, uint8_t hop_length_null, int64_t win_length_v, uint8_t win_length_null, tensor window, int normalized, int onesided, int return_complex, int align_to_window) { PROTECT( - auto outputs__ = torch::stft(*self, n_fft, hop_length_null ? c10::nullopt : c10::optional(hop_length_v), win_length_null ? c10::nullopt : c10::optional(win_length_v), (window ? *window : torch::Tensor()), (bool)normalized, (bool)onesided, (bool)return_complex); + auto outputs__ = torch::stft(*self, n_fft, hop_length_null ? c10::nullopt : c10::optional(hop_length_v), win_length_null ? c10::nullopt : c10::optional(win_length_v), (window ? *window : torch::Tensor()), (bool)normalized, (bool)onesided, (bool)return_complex, (bool)align_to_window); out__[0] = new torch::Tensor(outputs__); ) } -void atg_stft_center(tensor *out__, tensor self, int64_t n_fft, int64_t hop_length_v, uint8_t hop_length_null, int64_t win_length_v, uint8_t win_length_null, tensor window, int center, char* pad_mode_ptr, int pad_mode_len, int normalized, int onesided, int return_complex) { +void atg_stft_center(tensor *out__, tensor self, int64_t n_fft, int64_t hop_length_v, uint8_t hop_length_null, int64_t win_length_v, uint8_t win_length_null, tensor window, int center, char* pad_mode_ptr, int pad_mode_len, int normalized, int onesided, int return_complex, int align_to_window) { PROTECT( - auto outputs__ = torch::stft(*self, n_fft, hop_length_null ? c10::nullopt : c10::optional(hop_length_v), win_length_null ? c10::nullopt : c10::optional(win_length_v), (window ? *window : torch::Tensor()), (bool)center, std::string(pad_mode_ptr, pad_mode_len), (bool)normalized, (bool)onesided, (bool)return_complex); + auto outputs__ = torch::stft(*self, n_fft, hop_length_null ? c10::nullopt : c10::optional(hop_length_v), win_length_null ? c10::nullopt : c10::optional(win_length_v), (window ? *window : torch::Tensor()), (bool)center, std::string(pad_mode_ptr, pad_mode_len), (bool)normalized, (bool)onesided, (bool)return_complex, (bool)align_to_window); out__[0] = new torch::Tensor(outputs__); ) } @@ -17651,6 +18323,13 @@ void atg_upsample_bilinear2d_vec(tensor *out__, tensor input, int64_t *output_si ) } +void atg_upsample_bilinear2d_vec_out(tensor *out__, tensor out, tensor input, int64_t *output_size_data, int output_size_len, int align_corners, double *scale_factors_data, int scale_factors_len) { + PROTECT( + auto outputs__ = torch::upsample_bilinear2d_out(*out, *input, output_size_data == nullptr ? c10::nullopt : c10::optional(torch::IntArrayRef(output_size_data, output_size_len)), (bool)align_corners, at::ArrayRef(scale_factors_data, scale_factors_len)); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_upsample_linear1d(tensor *out__, tensor self, int64_t *output_size_data, int output_size_len, int align_corners, double scales_v, uint8_t scales_null) { PROTECT( auto outputs__ = torch::upsample_linear1d(*self, torch::IntArrayRef(output_size_data, output_size_len), (bool)align_corners, scales_null ? c10::nullopt : c10::optional(scales_v)); @@ -17756,6 +18435,13 @@ void atg_upsample_nearest2d_vec(tensor *out__, tensor input, int64_t *output_siz ) } +void atg_upsample_nearest2d_vec_out(tensor *out__, tensor out, tensor input, int64_t *output_size_data, int output_size_len, double *scale_factors_data, int scale_factors_len) { + PROTECT( + auto outputs__ = torch::upsample_nearest2d_out(*out, *input, output_size_data == nullptr ? c10::nullopt : c10::optional(torch::IntArrayRef(output_size_data, output_size_len)), at::ArrayRef(scale_factors_data, scale_factors_len)); + out__[0] = new torch::Tensor(outputs__); + ) +} + void atg_upsample_nearest3d(tensor *out__, tensor self, int64_t *output_size_data, int output_size_len, double scales_d_v, uint8_t scales_d_null, double scales_h_v, uint8_t scales_h_null, double scales_w_v, uint8_t scales_w_null) { PROTECT( auto outputs__ = torch::upsample_nearest3d(*self, torch::IntArrayRef(output_size_data, output_size_len), scales_d_null ? c10::nullopt : c10::optional(scales_d_v), scales_h_null ? c10::nullopt : c10::optional(scales_h_v), scales_w_null ? c10::nullopt : c10::optional(scales_w_v)); diff --git a/libtch/torch_api_generated.h b/libtch/torch_api_generated.h index d4d0615..f985b45 100644 --- a/libtch/torch_api_generated.h +++ b/libtch/torch_api_generated.h @@ -48,8 +48,14 @@ void atg__aminmax_out(tensor *, tensor out0, tensor out1, tensor self); void atg__amp_update_scale(tensor *, tensor self, tensor growth_tracker, tensor found_inf, double scale_growth_factor, double scale_backoff_factor, int64_t growth_interval); void atg__amp_update_scale_(tensor *, tensor self, tensor growth_tracker, tensor found_inf, double scale_growth_factor, double scale_backoff_factor, int64_t growth_interval); void atg__amp_update_scale_out(tensor *, tensor out, tensor self, tensor growth_tracker, tensor found_inf, double scale_growth_factor, double scale_backoff_factor, int64_t growth_interval); +void atg__assert_scalar(scalar self_scalar, char* assert_msg_ptr, int assert_msg_len); void atg__autocast_to_full_precision(tensor *, tensor self, int cuda_enabled, int cpu_enabled); void atg__autocast_to_reduced_precision(tensor *, tensor self, int cuda_enabled, int cpu_enabled, int cuda_dtype, int cpu_dtype); +void atg__batch_norm_no_update(tensor *, tensor input, tensor weight, tensor bias, tensor running_mean, tensor running_var, double momentum, double eps); +void atg__batch_norm_no_update_out(tensor *, tensor out0, tensor out1, tensor out2, tensor out3, tensor input, tensor weight, tensor bias, tensor running_mean, tensor running_var, double momentum, double eps); +void atg__batch_norm_with_update(tensor *, tensor input, tensor weight, tensor bias, tensor running_mean, tensor running_var, double momentum, double eps); +void atg__batch_norm_with_update_functional(tensor *, tensor input, tensor weight, tensor bias, tensor running_mean, tensor running_var, double momentum, double eps); +void atg__batch_norm_with_update_out(tensor *, tensor out, tensor save_mean, tensor save_invstd, tensor reserve, tensor input, tensor weight, tensor bias, tensor running_mean, tensor running_var, double momentum, double eps); void atg__cast_byte(tensor *, tensor self, int non_blocking); void atg__cast_char(tensor *, tensor self, int non_blocking); void atg__cast_double(tensor *, tensor self, int non_blocking); @@ -62,6 +68,8 @@ void atg__cdist_backward(tensor *, tensor grad, tensor x1, tensor x2, double p, void atg__cdist_backward_out(tensor *, tensor out, tensor grad, tensor x1, tensor x2, double p, tensor cdist); void atg__cholesky_solve_helper(tensor *, tensor self, tensor A, int upper); void atg__cholesky_solve_helper_out(tensor *, tensor out, tensor self, tensor A, int upper); +void atg__chunk_cat(tensor *, tensor *tensors_data, int tensors_len, int64_t dim, int64_t num_chunks); +void atg__chunk_cat_out(tensor *, tensor out, tensor *tensors_data, int tensors_len, int64_t dim, int64_t num_chunks); void atg__coalesce(tensor *, tensor self); void atg__coalesce_out(tensor *, tensor out, tensor self); void atg__coalesced(tensor *, tensor self, int coalesced); @@ -80,6 +88,8 @@ void atg__convert_indices_from_coo_to_csr(tensor *, tensor self, int64_t size, i void atg__convert_indices_from_coo_to_csr_out(tensor *, tensor out, tensor self, int64_t size, int out_int32); void atg__convert_indices_from_csr_to_coo(tensor *, tensor crow_indices, tensor col_indices, int out_int32, int transpose); void atg__convert_indices_from_csr_to_coo_out(tensor *, tensor out, tensor crow_indices, tensor col_indices, int out_int32, int transpose); +void atg__convert_weight_to_int4pack(tensor *, tensor self, int64_t innerKTiles); +void atg__convert_weight_to_int4pack_for_cpu(tensor *, tensor self, int64_t innerKTiles); void atg__convolution(tensor *, tensor input, tensor weight, tensor bias, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int64_t *dilation_data, int dilation_len, int transposed, int64_t *output_padding_data, int output_padding_len, int64_t groups, int benchmark, int deterministic, int cudnn_enabled, int allow_tf32); void atg__convolution_deprecated(tensor *, tensor input, tensor weight, tensor bias, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int64_t *dilation_data, int dilation_len, int transposed, int64_t *output_padding_data, int output_padding_len, int64_t groups, int benchmark, int deterministic, int cudnn_enabled); void atg__convolution_mode(tensor *, tensor input, tensor weight, tensor bias, int64_t *stride_data, int stride_len, char* padding_ptr, int padding_len, int64_t *dilation_data, int dilation_len, int64_t groups); @@ -89,7 +99,8 @@ void atg__copy_from_and_resize(tensor *, tensor self, tensor dst); void atg__copy_from_and_resize_out(tensor *, tensor out, tensor self, tensor dst); void atg__copy_from_out(tensor *, tensor out, tensor self, tensor dst, int non_blocking); void atg__cslt_compress(tensor *, tensor input); -void atg__cslt_sparse_mm(tensor *, tensor compressed_A, tensor dense_B, tensor bias, int transpose_result); +void atg__cslt_sparse_mm(tensor *, tensor compressed_A, tensor dense_B, tensor bias, tensor alpha, int out_dtype, int transpose_result, int64_t alg_id, int64_t split_k, int64_t split_k_mode); +int64_t atg__cslt_sparse_mm_search(tensor compressed_A, tensor dense_B, tensor bias, tensor alpha, int out_dtype, int transpose_result); void atg__ctc_loss(tensor *, tensor log_probs, tensor targets, int64_t *input_lengths_data, int input_lengths_len, int64_t *target_lengths_data, int target_lengths_len, int64_t blank, int zero_infinity); void atg__ctc_loss_backward(tensor *, tensor grad, tensor log_probs, tensor targets, int64_t *input_lengths_data, int input_lengths_len, int64_t *target_lengths_data, int target_lengths_len, tensor neg_log_likelihood, tensor log_alpha, int64_t blank, int zero_infinity); void atg__ctc_loss_backward_out(tensor *, tensor out, tensor grad, tensor log_probs, tensor targets, int64_t *input_lengths_data, int input_lengths_len, int64_t *target_lengths_data, int target_lengths_len, tensor neg_log_likelihood, tensor log_alpha, int64_t blank, int zero_infinity); @@ -97,6 +108,7 @@ void atg__ctc_loss_backward_tensor(tensor *, tensor grad, tensor log_probs, tens void atg__ctc_loss_out(tensor *, tensor out0, tensor out1, tensor log_probs, tensor targets, int64_t *input_lengths_data, int input_lengths_len, int64_t *target_lengths_data, int target_lengths_len, int64_t blank, int zero_infinity); void atg__ctc_loss_tensor(tensor *, tensor log_probs, tensor targets, tensor input_lengths, tensor target_lengths, int64_t blank, int zero_infinity); void atg__ctc_loss_tensor_out(tensor *, tensor out0, tensor out1, tensor log_probs, tensor targets, tensor input_lengths, tensor target_lengths, int64_t blank, int zero_infinity); +void atg__cudnn_attention_backward(tensor *, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, tensor philox_seed, tensor philox_offset, tensor attn_bias, tensor cum_seq_q, tensor cum_seq_k, int64_t max_q, int64_t max_k, double dropout_p, int is_causal, double scale_v, uint8_t scale_null); void atg__cudnn_ctc_loss(tensor *, tensor log_probs, tensor targets, int64_t *input_lengths_data, int input_lengths_len, int64_t *target_lengths_data, int target_lengths_len, int64_t blank, int deterministic, int zero_infinity); void atg__cudnn_ctc_loss_out(tensor *, tensor out0, tensor out1, tensor log_probs, tensor targets, int64_t *input_lengths_data, int input_lengths_len, int64_t *target_lengths_data, int target_lengths_len, int64_t blank, int deterministic, int zero_infinity); void atg__cudnn_ctc_loss_tensor(tensor *, tensor log_probs, tensor targets, tensor input_lengths, tensor target_lengths, int64_t blank, int deterministic, int zero_infinity); @@ -112,7 +124,9 @@ int64_t atg__dimi(tensor self); int64_t atg__dimv(tensor self); void atg__dirichlet_grad(tensor *, tensor x, tensor alpha, tensor total); void atg__dirichlet_grad_out(tensor *, tensor out, tensor x, tensor alpha, tensor total); -void atg__efficient_attention_backward(tensor *, tensor grad_out_, tensor query, tensor key, tensor value, tensor bias, tensor out, tensor cu_seqlens_q, tensor cu_seqlens_k, int64_t max_seqlen_k, int64_t max_seqlen_q, tensor logsumexp, double dropout_p, tensor philox_seed, tensor philox_offset, int64_t custom_mask_type, int bias_requires_grad, double scale_v, uint8_t scale_null, int64_t num_splits_key_v, uint8_t num_splits_key_null); +void atg__dyn_quant_matmul_4bit(tensor *, tensor inp, tensor packed_weights, int64_t block_size, int64_t in_features, int64_t out_features); +void atg__dyn_quant_pack_4bit_weight(tensor *, tensor weights, tensor scales_zeros, tensor bias, int64_t block_size, int64_t in_features, int64_t out_features); +void atg__efficient_attention_backward(tensor *, tensor grad_out_, tensor query, tensor key, tensor value, tensor bias, tensor out, tensor cu_seqlens_q, tensor cu_seqlens_k, int64_t max_seqlen_q, int64_t max_seqlen_k, tensor logsumexp, double dropout_p, tensor philox_seed, tensor philox_offset, int64_t custom_mask_type, int bias_requires_grad, double scale_v, uint8_t scale_null, int64_t num_splits_key_v, uint8_t num_splits_key_null, int64_t window_size_v, uint8_t window_size_null, int shared_storage_dqdkdv); void atg__efficientzerotensor(tensor *, int64_t *size_data, int size_len, int options_kind, int options_device); void atg__efficientzerotensor_out(tensor *, tensor out, int64_t *size_data, int size_len); void atg__embedding_bag(tensor *, tensor weight, tensor indices, tensor offsets, int scale_grad_by_freq, int64_t mode, int sparse, tensor per_sample_weights, int include_last_offset, int64_t padding_idx); @@ -146,18 +160,28 @@ void atg__fft_c2r_out(tensor *, tensor out, tensor self, int64_t *dim_data, int void atg__fft_r2c(tensor *, tensor self, int64_t *dim_data, int dim_len, int64_t normalization, int onesided); void atg__fft_r2c_out(tensor *, tensor out, tensor self, int64_t *dim_data, int dim_len, int64_t normalization, int onesided); void atg__fill_mem_eff_dropout_mask_(tensor *, tensor self, double dropout_p, int64_t seed, int64_t offset); -void atg__flash_attention_backward(tensor *, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, tensor cum_seq_q, tensor cum_seq_k, int64_t max_q, int64_t max_k, double dropout_p, int is_causal, tensor philox_seed, tensor philox_offset, double scale_v, uint8_t scale_null); +void atg__flash_attention_backward(tensor *, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, tensor cum_seq_q, tensor cum_seq_k, int64_t max_q, int64_t max_k, double dropout_p, int is_causal, tensor rng_state, tensor unused, double scale_v, uint8_t scale_null, int64_t window_size_left_v, uint8_t window_size_left_null, int64_t window_size_right_v, uint8_t window_size_right_null); void atg__foobar(tensor *, tensor self, int arg1, int arg2, int arg3); void atg__foobar_out(tensor *, tensor out, tensor self, int arg1, int arg2, int arg3); void atg__functional_assert_async(tensor *, tensor self, char* assert_msg_ptr, int assert_msg_len, tensor dep_token); +void atg__functional_assert_scalar(tensor *, scalar self_scalar, char* assert_msg_ptr, int assert_msg_len, tensor dep_token); void atg__functional_sym_constrain_range(tensor *, scalar size, int64_t min_v, uint8_t min_null, int64_t max_v, uint8_t max_null, tensor dep_token); void atg__functional_sym_constrain_range_for_size(tensor *, scalar size, int64_t min_v, uint8_t min_null, int64_t max_v, uint8_t max_null, tensor dep_token); +void atg__fused_adagrad(tensor *out_data, int out_len, tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *state_sums_data, int state_sums_len, tensor *state_steps_data, int state_steps_len, double lr, double lr_decay, double weight_decay, double eps, int maximize, tensor grad_scale, tensor found_inf); +void atg__fused_adagrad_(tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *state_sums_data, int state_sums_len, tensor *state_steps_data, int state_steps_len, double lr, double lr_decay, double weight_decay, double eps, int maximize, tensor grad_scale, tensor found_inf); +void atg__fused_adagrad_tensor_lr_(tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *state_sums_data, int state_sums_len, tensor *state_steps_data, int state_steps_len, tensor lr, double lr_decay, double weight_decay, double eps, int maximize, tensor grad_scale, tensor found_inf); +void atg__fused_adagrad_tensor_lr_out(tensor *out_data, int out_len, tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *state_sums_data, int state_sums_len, tensor *state_steps_data, int state_steps_len, tensor lr, double lr_decay, double weight_decay, double eps, int maximize, tensor grad_scale, tensor found_inf); void atg__fused_dropout(tensor *, tensor self, double p); void atg__fused_dropout_out(tensor *, tensor out0, tensor out1, tensor self, double p); void atg__fused_moving_avg_obs_fq_helper(tensor *, tensor self, tensor observer_on, tensor fake_quant_on, tensor running_min, tensor running_max, tensor scale, tensor zero_point, double averaging_const, int64_t quant_min, int64_t quant_max, int64_t ch_axis, int per_row_fake_quant, int symmetric_quant); void atg__fused_moving_avg_obs_fq_helper_functional(tensor *, tensor self, tensor observer_on, tensor fake_quant_on, tensor running_min, tensor running_max, tensor scale, tensor zero_point, double averaging_const, int64_t quant_min, int64_t quant_max, int64_t ch_axis, int per_row_fake_quant, int symmetric_quant); void atg__fused_moving_avg_obs_fq_helper_out(tensor *, tensor out0, tensor out1, tensor self, tensor observer_on, tensor fake_quant_on, tensor running_min, tensor running_max, tensor scale, tensor zero_point, double averaging_const, int64_t quant_min, int64_t quant_max, int64_t ch_axis, int per_row_fake_quant, int symmetric_quant); -int64_t atg__fused_sdp_choice(tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, double scale_v, uint8_t scale_null); +void atg__fused_rms_norm(tensor *, tensor input, int64_t *normalized_shape_data, int normalized_shape_len, tensor weight, double eps_v, uint8_t eps_null); +int64_t atg__fused_sdp_choice(tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, double scale_v, uint8_t scale_null, int enable_gqa); +void atg__fused_sgd(tensor *out_data, int out_len, tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *momentum_buffer_list_data, int momentum_buffer_list_len, double weight_decay, double momentum, double lr, double dampening, int nesterov, int maximize, int is_first_step, tensor grad_scale, tensor found_inf); +void atg__fused_sgd_(tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *momentum_buffer_list_data, int momentum_buffer_list_len, double weight_decay, double momentum, double lr, double dampening, int nesterov, int maximize, int is_first_step, tensor grad_scale, tensor found_inf); +void atg__fused_sgd_tensor_lr_(tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *momentum_buffer_list_data, int momentum_buffer_list_len, double weight_decay, double momentum, tensor lr, double dampening, int nesterov, int maximize, int is_first_step, tensor grad_scale, tensor found_inf); +void atg__fused_sgd_tensor_lr_out(tensor *out_data, int out_len, tensor *self_data, int self_len, tensor *grads_data, int grads_len, tensor *momentum_buffer_list_data, int momentum_buffer_list_len, double weight_decay, double momentum, tensor lr, double dampening, int nesterov, int maximize, int is_first_step, tensor grad_scale, tensor found_inf); void atg__fw_primal(tensor *, tensor self, int64_t level); void atg__fw_primal_copy(tensor *, tensor self, int64_t level); void atg__fw_primal_copy_out(tensor *, tensor out, tensor self, int64_t level); @@ -165,14 +189,12 @@ void atg__gather_sparse_backward(tensor *, tensor self, int64_t dim, tensor inde void atg__grid_sampler_2d_cpu_fallback(tensor *, tensor input, tensor grid, int64_t interpolation_mode, int64_t padding_mode, int align_corners); void atg__grid_sampler_2d_cpu_fallback_backward(tensor *, tensor grad_output, tensor input, tensor grid, int64_t interpolation_mode, int64_t padding_mode, int align_corners); void atg__grid_sampler_2d_cpu_fallback_out(tensor *, tensor out, tensor input, tensor grid, int64_t interpolation_mode, int64_t padding_mode, int align_corners); +void atg__grouped_mm(tensor *, tensor self, tensor mat2, tensor offs, tensor bias, int out_dtype); int atg__has_compatible_shallow_copy_type(tensor self, tensor from); int atg__has_same_storage_numel(tensor self, tensor other); void atg__histogramdd_from_bin_cts(tensor *, tensor out, tensor self, int64_t *bins_data, int bins_len, double *range_data, int range_len, tensor weight, int density); void atg__histogramdd_from_bin_tensors(tensor *, tensor self, tensor *bins_data, int bins_len, tensor weight, int density); void atg__histogramdd_from_bin_tensors_out(tensor *, tensor out, tensor self, tensor *bins_data, int bins_len, tensor weight, int density); -void atg__index_put_impl(tensor *, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate, int unsafe); -void atg__index_put_impl_(tensor *, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate, int unsafe); -void atg__index_put_impl_out(tensor *, tensor out, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate, int unsafe); void atg__indices(tensor *, tensor self); void atg__indices_copy(tensor *, tensor self); void atg__indices_copy_out(tensor *, tensor out, tensor self); @@ -181,10 +203,12 @@ void atg__int_mm_out(tensor *, tensor out, tensor self, tensor mat2); void atg__is_all_true(tensor *, tensor self); void atg__is_any_true(tensor *, tensor self); int atg__is_zerotensor(tensor self); +void atg__lazy_clone(tensor *, tensor self); void atg__linalg_det(tensor *, tensor A); void atg__linalg_det_result(tensor *, tensor result, tensor LU, tensor pivots, tensor A); void atg__linalg_eigh(tensor *, tensor A, char* UPLO_ptr, int UPLO_len, int compute_v); void atg__linalg_eigh_eigenvalues(tensor *, tensor eigenvalues, tensor eigenvectors, tensor A, char* UPLO_ptr, int UPLO_len, int compute_v); +void atg__linalg_eigvals(tensor *, tensor self); void atg__linalg_slogdet(tensor *, tensor A); void atg__linalg_slogdet_sign(tensor *, tensor sign, tensor logabsdet, tensor LU, tensor pivots, tensor A); void atg__linalg_solve_ex(tensor *, tensor A, tensor B, int left, int check_errors); @@ -214,6 +238,7 @@ void atg__masked_softmax(tensor *, tensor self, tensor mask, int64_t dim_v, uint void atg__masked_softmax_backward(tensor *, tensor grad_output, tensor output, tensor mask, int64_t dim_v, uint8_t dim_null); void atg__masked_softmax_backward_out(tensor *, tensor out, tensor grad_output, tensor output, tensor mask, int64_t dim_v, uint8_t dim_null); void atg__masked_softmax_out(tensor *, tensor out, tensor self, tensor mask, int64_t dim_v, uint8_t dim_null, int64_t mask_type_v, uint8_t mask_type_null); +void atg__mixed_dtypes_linear(tensor *, tensor input, tensor weight, tensor scale, tensor bias, char* activation_ptr, int activation_len); void atg__mkldnn_reshape(tensor *, tensor self, int64_t *shape_data, int shape_len); void atg__mkldnn_reshape_out(tensor *, tensor out, tensor self, int64_t *shape_data, int shape_len); void atg__mkldnn_transpose(tensor *, tensor self, int64_t dim0, int64_t dim1); @@ -235,15 +260,29 @@ void atg__native_multi_head_attention_out(tensor *, tensor out0, tensor out1, te void atg__neg_view(tensor *, tensor self); void atg__neg_view_copy(tensor *, tensor self); void atg__neg_view_copy_out(tensor *, tensor out, tensor self); +void atg__nested_compute_contiguous_strides_offsets(tensor *, tensor nested_size); void atg__nested_from_padded(tensor *, tensor padded, tensor cpu_nested_shape_example, int fuse_transform_0213); void atg__nested_from_padded_and_nested_example(tensor *, tensor padded, tensor nt_example); void atg__nested_from_padded_and_nested_example_out(tensor *, tensor out, tensor padded, tensor nt_example); void atg__nested_from_padded_out(tensor *, tensor out, tensor padded, tensor cpu_nested_shape_example, int fuse_transform_0213); +void atg__nested_from_padded_tensor(tensor *, tensor padded, tensor offsets, tensor dummy, int64_t ragged_idx, tensor min_seqlen, tensor max_seqlen, int64_t sum_S_v, uint8_t sum_S_null); +void atg__nested_get_jagged_dummy(tensor *, tensor any); +void atg__nested_get_lengths(tensor *, tensor self); +void atg__nested_get_max_seqlen(tensor *, tensor self); +void atg__nested_get_min_seqlen(tensor *, tensor self); +void atg__nested_get_offsets(tensor *, tensor self); +int64_t atg__nested_get_ragged_idx(tensor self); +void atg__nested_get_values(tensor *, tensor self); +void atg__nested_get_values_copy(tensor *, tensor self); +void atg__nested_get_values_copy_out(tensor *, tensor out, tensor self); void atg__nested_select_backward(tensor *, tensor grad_output, tensor self, int64_t dim, int64_t index); void atg__nested_sum_backward(tensor *, tensor grad, tensor self, int64_t *dim_data, int dim_len, int keepdim); void atg__nested_view_from_buffer(tensor *, tensor self, tensor nested_size, tensor nested_strides, tensor offsets); void atg__nested_view_from_buffer_copy(tensor *, tensor self, tensor nested_size, tensor nested_strides, tensor offsets); void atg__nested_view_from_buffer_copy_out(tensor *, tensor out, tensor self, tensor nested_size, tensor nested_strides, tensor offsets); +void atg__nested_view_from_jagged(tensor *, tensor self, tensor offsets, tensor dummy, tensor lengths, int64_t ragged_idx, tensor min_seqlen, tensor max_seqlen); +void atg__nested_view_from_jagged_copy(tensor *, tensor self, tensor offsets, tensor dummy, tensor lengths, int64_t ragged_idx, tensor min_seqlen, tensor max_seqlen); +void atg__nested_view_from_jagged_copy_out(tensor *, tensor out, tensor self, tensor offsets, tensor dummy, tensor lengths, int64_t ragged_idx, tensor min_seqlen, tensor max_seqlen); void atg__new_zeros_with_same_feature_meta(tensor *, tensor self, tensor other, int64_t self_num_batch_dims); void atg__new_zeros_with_same_feature_meta_out(tensor *, tensor out, tensor self, tensor other, int64_t self_num_batch_dims); int atg__nnpack_available(); @@ -262,6 +301,7 @@ void atg__pin_memory(tensor *, tensor self, int device); void atg__pin_memory_out(tensor *, tensor out, tensor self, int device); void atg__prelu_kernel(tensor *, tensor self, tensor weight); void atg__prelu_kernel_backward(tensor *, tensor grad_output, tensor self, tensor weight); +void atg__print(char* s_ptr, int s_len); void atg__propagate_xla_data(tensor input, tensor output); void atg__remove_batch_dim(tensor *, tensor self, int64_t level, int64_t batch_size, int64_t out_dim); void atg__reshape_alias(tensor *, tensor self, int64_t *size_data, int size_len, int64_t *stride_data, int stride_len); @@ -273,14 +313,23 @@ void atg__resize_output(tensor *, tensor self, int64_t *size_data, int size_len, void atg__resize_output_(tensor *, tensor self, int64_t *size_data, int size_len, int device); void atg__resize_output_out(tensor *, tensor out, tensor self, int64_t *size_data, int size_len, int device); void atg__rowwise_prune(tensor *, tensor weight, tensor mask, int compressed_indices_dtype); +void atg__safe_softmax(tensor *, tensor self, int64_t dim, int dtype); void atg__sample_dirichlet(tensor *, tensor self); void atg__sample_dirichlet_out(tensor *, tensor out, tensor self); void atg__saturate_weight_to_fp16(tensor *, tensor weight); -void atg__scaled_dot_product_attention_math(tensor *, tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, tensor dropout_mask, double scale_v, uint8_t scale_null); +void atg__scaled_dot_product_attention_math(tensor *, tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, tensor dropout_mask, double scale_v, uint8_t scale_null, int enable_gqa); +void atg__scaled_dot_product_attention_math_for_mps(tensor *, tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, tensor dropout_mask, double scale_v, uint8_t scale_null); +void atg__scaled_dot_product_cudnn_attention_backward(tensor *, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, tensor philox_seed, tensor philox_offset, tensor attn_bias, tensor cum_seq_q, tensor cum_seq_k, int64_t max_q, int64_t max_k, double dropout_p, int is_causal, double scale_v, uint8_t scale_null); void atg__scaled_dot_product_efficient_attention(tensor *, tensor query, tensor key, tensor value, tensor attn_bias, int compute_log_sumexp, double dropout_p, int is_causal, double scale_v, uint8_t scale_null); void atg__scaled_dot_product_flash_attention_backward(tensor *, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, tensor cum_seq_q, tensor cum_seq_k, int64_t max_q, int64_t max_k, double dropout_p, int is_causal, tensor philox_seed, tensor philox_offset, double scale_v, uint8_t scale_null); -void atg__scaled_mm(tensor *, tensor self, tensor mat2, tensor bias, int out_dtype, tensor scale_a, tensor scale_b, tensor scale_result); -void atg__scaled_mm_out(tensor *, tensor out, tensor out_amax, tensor self, tensor mat2, tensor bias, int out_dtype, tensor scale_a, tensor scale_b, tensor scale_result); +void atg__scaled_dot_product_flash_attention_for_cpu(tensor *, tensor query, tensor key, tensor value, double dropout_p, int is_causal, tensor attn_mask, double scale_v, uint8_t scale_null); +void atg__scaled_dot_product_flash_attention_for_cpu_backward(tensor *, tensor grad_out, tensor query, tensor key, tensor value, tensor out, tensor logsumexp, double dropout_p, int is_causal, tensor attn_mask, double scale_v, uint8_t scale_null); +void atg__scaled_grouped_mm(tensor *, tensor self, tensor mat2, tensor scale_a, tensor scale_b, tensor offs, tensor bias, tensor scale_result, int out_dtype, int use_fast_accum); +void atg__scaled_grouped_mm_v2(tensor *, tensor self, tensor mat2, tensor *scale_a_data, int scale_a_len, int64_t *recipe_a_data, int recipe_a_len, int64_t *swizzle_a_data, int swizzle_a_len, tensor *scale_b_data, int scale_b_len, int64_t *recipe_b_data, int recipe_b_len, int64_t *swizzle_b_data, int swizzle_b_len, tensor offs, tensor bias, int out_dtype, int64_t *contraction_dim_data, int contraction_dim_len, int use_fast_accum); +void atg__scaled_mm(tensor *, tensor self, tensor mat2, tensor scale_a, tensor scale_b, tensor bias, tensor scale_result, int out_dtype, int use_fast_accum); +void atg__scaled_mm_out(tensor *, tensor out, tensor self, tensor mat2, tensor scale_a, tensor scale_b, tensor bias, tensor scale_result, int out_dtype, int use_fast_accum); +void atg__scaled_mm_v2(tensor *, tensor self, tensor mat2, tensor *scale_a_data, int scale_a_len, int64_t *recipe_a_data, int recipe_a_len, int64_t *swizzle_a_data, int swizzle_a_len, tensor *scale_b_data, int scale_b_len, int64_t *recipe_b_data, int recipe_b_len, int64_t *swizzle_b_data, int swizzle_b_len, tensor bias, int out_dtype, int64_t *contraction_dim_data, int contraction_dim_len, int use_fast_accum); +void atg__scaled_mm_v2_out(tensor *, tensor out, tensor self, tensor mat2, tensor *scale_a_data, int scale_a_len, int64_t *recipe_a_data, int recipe_a_len, int64_t *swizzle_a_data, int swizzle_a_len, tensor *scale_b_data, int scale_b_len, int64_t *recipe_b_data, int recipe_b_len, int64_t *swizzle_b_data, int swizzle_b_len, tensor bias, int out_dtype, int64_t *contraction_dim_data, int contraction_dim_len, int use_fast_accum); void atg__scatter_reduce(tensor *, tensor self, int64_t dim, tensor index, tensor src, char* reduce_ptr, int reduce_len, int include_self); void atg__scatter_reduce_(tensor *, tensor self, int64_t dim, tensor index, tensor src, char* reduce_ptr, int reduce_len, int include_self); void atg__scatter_reduce_two_out(tensor *, tensor out, tensor self, int64_t dim, tensor index, tensor src, char* reduce_ptr, int reduce_len, int include_self); @@ -304,6 +353,7 @@ void atg__sparse_broadcast_to_copy_out(tensor *, tensor out, tensor self, int64_ void atg__sparse_bsc_tensor_unsafe(tensor *, tensor ccol_indices, tensor row_indices, tensor values, int64_t *size_data, int size_len, int options_kind, int options_device); void atg__sparse_bsr_tensor_unsafe(tensor *, tensor crow_indices, tensor col_indices, tensor values, int64_t *size_data, int size_len, int options_kind, int options_device); void atg__sparse_compressed_tensor_unsafe(tensor *, tensor compressed_indices, tensor plain_indices, tensor values, int64_t *size_data, int size_len, int options_kind, int options_device); +void atg__sparse_compressed_tensor_with_dims(tensor *, int64_t nnz, int64_t dense_dim, int64_t *size_data, int size_len, int64_t *blocksize_data, int blocksize_len, int index_dtype, int options_kind, int options_device); void atg__sparse_coo_tensor_unsafe(tensor *, tensor indices, tensor values, int64_t *size_data, int size_len, int options_kind, int options_device, int is_coalesced); void atg__sparse_coo_tensor_with_dims(tensor *, int64_t sparse_dim, int64_t dense_dim, int64_t *size_data, int size_len, int options_kind, int options_device); void atg__sparse_coo_tensor_with_dims_and_tensors(tensor *, int64_t sparse_dim, int64_t dense_dim, int64_t *size_data, int size_len, tensor indices, tensor values, int options_kind, int options_device, int is_coalesced); @@ -325,7 +375,11 @@ void atg__sparse_mask_projection_out(tensor *, tensor out, tensor self, tensor m void atg__sparse_mm(tensor *, tensor sparse, tensor dense); void atg__sparse_mm_reduce(tensor *, tensor sparse, tensor dense, char* reduce_ptr, int reduce_len); void atg__sparse_mm_reduce_impl(tensor *, tensor self, tensor other, char* reduce_ptr, int reduce_len); -void atg__sparse_semi_structured_linear(tensor *, tensor input, tensor weight, tensor meta, tensor bias, char* activation_ptr, int activation_len); +void atg__sparse_semi_structured_apply(tensor *, tensor input, tensor thread_masks); +void atg__sparse_semi_structured_apply_dense(tensor *, tensor input, tensor thread_masks); +void atg__sparse_semi_structured_linear(tensor *, tensor input, tensor weight, tensor meta, tensor bias, char* activation_ptr, int activation_len, int out_dtype); +void atg__sparse_semi_structured_mm(tensor *, tensor mat1, tensor mat1_meta, tensor mat2, int out_dtype); +void atg__sparse_semi_structured_tile(tensor *, tensor input, char* algorithm_ptr, int algorithm_len, int use_cutlass); void atg__sparse_softmax(tensor *, tensor self, int64_t dim, int half_to_float); void atg__sparse_softmax_backward_data(tensor *, tensor grad_output, tensor output, int64_t dim, tensor self); void atg__sparse_softmax_backward_data_out(tensor *, tensor out, tensor grad_output, tensor output, int64_t dim, tensor self); @@ -342,6 +396,7 @@ void atg__sparse_sum_dim_out(tensor *, tensor out, tensor self, int64_t *dim_dat void atg__sparse_sum_dtype(tensor *, tensor self, int dtype); void atg__spdiags(tensor *, tensor diagonals, tensor offsets, int64_t *shape_data, int shape_len, int8_t layout); void atg__spdiags_out(tensor *, tensor out, tensor diagonals, tensor offsets, int64_t *shape_data, int shape_len, int8_t layout); +void atg__spsolve(tensor *, tensor A, tensor B, int left); void atg__stack(tensor *, tensor *tensors_data, int tensors_len, int64_t dim); void atg__stack_out(tensor *, tensor out, tensor *tensors_data, int tensors_len, int64_t dim); void atg__standard_gamma(tensor *, tensor self); @@ -365,6 +420,7 @@ void atg__test_optional_floatlist(tensor *, tensor values, double *addends_data, void atg__test_optional_floatlist_out(tensor *, tensor out, tensor values, double *addends_data, int addends_len); void atg__test_optional_intlist(tensor *, tensor values, int64_t *addends_data, int addends_len); void atg__test_optional_intlist_out(tensor *, tensor out, tensor values, int64_t *addends_data, int addends_len); +void atg__test_parallel_materialize(tensor *, tensor self, int64_t num_parallel, int skip_first); void atg__test_serialization_subcmul(tensor *, tensor self, tensor other); void atg__test_string_default(tensor *, tensor dummy, char* a_ptr, int a_len, char* b_ptr, int b_len); void atg__test_warn_in_autograd(tensor *, tensor self); @@ -402,8 +458,6 @@ void atg__unique2(tensor *, tensor self, int sorted, int return_inverse, int ret void atg__unique2_out(tensor *, tensor out0, tensor out1, tensor out2, tensor self, int sorted, int return_inverse, int return_counts); void atg__unique_out(tensor *, tensor out0, tensor out1, tensor self, int sorted, int return_inverse); void atg__unpack_dual(tensor *, tensor dual, int64_t level); -void atg__unsafe_index(tensor *, tensor self, tensor *indices_data, int indices_len); -void atg__unsafe_index_put(tensor *, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate); void atg__unsafe_view(tensor *, tensor self, int64_t *size_data, int size_len); void atg__unsafe_view_out(tensor *, tensor out, tensor self, int64_t *size_data, int size_len); void atg__upsample_bicubic2d_aa(tensor *, tensor self, int64_t *output_size_data, int output_size_len, int align_corners, double scales_h_v, uint8_t scales_h_null, double scales_w_v, uint8_t scales_w_null); @@ -438,12 +492,18 @@ void atg__values(tensor *, tensor self); void atg__values_copy(tensor *, tensor self); void atg__values_copy_out(tensor *, tensor out, tensor self); int64_t atg__version(tensor self); +void atg__weight_int4pack_mm(tensor *, tensor self, tensor mat2, int64_t qGroupSize, tensor qScaleAndZeros); +void atg__weight_int4pack_mm_for_cpu(tensor *, tensor self, tensor mat2, int64_t qGroupSize, tensor qScaleAndZeros); +void atg__weight_int4pack_mm_with_scales_and_zeros(tensor *, tensor self, tensor mat2, int64_t qGroupSize, tensor qScale, tensor qZeros); +void atg__weight_int8pack_mm(tensor *, tensor self, tensor mat2, tensor scales); void atg__weight_norm(tensor *, tensor v, tensor g, int64_t dim); void atg__weight_norm_differentiable_backward(tensor *, tensor grad_w, tensor saved_v, tensor saved_g, tensor saved_norms, int64_t dim); void atg__weight_norm_interface(tensor *, tensor v, tensor g, int64_t dim); void atg__weight_norm_interface_backward(tensor *, tensor grad_w, tensor saved_v, tensor saved_g, tensor saved_norms, int64_t dim); void atg__weight_norm_interface_backward_out(tensor *, tensor out0, tensor out1, tensor grad_w, tensor saved_v, tensor saved_g, tensor saved_norms, int64_t dim); void atg__weight_norm_interface_out(tensor *, tensor out0, tensor out1, tensor v, tensor g, int64_t dim); +void atg__wrapped_linear_prepack(tensor *, tensor weight, tensor weight_scale, tensor weight_zero_point, tensor bias); +void atg__wrapped_quantized_linear_prepacked(tensor *, tensor input, tensor input_scale, tensor input_zero_point, tensor packed_weight, tensor output_scale, tensor output_zero_point, int64_t out_channel); void atg_abs(tensor *, tensor self); void atg_abs_(tensor *, tensor self); void atg_abs_out(tensor *, tensor out, tensor self); @@ -457,6 +517,7 @@ void atg_acosh(tensor *, tensor self); void atg_acosh_(tensor *, tensor self); void atg_acosh_out(tensor *, tensor out, tensor self); void atg_adaptive_avg_pool1d(tensor *, tensor self, int64_t *output_size_data, int output_size_len); +void atg_adaptive_avg_pool1d_out(tensor *, tensor out, tensor self, int64_t *output_size_data, int output_size_len); void atg_adaptive_avg_pool2d(tensor *, tensor self, int64_t *output_size_data, int output_size_len); void atg_adaptive_avg_pool2d_out(tensor *, tensor out, tensor self, int64_t *output_size_data, int output_size_len); void atg_adaptive_avg_pool3d(tensor *, tensor self, int64_t *output_size_data, int output_size_len); @@ -488,6 +549,8 @@ void atg_addcmul_(tensor *, tensor self, tensor tensor1, tensor tensor2); void atg_addcmul_out(tensor *, tensor out, tensor self, tensor tensor1, tensor tensor2); void atg_addmm(tensor *, tensor self, tensor mat1, tensor mat2); void atg_addmm_(tensor *, tensor self, tensor mat1, tensor mat2); +void atg_addmm_dtype(tensor *, tensor self, tensor mat1, tensor mat2, int out_dtype); +void atg_addmm_dtype_out(tensor *, tensor out, tensor self, tensor mat1, tensor mat2, int out_dtype); void atg_addmm_out(tensor *, tensor out, tensor self, tensor mat1, tensor mat2); void atg_addmv(tensor *, tensor self, tensor mat, tensor vec); void atg_addmv_(tensor *, tensor self, tensor mat, tensor vec); @@ -507,6 +570,8 @@ tensor *atg_align_tensors(tensor *tensors_data, int tensors_len); void atg_all(tensor *, tensor self); void atg_all_all_out(tensor *, tensor out, tensor self); void atg_all_dim(tensor *, tensor self, int64_t dim, int keepdim); +void atg_all_dims(tensor *, tensor self, int64_t *dim_data, int dim_len, int keepdim); +void atg_all_dims_out(tensor *, tensor out, tensor self, int64_t *dim_data, int dim_len, int keepdim); void atg_all_out(tensor *, tensor out, tensor self, int64_t dim, int keepdim); int atg_allclose(tensor self, tensor other, double rtol, double atol, int equal_nan); void atg_alpha_dropout(tensor *, tensor input, double p, int train); @@ -522,6 +587,8 @@ void atg_angle_out(tensor *, tensor out, tensor self); void atg_any(tensor *, tensor self); void atg_any_all_out(tensor *, tensor out, tensor self); void atg_any_dim(tensor *, tensor self, int64_t dim, int keepdim); +void atg_any_dims(tensor *, tensor self, int64_t *dim_data, int dim_len, int keepdim); +void atg_any_dims_out(tensor *, tensor out, tensor self, int64_t *dim_data, int dim_len, int keepdim); void atg_any_out(tensor *, tensor out, tensor self, int64_t dim, int keepdim); void atg_arange(tensor *, scalar end, int options_kind, int options_device); void atg_arange_start(tensor *, scalar start, scalar end, int options_kind, int options_device); @@ -583,6 +650,7 @@ tensor *atg_atleast_2d_sequence(tensor *tensors_data, int tensors_len); void atg_atleast_3d(tensor *, tensor self); tensor *atg_atleast_3d_sequence(tensor *tensors_data, int tensors_len); void atg_avg_pool1d(tensor *, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int ceil_mode, int count_include_pad); +void atg_avg_pool1d_out(tensor *, tensor out, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int ceil_mode, int count_include_pad); void atg_avg_pool2d(tensor *, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int ceil_mode, int count_include_pad, int64_t divisor_override_v, uint8_t divisor_override_null); void atg_avg_pool2d_backward(tensor *, tensor grad_output, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int ceil_mode, int count_include_pad, int64_t divisor_override_v, uint8_t divisor_override_null); void atg_avg_pool2d_backward_grad_input(tensor *, tensor grad_input, tensor grad_output, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int ceil_mode, int count_include_pad, int64_t divisor_override_v, uint8_t divisor_override_null); @@ -593,6 +661,8 @@ void atg_avg_pool3d_backward_grad_input(tensor *, tensor grad_input, tensor grad void atg_avg_pool3d_out(tensor *, tensor out, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int ceil_mode, int count_include_pad, int64_t divisor_override_v, uint8_t divisor_override_null); void atg_baddbmm(tensor *, tensor self, tensor batch1, tensor batch2, scalar beta, scalar alpha); void atg_baddbmm_(tensor *, tensor self, tensor batch1, tensor batch2); +void atg_baddbmm_dtype(tensor *, tensor self, tensor batch1, tensor batch2, int out_dtype, scalar beta, scalar alpha); +void atg_baddbmm_dtype_out(tensor *, tensor out, tensor self, tensor batch1, tensor batch2, int out_dtype); void atg_baddbmm_out(tensor *, tensor out, tensor self, tensor batch1, tensor batch2); void atg_bartlett_window(tensor *, int64_t window_length, int options_kind, int options_device); void atg_bartlett_window_out(tensor *, tensor out, int64_t window_length); @@ -679,6 +749,8 @@ void atg_blackman_window_periodic_out(tensor *, tensor out, int64_t window_lengt void atg_block_diag(tensor *, tensor *tensors_data, int tensors_len); void atg_block_diag_out(tensor *, tensor out, tensor *tensors_data, int tensors_len); void atg_bmm(tensor *, tensor self, tensor mat2); +void atg_bmm_dtype(tensor *, tensor self, tensor mat2, int out_dtype); +void atg_bmm_dtype_out(tensor *, tensor out, tensor self, tensor mat2, int out_dtype); void atg_bmm_out(tensor *, tensor out, tensor self, tensor mat2); tensor *atg_broadcast_tensors(tensor *tensors_data, int tensors_len); void atg_broadcast_to(tensor *, tensor self, int64_t *size_data, int size_len); @@ -686,7 +758,7 @@ void atg_bucketize(tensor *, tensor self, tensor boundaries, int out_int32, int void atg_bucketize_scalar(tensor *, scalar self_scalar, tensor boundaries, int out_int32, int right); void atg_bucketize_scalar_out(tensor *, tensor out, scalar self_scalar, tensor boundaries, int out_int32, int right); void atg_bucketize_tensor_out(tensor *, tensor out, tensor self, tensor boundaries, int out_int32, int right); -int atg_can_cast(int from, int to); +int atg_can_cast(int from_, int to); void atg_cartesian_prod(tensor *, tensor *tensors_data, int tensors_len); void atg_cat(tensor *, tensor *tensors_data, int tensors_len, int64_t dim); void atg_cat_out(tensor *, tensor out, tensor *tensors_data, int tensors_len, int64_t dim); @@ -984,6 +1056,8 @@ void atg_fake_quantize_per_tensor_affine_cachemask_out(tensor *, tensor out0, te void atg_fake_quantize_per_tensor_affine_tensor_qparams(tensor *, tensor self, tensor scale, tensor zero_point, int64_t quant_min, int64_t quant_max); void atg_fbgemm_linear_fp16_weight(tensor *, tensor input, tensor packed_weight, tensor bias); void atg_fbgemm_linear_fp16_weight_fp32_activation(tensor *, tensor input, tensor packed_weight, tensor bias); +void atg_fbgemm_linear_fp16_weight_fp32_activation_out(tensor *, tensor input, tensor packed_weight, tensor bias, tensor output); +void atg_fbgemm_linear_fp16_weight_out(tensor *, tensor input, tensor packed_weight, tensor bias, tensor output); void atg_fbgemm_linear_int8_weight(tensor *, tensor input, tensor weight, tensor packed, tensor col_offsets, scalar weight_scale, scalar weight_zero_point, tensor bias); void atg_fbgemm_linear_int8_weight_fp32_activation(tensor *, tensor input, tensor weight, tensor packed, tensor col_offsets, scalar weight_scale, scalar weight_zero_point, tensor bias); void atg_fbgemm_pack_gemm_matrix_fp16(tensor *, tensor input); @@ -1066,6 +1140,7 @@ void atg_floor_divide_(tensor *, tensor self, tensor other); void atg_floor_divide_out(tensor *, tensor out, tensor self, tensor other); void atg_floor_divide_scalar(tensor *, tensor self, scalar other); void atg_floor_divide_scalar_(tensor *, tensor self, scalar other); +void atg_floor_divide_scalar_out(tensor *, tensor out, tensor self, scalar other); void atg_floor_out(tensor *, tensor out, tensor self); void atg_fmax(tensor *, tensor self, tensor other); void atg_fmax_out(tensor *, tensor out, tensor self, tensor other); @@ -1190,6 +1265,8 @@ void atg_hardtanh_(tensor *, tensor self); void atg_hardtanh_backward(tensor *, tensor grad_output, tensor self, scalar min_val, scalar max_val); void atg_hardtanh_backward_grad_input(tensor *, tensor grad_input, tensor grad_output, tensor self, scalar min_val, scalar max_val); void atg_hardtanh_out(tensor *, tensor out, tensor self); +void atg_hash_tensor(tensor *, tensor self, int64_t *dim_data, int dim_len, int keepdim, int64_t mode); +void atg_hash_tensor_out(tensor *, tensor out, tensor self, int64_t *dim_data, int dim_len, int keepdim, int64_t mode); void atg_heaviside(tensor *, tensor self, tensor values); void atg_heaviside_(tensor *, tensor self, tensor values); void atg_heaviside_out(tensor *, tensor out, tensor self, tensor values); @@ -1221,7 +1298,6 @@ void atg_igammac_out(tensor *, tensor out, tensor self, tensor other); void atg_im2col(tensor *, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *dilation_data, int dilation_len, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len); void atg_im2col_out(tensor *, tensor out, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *dilation_data, int dilation_len, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len); void atg_imag(tensor *, tensor self); -void atg_index(tensor *, tensor self, tensor *indices_data, int indices_len); void atg_index_add(tensor *, tensor self, int64_t dim, tensor index, tensor source); void atg_index_add_(tensor *, tensor self, int64_t dim, tensor index, tensor source); void atg_index_add_out(tensor *, tensor out, tensor self, int64_t dim, tensor index, tensor source); @@ -1234,16 +1310,12 @@ void atg_index_fill_int_scalar_out(tensor *, tensor out, tensor self, int64_t di void atg_index_fill_int_tensor(tensor *, tensor self, int64_t dim, tensor index, tensor value); void atg_index_fill_int_tensor_(tensor *, tensor self, int64_t dim, tensor index, tensor value); void atg_index_fill_int_tensor_out(tensor *, tensor out, tensor self, int64_t dim, tensor index, tensor value); -void atg_index_put(tensor *, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate); -void atg_index_put_(tensor *, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate); -void atg_index_put_out(tensor *, tensor out, tensor self, tensor *indices_data, int indices_len, tensor values, int accumulate); void atg_index_reduce(tensor *, tensor self, int64_t dim, tensor index, tensor source, char* reduce_ptr, int reduce_len, int include_self); void atg_index_reduce_(tensor *, tensor self, int64_t dim, tensor index, tensor source, char* reduce_ptr, int reduce_len, int include_self); void atg_index_reduce_out(tensor *, tensor out, tensor self, int64_t dim, tensor index, tensor source, char* reduce_ptr, int reduce_len, int include_self); void atg_index_select(tensor *, tensor self, int64_t dim, tensor index); void atg_index_select_backward(tensor *, tensor grad, int64_t *self_sizes_data, int self_sizes_len, int64_t dim, tensor index); void atg_index_select_out(tensor *, tensor out, tensor self, int64_t dim, tensor index); -void atg_index_tensor_out(tensor *, tensor out, tensor self, tensor *indices_data, int indices_len); void atg_indices(tensor *, tensor self); void atg_indices_copy(tensor *, tensor self); void atg_indices_copy_out(tensor *, tensor out, tensor self); @@ -1439,6 +1511,12 @@ void atg_linear(tensor *, tensor input, tensor weight, tensor bias); void atg_linear_out(tensor *, tensor out, tensor input, tensor weight, tensor bias); void atg_linspace(tensor *, scalar start, scalar end, int64_t steps, int options_kind, int options_device); void atg_linspace_out(tensor *, tensor out, scalar start, scalar end, int64_t steps); +void atg_linspace_scalar_tensor(tensor *, scalar start, tensor end, int64_t steps, int options_kind, int options_device); +void atg_linspace_scalar_tensor_out(tensor *, tensor out, scalar start, tensor end, int64_t steps); +void atg_linspace_tensor_scalar(tensor *, tensor start, scalar end, int64_t steps, int options_kind, int options_device); +void atg_linspace_tensor_scalar_out(tensor *, tensor out, tensor start, scalar end, int64_t steps); +void atg_linspace_tensor_tensor(tensor *, tensor start, tensor end, int64_t steps, int options_kind, int options_device); +void atg_linspace_tensor_tensor_out(tensor *, tensor out, tensor start, tensor end, int64_t steps); void atg_log(tensor *, tensor self); void atg_log10(tensor *, tensor self); void atg_log10_(tensor *, tensor self); @@ -1486,6 +1564,12 @@ void atg_logit_backward_grad_input(tensor *, tensor grad_input, tensor grad_outp void atg_logit_out(tensor *, tensor out, tensor self, double eps_v, uint8_t eps_null); void atg_logspace(tensor *, scalar start, scalar end, int64_t steps, double base, int options_kind, int options_device); void atg_logspace_out(tensor *, tensor out, scalar start, scalar end, int64_t steps, double base); +void atg_logspace_scalar_tensor(tensor *, scalar start, tensor end, int64_t steps, double base, int options_kind, int options_device); +void atg_logspace_scalar_tensor_out(tensor *, tensor out, scalar start, tensor end, int64_t steps, double base); +void atg_logspace_tensor_scalar(tensor *, tensor start, scalar end, int64_t steps, double base, int options_kind, int options_device); +void atg_logspace_tensor_scalar_out(tensor *, tensor out, tensor start, scalar end, int64_t steps, double base); +void atg_logspace_tensor_tensor(tensor *, tensor start, tensor end, int64_t steps, double base, int options_kind, int options_device); +void atg_logspace_tensor_tensor_out(tensor *, tensor out, tensor start, tensor end, int64_t steps, double base); void atg_logsumexp(tensor *, tensor self, int64_t *dim_data, int dim_len, int keepdim); void atg_logsumexp_out(tensor *, tensor out, tensor self, int64_t *dim_data, int dim_len, int keepdim); void atg_lstm(tensor *, tensor input, tensor *hx_data, int hx_len, tensor *params_data, int params_len, int has_biases, int64_t num_layers, double dropout, int train, int bidirectional, int batch_first); @@ -1510,6 +1594,7 @@ void atg_masked_fill_tensor_(tensor *, tensor self, tensor mask, tensor value); void atg_masked_fill_tensor_out(tensor *, tensor out, tensor self, tensor mask, tensor value); void atg_masked_scatter(tensor *, tensor self, tensor mask, tensor source); void atg_masked_scatter_(tensor *, tensor self, tensor mask, tensor source); +void atg_masked_scatter_backward(tensor *, tensor grad_output, tensor mask, int64_t *sizes_data, int sizes_len); void atg_masked_scatter_out(tensor *, tensor out, tensor self, tensor mask, tensor source); void atg_masked_select(tensor *, tensor self, tensor mask); void atg_masked_select_backward(tensor *, tensor grad, tensor input, tensor mask); @@ -1549,6 +1634,7 @@ void atg_maximum(tensor *, tensor self, tensor other); void atg_maximum_out(tensor *, tensor out, tensor self, tensor other); void atg_mean(tensor *, tensor self, int dtype); void atg_mean_dim(tensor *, tensor self, int64_t *dim_data, int dim_len, int keepdim, int dtype); +void atg_mean_dtype_out(tensor *, tensor out, tensor self, int dtype); void atg_mean_out(tensor *, tensor out, tensor self, int64_t *dim_data, int dim_len, int keepdim, int dtype); void atg_median(tensor *, tensor self); void atg_median_dim(tensor *, tensor self, int64_t dim, int keepdim); @@ -1605,13 +1691,15 @@ void atg_mkldnn_max_pool3d_backward_out(tensor *, tensor out, tensor grad_output void atg_mkldnn_max_pool3d_out(tensor *, tensor out, tensor self, int64_t *kernel_size_data, int kernel_size_len, int64_t *stride_data, int stride_len, int64_t *padding_data, int padding_len, int64_t *dilation_data, int dilation_len, int ceil_mode); void atg_mkldnn_reorder_conv2d_weight(tensor *, tensor self, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len, int64_t *dilation_data, int dilation_len, int64_t groups, int64_t *input_size_data, int input_size_len); void atg_mkldnn_reorder_conv2d_weight_out(tensor *, tensor out, tensor self, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len, int64_t *dilation_data, int dilation_len, int64_t groups, int64_t *input_size_data, int input_size_len); -void atg_mkldnn_reorder_conv3d_weight(tensor *, tensor self, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len, int64_t *dilation_data, int dilation_len, int64_t groups); -void atg_mkldnn_reorder_conv3d_weight_out(tensor *, tensor out, tensor self, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len, int64_t *dilation_data, int dilation_len, int64_t groups); +void atg_mkldnn_reorder_conv3d_weight(tensor *, tensor self, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len, int64_t *dilation_data, int dilation_len, int64_t groups, int64_t *input_size_data, int input_size_len); +void atg_mkldnn_reorder_conv3d_weight_out(tensor *, tensor out, tensor self, int64_t *padding_data, int padding_len, int64_t *stride_data, int stride_len, int64_t *dilation_data, int dilation_len, int64_t groups, int64_t *input_size_data, int input_size_len); void atg_mkldnn_rnn_layer(tensor *, tensor input, tensor weight0, tensor weight1, tensor weight2, tensor weight3, tensor hx_, tensor cx_, int reverse, int64_t *batch_sizes_data, int batch_sizes_len, int64_t mode, int64_t hidden_size, int64_t num_layers, int has_biases, int bidirectional, int batch_first, int train); void atg_mkldnn_rnn_layer_backward(tensor *, tensor input, tensor weight1, tensor weight2, tensor weight3, tensor weight4, tensor hx_, tensor cx_tmp, tensor output, tensor hy_, tensor cy_, tensor grad_output, tensor grad_hy, tensor grad_cy, int reverse, int64_t mode, int64_t hidden_size, int64_t num_layers, int has_biases, int train, int bidirectional, int64_t *batch_sizes_data, int batch_sizes_len, int batch_first, tensor workspace); void atg_mkldnn_rnn_layer_backward_out(tensor *, tensor out0, tensor out1, tensor out2, tensor out3, tensor out4, tensor out5, tensor out6, tensor input, tensor weight1, tensor weight2, tensor weight3, tensor weight4, tensor hx_, tensor cx_tmp, tensor output, tensor hy_, tensor cy_, tensor grad_output, tensor grad_hy, tensor grad_cy, int reverse, int64_t mode, int64_t hidden_size, int64_t num_layers, int has_biases, int train, int bidirectional, int64_t *batch_sizes_data, int batch_sizes_len, int batch_first, tensor workspace); void atg_mkldnn_rnn_layer_out(tensor *, tensor out0, tensor out1, tensor out2, tensor out3, tensor input, tensor weight0, tensor weight1, tensor weight2, tensor weight3, tensor hx_, tensor cx_, int reverse, int64_t *batch_sizes_data, int batch_sizes_len, int64_t mode, int64_t hidden_size, int64_t num_layers, int has_biases, int bidirectional, int batch_first, int train); void atg_mm(tensor *, tensor self, tensor mat2); +void atg_mm_dtype(tensor *, tensor self, tensor mat2, int out_dtype); +void atg_mm_dtype_out(tensor *, tensor out, tensor self, tensor mat2, int out_dtype); void atg_mm_out(tensor *, tensor out, tensor self, tensor mat2); void atg_mode(tensor *, tensor self, int64_t dim, int keepdim); void atg_mode_values(tensor *, tensor values, tensor indices, tensor self, int64_t dim, int keepdim); @@ -1759,7 +1847,7 @@ void atg_outer(tensor *, tensor self, tensor vec2); void atg_outer_out(tensor *, tensor out, tensor self, tensor vec2); int64_t atg_output_nr(tensor self); void atg_pad(tensor *, tensor self, int64_t *pad_data, int pad_len, char* mode_ptr, int mode_len, double value_v, uint8_t value_null); -void atg_pad_sequence(tensor *, tensor *sequences_data, int sequences_len, int batch_first, double padding_value); +void atg_pad_sequence(tensor *, tensor *sequences_data, int sequences_len, int batch_first, double padding_value, char* padding_side_ptr, int padding_side_len); void atg_pairwise_distance(tensor *, tensor x1, tensor x2, double p, double eps, int keepdim); void atg_pdist(tensor *, tensor self, double p); void atg_permute(tensor *, tensor self, int64_t *dims_data, int dims_len); @@ -1840,6 +1928,8 @@ void atg_randint_like(tensor *, tensor self, int64_t high); void atg_randint_like_low_dtype(tensor *, tensor self, int64_t low, int64_t high); void atg_randint_like_low_dtype_out(tensor *, tensor out, tensor self, int64_t low, int64_t high); void atg_randint_like_out(tensor *, tensor out, tensor self, int64_t high); +void atg_randint_like_tensor(tensor *, tensor self, tensor high); +void atg_randint_like_tensor_out(tensor *, tensor out, tensor self, tensor high); void atg_randint_low(tensor *, int64_t low, int64_t high, int64_t *size_data, int size_len, int options_kind, int options_device); void atg_randint_low_out(tensor *, tensor out, int64_t low, int64_t high, int64_t *size_data, int size_len); void atg_randint_out(tensor *, tensor out, int64_t high, int64_t *size_data, int size_len); @@ -1928,6 +2018,7 @@ void atg_resize_out(tensor *, tensor out, tensor self, int64_t *size_data, int s void atg_resolve_conj(tensor *, tensor self); void atg_resolve_neg(tensor *, tensor self); int atg_retains_grad(tensor self); +void atg_rms_norm(tensor *, tensor input, int64_t *normalized_shape_data, int normalized_shape_len, tensor weight, double eps_v, uint8_t eps_null); void atg_rnn_relu(tensor *, tensor input, tensor hx, tensor *params_data, int params_len, int has_biases, int64_t num_layers, double dropout, int train, int bidirectional, int batch_first); void atg_rnn_relu_cell(tensor *, tensor input, tensor hx, tensor w_ih, tensor w_hh, tensor b_ih, tensor b_hh); void atg_rnn_relu_data(tensor *, tensor data, tensor batch_sizes, tensor hx, tensor *params_data, int params_len, int has_biases, int64_t num_layers, double dropout, int train, int bidirectional); @@ -1955,6 +2046,7 @@ void atg_rrelu_with_noise(tensor *, tensor self, tensor noise, int training); void atg_rrelu_with_noise_(tensor *, tensor self, tensor noise, int training); void atg_rrelu_with_noise_backward(tensor *, tensor grad_output, tensor self, tensor noise, scalar lower, scalar upper, int training, int self_is_result); void atg_rrelu_with_noise_backward_out(tensor *, tensor out, tensor grad_output, tensor self, tensor noise, scalar lower, scalar upper, int training, int self_is_result); +void atg_rrelu_with_noise_functional(tensor *, tensor self, tensor noise, int training); void atg_rrelu_with_noise_out(tensor *, tensor out, tensor self, tensor noise, int training); void atg_rsqrt(tensor *, tensor self); void atg_rsqrt_(tensor *, tensor self); @@ -1965,7 +2057,7 @@ void atg_rsub_scalar_out(tensor *, tensor out, tensor self, scalar other); void atg_rsub_tensor_out(tensor *, tensor out, tensor self, tensor other); void atg_scalar_tensor(tensor *, scalar s, int options_kind, int options_device); void atg_scalar_tensor_out(tensor *, tensor out, scalar s); -void atg_scaled_dot_product_attention(tensor *, tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, double scale_v, uint8_t scale_null); +void atg_scaled_dot_product_attention(tensor *, tensor query, tensor key, tensor value, tensor attn_mask, double dropout_p, int is_causal, double scale_v, uint8_t scale_null, int enable_gqa); void atg_scatter(tensor *, tensor self, int64_t dim, tensor index, tensor src); void atg_scatter_(tensor *, tensor self, int64_t dim, tensor index, tensor src); void atg_scatter_add(tensor *, tensor self, int64_t dim, tensor index, tensor src); @@ -2036,6 +2128,7 @@ void atg_slice_backward(tensor *, tensor grad_output, int64_t *input_sizes_data, void atg_slice_backward_out(tensor *, tensor out, tensor grad_output, int64_t *input_sizes_data, int input_sizes_len, int64_t dim, int64_t start, int64_t end, int64_t step); void atg_slice_copy(tensor *, tensor self, int64_t dim, int64_t start_v, uint8_t start_null, int64_t end_v, uint8_t end_null, int64_t step); void atg_slice_copy_tensor_out(tensor *, tensor out, tensor self, int64_t dim, int64_t start_v, uint8_t start_null, int64_t end_v, uint8_t end_null, int64_t step); +void atg_slice_inverse(tensor *, tensor self, tensor src, int64_t dim, int64_t start_v, uint8_t start_null, int64_t end_v, uint8_t end_null, int64_t step); void atg_slice_scatter(tensor *, tensor self, tensor src, int64_t dim, int64_t start_v, uint8_t start_null, int64_t end_v, uint8_t end_null, int64_t step); void atg_slice_scatter_out(tensor *, tensor out, tensor self, tensor src, int64_t dim, int64_t start_v, uint8_t start_null, int64_t end_v, uint8_t end_null, int64_t step); void atg_slogdet(tensor *, tensor self); @@ -2304,8 +2397,8 @@ void atg_std_mean_correction(tensor *, tensor self, int64_t *dim_data, int dim_l void atg_std_mean_correction_out(tensor *, tensor out0, tensor out1, tensor self, int64_t *dim_data, int dim_len, scalar correction, int keepdim); void atg_std_mean_dim(tensor *, tensor self, int64_t *dim_data, int dim_len, int unbiased, int keepdim); void atg_std_out(tensor *, tensor out, tensor self, int64_t *dim_data, int dim_len, int unbiased, int keepdim); -void atg_stft(tensor *, tensor self, int64_t n_fft, int64_t hop_length_v, uint8_t hop_length_null, int64_t win_length_v, uint8_t win_length_null, tensor window, int normalized, int onesided, int return_complex); -void atg_stft_center(tensor *, tensor self, int64_t n_fft, int64_t hop_length_v, uint8_t hop_length_null, int64_t win_length_v, uint8_t win_length_null, tensor window, int center, char* pad_mode_ptr, int pad_mode_len, int normalized, int onesided, int return_complex); +void atg_stft(tensor *, tensor self, int64_t n_fft, int64_t hop_length_v, uint8_t hop_length_null, int64_t win_length_v, uint8_t win_length_null, tensor window, int normalized, int onesided, int return_complex, int align_to_window); +void atg_stft_center(tensor *, tensor self, int64_t n_fft, int64_t hop_length_v, uint8_t hop_length_null, int64_t win_length_v, uint8_t win_length_null, tensor window, int center, char* pad_mode_ptr, int pad_mode_len, int normalized, int onesided, int return_complex, int align_to_window); void atg_sub(tensor *, tensor self, tensor other); void atg_sub_(tensor *, tensor self, tensor other); void atg_sub_out(tensor *, tensor out, tensor self, tensor other); @@ -2446,6 +2539,7 @@ void atg_upsample_bilinear2d_backward(tensor *, tensor grad_output, int64_t *out void atg_upsample_bilinear2d_backward_grad_input(tensor *, tensor grad_input, tensor grad_output, int64_t *output_size_data, int output_size_len, int64_t *input_size_data, int input_size_len, int align_corners, double scales_h_v, uint8_t scales_h_null, double scales_w_v, uint8_t scales_w_null); void atg_upsample_bilinear2d_out(tensor *, tensor out, tensor self, int64_t *output_size_data, int output_size_len, int align_corners, double scales_h_v, uint8_t scales_h_null, double scales_w_v, uint8_t scales_w_null); void atg_upsample_bilinear2d_vec(tensor *, tensor input, int64_t *output_size_data, int output_size_len, int align_corners, double *scale_factors_data, int scale_factors_len); +void atg_upsample_bilinear2d_vec_out(tensor *, tensor out, tensor input, int64_t *output_size_data, int output_size_len, int align_corners, double *scale_factors_data, int scale_factors_len); void atg_upsample_linear1d(tensor *, tensor self, int64_t *output_size_data, int output_size_len, int align_corners, double scales_v, uint8_t scales_null); void atg_upsample_linear1d_backward(tensor *, tensor grad_output, int64_t *output_size_data, int output_size_len, int64_t *input_size_data, int input_size_len, int align_corners, double scales_v, uint8_t scales_null); void atg_upsample_linear1d_backward_grad_input(tensor *, tensor grad_input, tensor grad_output, int64_t *output_size_data, int output_size_len, int64_t *input_size_data, int input_size_len, int align_corners, double scales_v, uint8_t scales_null); @@ -2461,6 +2555,7 @@ void atg_upsample_nearest2d_backward(tensor *, tensor grad_output, int64_t *outp void atg_upsample_nearest2d_backward_grad_input(tensor *, tensor grad_input, tensor grad_output, int64_t *output_size_data, int output_size_len, int64_t *input_size_data, int input_size_len, double scales_h_v, uint8_t scales_h_null, double scales_w_v, uint8_t scales_w_null); void atg_upsample_nearest2d_out(tensor *, tensor out, tensor self, int64_t *output_size_data, int output_size_len, double scales_h_v, uint8_t scales_h_null, double scales_w_v, uint8_t scales_w_null); void atg_upsample_nearest2d_vec(tensor *, tensor input, int64_t *output_size_data, int output_size_len, double *scale_factors_data, int scale_factors_len); +void atg_upsample_nearest2d_vec_out(tensor *, tensor out, tensor input, int64_t *output_size_data, int output_size_len, double *scale_factors_data, int scale_factors_len); void atg_upsample_nearest3d(tensor *, tensor self, int64_t *output_size_data, int output_size_len, double scales_d_v, uint8_t scales_d_null, double scales_h_v, uint8_t scales_h_null, double scales_w_v, uint8_t scales_w_null); void atg_upsample_nearest3d_backward(tensor *, tensor grad_output, int64_t *output_size_data, int output_size_len, int64_t *input_size_data, int input_size_len, double scales_d_v, uint8_t scales_d_null, double scales_h_v, uint8_t scales_h_null, double scales_w_v, uint8_t scales_w_null); void atg_upsample_nearest3d_backward_grad_input(tensor *, tensor grad_input, tensor grad_output, int64_t *output_size_data, int output_size_len, int64_t *input_size_data, int input_size_len, double scales_d_v, uint8_t scales_d_null, double scales_h_v, uint8_t scales_h_null, double scales_w_v, uint8_t scales_w_null); diff --git a/setup-gotch.sh b/setup-gotch.sh index 3c72e96..d083057 100644 --- a/setup-gotch.sh +++ b/setup-gotch.sh @@ -1,7 +1,7 @@ #!/bin/bash -GOTCH_VERSION="${GOTCH_VER:-v0.9.0}" -CUDA_VERSION="${CUDA_VER:-11.8}" +GOTCH_VERSION="${GOTCH_VER:-v0.10.0}" +CUDA_VERSION="${CUDA_VER:-12.6}" if [ -z $GOPATH ]; then GOPATH="$HOME/go" diff --git a/setup-libtorch.sh b/setup-libtorch.sh old mode 100644 new mode 100755 index c2ec6dd..f6c996b --- a/setup-libtorch.sh +++ b/setup-libtorch.sh @@ -1,7 +1,7 @@ #!/bin/bash -LIBTORCH_VERSION="${LIBTORCH_VER:-2.1.0}" -CUDA_VERSION="${CUDA_VER:-11.8}" +LIBTORCH_VERSION="${LIBTORCH_VER:-2.10.0}" +CUDA_VERSION="${CUDA_VER:-12.6}" if [ "${CUDA_VERSION}" == "cpu" ]; then CU_VERSION="cpu" diff --git a/ts/must-tensor-generated.go b/ts/must-tensor-generated.go index 79655ab..e717608 100644 --- a/ts/must-tensor-generated.go +++ b/ts/must-tensor-generated.go @@ -409,6 +409,46 @@ func(ts *Tensor) Must_AutocastToReducedPrecision(cudaEnabled bool, cpuEnabled bo return retVal } +func Must_BatchNormNoUpdate(input *Tensor, weight *Tensor, bias *Tensor, runningMean *Tensor, runningVar *Tensor, momentum float64, eps float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor) { + + retVal0, retVal1, retVal2, retVal3, err := _BatchNormNoUpdate(input, weight, bias, runningMean, runningVar, momentum, eps) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1, retVal2, retVal3 +} + +func Must_BatchNormNoUpdateOut(out0 *Tensor, out1 *Tensor, out2 *Tensor, out3 *Tensor, input *Tensor, weight *Tensor, bias *Tensor, runningMean *Tensor, runningVar *Tensor, momentum float64, eps float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor) { + + retVal0, retVal1, retVal2, retVal3, err := _BatchNormNoUpdateOut(out0, out1, out2, out3, input, weight, bias, runningMean, runningVar, momentum, eps) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1, retVal2, retVal3 +} + +func Must_BatchNormWithUpdate(input *Tensor, weight *Tensor, bias *Tensor, runningMean *Tensor, runningVar *Tensor, momentum float64, eps float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor) { + + retVal0, retVal1, retVal2, retVal3, err := _BatchNormWithUpdate(input, weight, bias, runningMean, runningVar, momentum, eps) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1, retVal2, retVal3 +} + +func Must_BatchNormWithUpdateFunctional(input *Tensor, weight *Tensor, bias *Tensor, runningMean *Tensor, runningVar *Tensor, momentum float64, eps float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, retVal4 *Tensor, retVal5 *Tensor) { + + retVal0, retVal1, retVal2, retVal3, retVal4, retVal5, err := _BatchNormWithUpdateFunctional(input, weight, bias, runningMean, runningVar, momentum, eps) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1, retVal2, retVal3, retVal4, retVal5 +} + +func Must_BatchNormWithUpdateOut(out *Tensor, saveMean *Tensor, saveInvstd *Tensor, reserve *Tensor, input *Tensor, weight *Tensor, bias *Tensor, runningMean *Tensor, runningVar *Tensor, momentum float64, eps float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor) { + + retVal0, retVal1, retVal2, retVal3, err := _BatchNormWithUpdateOut(out, saveMean, saveInvstd, reserve, input, weight, bias, runningMean, runningVar, momentum, eps) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1, retVal2, retVal3 +} + func(ts *Tensor) Must_CastByte(nonBlocking bool, del bool)(retVal *Tensor) { retVal, err := ts._CastByte(nonBlocking, del) @@ -505,6 +545,22 @@ func(ts *Tensor) Must_CholeskySolveHelperOut(out *Tensor, a *Tensor, upper bool, return retVal } +func Must_ChunkCat(tensors []*Tensor, dim int64, numChunks int64)(retVal *Tensor) { + + retVal, err := _ChunkCat(tensors, dim, numChunks) + if err != nil { log.Fatal(err) } + + return retVal +} + +func Must_ChunkCatOut(out *Tensor, tensors []*Tensor, dim int64, numChunks int64)(retVal *Tensor) { + + retVal, err := _ChunkCatOut(out, tensors, dim, numChunks) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) Must_Coalesce(del bool)(retVal *Tensor) { retVal, err := ts._Coalesce(del) @@ -649,6 +705,22 @@ func Must_ConvertIndicesFromCsrToCooOut(out *Tensor, crowIndices *Tensor, colInd return retVal } +func(ts *Tensor) Must_ConvertWeightToInt4pack(innerKTiles int64, del bool)(retVal *Tensor) { + + retVal, err := ts._ConvertWeightToInt4pack(innerKTiles, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_ConvertWeightToInt4packForCpu(innerKTiles int64, del bool)(retVal *Tensor) { + + retVal, err := ts._ConvertWeightToInt4packForCpu(innerKTiles, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func Must_Convolution(input *Tensor, weight *Tensor, bias *Tensor, stride []int64, padding []int64, dilation []int64, transposed bool, outputPadding []int64, groups int64, benchmark bool, deterministic bool, cudnnEnabled bool, allowTf32 bool)(retVal *Tensor) { retVal, err := _Convolution(input, weight, bias, stride, padding, dilation, transposed, outputPadding, groups, benchmark, deterministic, cudnnEnabled, allowTf32) @@ -721,9 +793,17 @@ func Must_CsltCompress(input *Tensor)(retVal *Tensor) { return retVal } -func Must_CsltSparseMm(compressedA *Tensor, denseB *Tensor, bias *Tensor, transposeResult bool)(retVal *Tensor) { +func Must_CsltSparseMm(compressedA *Tensor, denseB *Tensor, bias *Tensor, alpha *Tensor, outDtype gotch.DType, transposeResult bool, algId int64, splitK int64, splitKMode int64)(retVal *Tensor) { - retVal, err := _CsltSparseMm(compressedA, denseB, bias, transposeResult) + retVal, err := _CsltSparseMm(compressedA, denseB, bias, alpha, outDtype, transposeResult, algId, splitK, splitKMode) + if err != nil { log.Fatal(err) } + + return retVal +} + +func Must_CsltSparseMmSearch(compressedA *Tensor, denseB *Tensor, bias *Tensor, alpha *Tensor, outDtype gotch.DType, transposeResult bool)(retVal int64) { + + retVal, err := _CsltSparseMmSearch(compressedA, denseB, bias, alpha, outDtype, transposeResult) if err != nil { log.Fatal(err) } return retVal @@ -785,6 +865,14 @@ func Must_CtcLossTensorOut(out0 *Tensor, out1 *Tensor, logProbs *Tensor, targets return retVal0, retVal1 } +func Must_CudnnAttentionBackward(gradOut *Tensor, query *Tensor, key *Tensor, value *Tensor, out *Tensor, logsumexp *Tensor, philoxSeed *Tensor, philoxOffset *Tensor, attnBias *Tensor, cumSeqQ *Tensor, cumSeqK *Tensor, maxQ int64, maxK int64, dropoutP float64, isCausal bool, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor) { + + retVal0, retVal1, retVal2, err := _CudnnAttentionBackward(gradOut, query, key, value, out, logsumexp, philoxSeed, philoxOffset, attnBias, cumSeqQ, cumSeqK, maxQ, maxK, dropoutP, isCausal, scale) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1, retVal2 +} + func Must_CudnnCtcLoss(logProbs *Tensor, targets *Tensor, inputLengths []int64, targetLengths []int64, blank int64, deterministic bool, zeroInfinity bool)(retVal0 *Tensor, retVal1 *Tensor) { retVal0, retVal1, err := _CudnnCtcLoss(logProbs, targets, inputLengths, targetLengths, blank, deterministic, zeroInfinity) @@ -905,9 +993,25 @@ func Must_DirichletGradOut(out *Tensor, x *Tensor, alpha *Tensor, total *Tensor) return retVal } -func Must_EfficientAttentionBackward(gradOut_ *Tensor, query *Tensor, key *Tensor, value *Tensor, bias *Tensor, out *Tensor, cuSeqlensQ *Tensor, cuSeqlensK *Tensor, maxSeqlenK int64, maxSeqlenQ int64, logsumexp *Tensor, dropoutP float64, philoxSeed *Tensor, philoxOffset *Tensor, customMaskType int64, biasRequiresGrad bool, scale []float64, numSplitsKey []int64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor) { +func Must_DynQuantMatmul4bit(inp *Tensor, packedWeights *Tensor, blockSize int64, inFeatures int64, outFeatures int64)(retVal *Tensor) { - retVal0, retVal1, retVal2, retVal3, err := _EfficientAttentionBackward(gradOut_, query, key, value, bias, out, cuSeqlensQ, cuSeqlensK, maxSeqlenK, maxSeqlenQ, logsumexp, dropoutP, philoxSeed, philoxOffset, customMaskType, biasRequiresGrad, scale, numSplitsKey) + retVal, err := _DynQuantMatmul4bit(inp, packedWeights, blockSize, inFeatures, outFeatures) + if err != nil { log.Fatal(err) } + + return retVal +} + +func Must_DynQuantPack4bitWeight(weights *Tensor, scalesZeros *Tensor, bias *Tensor, blockSize int64, inFeatures int64, outFeatures int64)(retVal *Tensor) { + + retVal, err := _DynQuantPack4bitWeight(weights, scalesZeros, bias, blockSize, inFeatures, outFeatures) + if err != nil { log.Fatal(err) } + + return retVal +} + +func Must_EfficientAttentionBackward(gradOut_ *Tensor, query *Tensor, key *Tensor, value *Tensor, bias *Tensor, out *Tensor, cuSeqlensQ *Tensor, cuSeqlensK *Tensor, maxSeqlenQ int64, maxSeqlenK int64, logsumexp *Tensor, dropoutP float64, philoxSeed *Tensor, philoxOffset *Tensor, customMaskType int64, biasRequiresGrad bool, scale []float64, numSplitsKey []int64, windowSize []int64, sharedStorageDqdkdv bool)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor) { + + retVal0, retVal1, retVal2, retVal3, err := _EfficientAttentionBackward(gradOut_, query, key, value, bias, out, cuSeqlensQ, cuSeqlensK, maxSeqlenQ, maxSeqlenK, logsumexp, dropoutP, philoxSeed, philoxOffset, customMaskType, biasRequiresGrad, scale, numSplitsKey, windowSize, sharedStorageDqdkdv) if err != nil { log.Fatal(err) } return retVal0, retVal1, retVal2, retVal3 @@ -1177,9 +1281,9 @@ func(ts *Tensor) Must_FillMemEffDropoutMask_(dropoutP float64, seed int64, offse return } -func Must_FlashAttentionBackward(gradOut *Tensor, query *Tensor, key *Tensor, value *Tensor, out *Tensor, logsumexp *Tensor, cumSeqQ *Tensor, cumSeqK *Tensor, maxQ int64, maxK int64, dropoutP float64, isCausal bool, philoxSeed *Tensor, philoxOffset *Tensor, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor) { +func Must_FlashAttentionBackward(gradOut *Tensor, query *Tensor, key *Tensor, value *Tensor, out *Tensor, logsumexp *Tensor, cumSeqQ *Tensor, cumSeqK *Tensor, maxQ int64, maxK int64, dropoutP float64, isCausal bool, rngState *Tensor, unused *Tensor, scale []float64, windowSizeLeft []int64, windowSizeRight []int64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor) { - retVal0, retVal1, retVal2, err := _FlashAttentionBackward(gradOut, query, key, value, out, logsumexp, cumSeqQ, cumSeqK, maxQ, maxK, dropoutP, isCausal, philoxSeed, philoxOffset, scale) + retVal0, retVal1, retVal2, err := _FlashAttentionBackward(gradOut, query, key, value, out, logsumexp, cumSeqQ, cumSeqK, maxQ, maxK, dropoutP, isCausal, rngState, unused, scale, windowSizeLeft, windowSizeRight) if err != nil { log.Fatal(err) } return retVal0, retVal1, retVal2 @@ -1209,6 +1313,14 @@ func(ts *Tensor) Must_FunctionalAssertAsync(assertMsg string, depToken *Tensor, return retVal } +func Must_FunctionalAssertScalar(selfScalar *Scalar, assertMsg string, depToken *Tensor)(retVal *Tensor) { + + retVal, err := _FunctionalAssertScalar(selfScalar, assertMsg, depToken) + if err != nil { log.Fatal(err) } + + return retVal +} + func Must_FunctionalSymConstrainRange(size *Scalar, min []int64, max []int64, depToken *Tensor)(retVal *Tensor) { retVal, err := _FunctionalSymConstrainRange(size, min, max, depToken) @@ -1265,9 +1377,17 @@ func(ts *Tensor) Must_FusedMovingAvgObsFqHelperOut(out0 *Tensor, out1 *Tensor, o return retVal0, retVal1 } -func Must_FusedSdpChoice(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, scale []float64)(retVal int64) { +func Must_FusedRmsNorm(input *Tensor, normalizedShape []int64, weight *Tensor, eps []float64)(retVal0 *Tensor, retVal1 *Tensor) { + + retVal0, retVal1, err := _FusedRmsNorm(input, normalizedShape, weight, eps) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1 +} + +func Must_FusedSdpChoice(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, scale []float64, enableGqa bool)(retVal int64) { - retVal, err := _FusedSdpChoice(query, key, value, attnMask, dropoutP, isCausal, scale) + retVal, err := _FusedSdpChoice(query, key, value, attnMask, dropoutP, isCausal, scale, enableGqa) if err != nil { log.Fatal(err) } return retVal @@ -1329,6 +1449,14 @@ func Must_GridSampler2dCpuFallbackOut(out *Tensor, input *Tensor, grid *Tensor, return retVal } +func(ts *Tensor) Must_GroupedMm(mat2 *Tensor, offs *Tensor, bias *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor) { + + retVal, err := ts._GroupedMm(mat2, offs, bias, outDtype, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) Must_HasCompatibleShallowCopyType(from *Tensor, del bool)(retVal bool) { retVal, err := ts._HasCompatibleShallowCopyType(from, del) @@ -1369,22 +1497,6 @@ func(ts *Tensor) Must_HistogramddFromBinTensorsOut(out *Tensor, bins []*Tensor, return retVal } -func(ts *Tensor) Must_IndexPutImpl(indices []*Tensor, values *Tensor, accumulate bool, unsafety bool, del bool)(retVal *Tensor) { - - retVal, err := ts._IndexPutImpl(indices, values, accumulate, unsafety, del) - if err != nil { log.Fatal(err) } - - return retVal -} - -func(ts *Tensor) Must_IndexPutImplOut(out *Tensor, indices []*Tensor, values *Tensor, accumulate bool, unsafety bool, del bool)(retVal *Tensor) { - - retVal, err := ts._IndexPutImplOut(out, indices, values, accumulate, unsafety, del) - if err != nil { log.Fatal(err) } - - return retVal -} - func(ts *Tensor) Must_Indices(del bool)(retVal *Tensor) { retVal, err := ts._Indices(del) @@ -1449,6 +1561,14 @@ func(ts *Tensor) Must_IsZerotensor(del bool)(retVal bool) { return retVal } +func(ts *Tensor) Must_LazyClone(del bool)(retVal *Tensor) { + + retVal, err := ts._LazyClone(del) + if err != nil { log.Fatal(err) } + + return retVal +} + func Must_LinalgDet(a *Tensor)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor) { retVal0, retVal1, retVal2, err := _LinalgDet(a) @@ -1481,6 +1601,14 @@ func Must_LinalgEighEigenvalues(eigenvalues *Tensor, eigenvectors *Tensor, a *Te return retVal0, retVal1 } +func(ts *Tensor) Must_LinalgEigvals(del bool)(retVal *Tensor) { + + retVal, err := ts._LinalgEigvals(del) + if err != nil { log.Fatal(err) } + + return retVal +} + func Must_LinalgSlogdet(a *Tensor)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor) { retVal0, retVal1, retVal2, retVal3, err := _LinalgSlogdet(a) @@ -1713,6 +1841,14 @@ func(ts *Tensor) Must_MaskedSoftmaxOut(out *Tensor, mask *Tensor, dim []int64, m return retVal } +func Must_MixedDtypesLinear(input *Tensor, weight *Tensor, scale *Tensor, bias *Tensor, activation string)(retVal *Tensor) { + + retVal, err := _MixedDtypesLinear(input, weight, scale, bias, activation) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) Must_MkldnnReshape(shape []int64, del bool)(retVal *Tensor) { retVal, err := ts._MkldnnReshape(shape, del) @@ -1881,6 +2017,14 @@ func(ts *Tensor) Must_NegViewCopyOut(out *Tensor, del bool)(retVal *Tensor) { return retVal } +func Must_NestedComputeContiguousStridesOffsets(nestedSize *Tensor)(retVal0 *Tensor, retVal1 *Tensor) { + + retVal0, retVal1, err := _NestedComputeContiguousStridesOffsets(nestedSize) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1 +} + func Must_NestedFromPadded(padded *Tensor, cpuNestedShapeExample *Tensor, fuseTransform0213 bool)(retVal *Tensor) { retVal, err := _NestedFromPadded(padded, cpuNestedShapeExample, fuseTransform0213) @@ -1913,6 +2057,86 @@ func Must_NestedFromPaddedOut(out *Tensor, padded *Tensor, cpuNestedShapeExample return retVal } +func Must_NestedFromPaddedTensor(padded *Tensor, offsets *Tensor, dummy *Tensor, raggedIdx int64, minSeqlen *Tensor, maxSeqlen *Tensor, sumS []int64)(retVal *Tensor) { + + retVal, err := _NestedFromPaddedTensor(padded, offsets, dummy, raggedIdx, minSeqlen, maxSeqlen, sumS) + if err != nil { log.Fatal(err) } + + return retVal +} + +func Must_NestedGetJaggedDummy(any *Tensor)(retVal *Tensor) { + + retVal, err := _NestedGetJaggedDummy(any) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_NestedGetLengths(del bool)(retVal *Tensor) { + + retVal, err := ts._NestedGetLengths(del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_NestedGetMaxSeqlen(del bool)(retVal *Tensor) { + + retVal, err := ts._NestedGetMaxSeqlen(del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_NestedGetMinSeqlen(del bool)(retVal *Tensor) { + + retVal, err := ts._NestedGetMinSeqlen(del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_NestedGetOffsets(del bool)(retVal *Tensor) { + + retVal, err := ts._NestedGetOffsets(del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_NestedGetRaggedIdx(del bool)(retVal int64) { + + retVal, err := ts._NestedGetRaggedIdx(del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_NestedGetValues(del bool)(retVal *Tensor) { + + retVal, err := ts._NestedGetValues(del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_NestedGetValuesCopy(del bool)(retVal *Tensor) { + + retVal, err := ts._NestedGetValuesCopy(del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_NestedGetValuesCopyOut(out *Tensor, del bool)(retVal *Tensor) { + + retVal, err := ts._NestedGetValuesCopyOut(out, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) Must_NestedSelectBackward(gradOutput *Tensor, dim int64, index int64, del bool)(retVal *Tensor) { retVal, err := ts._NestedSelectBackward(gradOutput, dim, index, del) @@ -1953,6 +2177,30 @@ func(ts *Tensor) Must_NestedViewFromBufferCopyOut(out *Tensor, nestedSize *Tenso return retVal } +func(ts *Tensor) Must_NestedViewFromJagged(offsets *Tensor, dummy *Tensor, lengths *Tensor, raggedIdx int64, minSeqlen *Tensor, maxSeqlen *Tensor, del bool)(retVal *Tensor) { + + retVal, err := ts._NestedViewFromJagged(offsets, dummy, lengths, raggedIdx, minSeqlen, maxSeqlen, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_NestedViewFromJaggedCopy(offsets *Tensor, dummy *Tensor, lengths *Tensor, raggedIdx int64, minSeqlen *Tensor, maxSeqlen *Tensor, del bool)(retVal *Tensor) { + + retVal, err := ts._NestedViewFromJaggedCopy(offsets, dummy, lengths, raggedIdx, minSeqlen, maxSeqlen, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_NestedViewFromJaggedCopyOut(out *Tensor, offsets *Tensor, dummy *Tensor, lengths *Tensor, raggedIdx int64, minSeqlen *Tensor, maxSeqlen *Tensor, del bool)(retVal *Tensor) { + + retVal, err := ts._NestedViewFromJaggedCopyOut(out, offsets, dummy, lengths, raggedIdx, minSeqlen, maxSeqlen, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) Must_NewZerosWithSameFeatureMeta(other *Tensor, selfNumBatchDims int64, del bool)(retVal *Tensor) { retVal, err := ts._NewZerosWithSameFeatureMeta(other, selfNumBatchDims, del) @@ -2177,6 +2425,14 @@ func Must_RowwisePrune(weight *Tensor, mask *Tensor, compressedIndicesDtype gotc return retVal0, retVal1 } +func(ts *Tensor) Must_SafeSoftmax(dim int64, dtype gotch.DType, del bool)(retVal *Tensor) { + + retVal, err := ts._SafeSoftmax(dim, dtype, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) Must_SampleDirichlet(del bool)(retVal *Tensor) { retVal, err := ts._SampleDirichlet(del) @@ -2201,14 +2457,30 @@ func Must_SaturateWeightToFp16(weight *Tensor)(retVal *Tensor) { return retVal } -func Must_ScaledDotProductAttentionMath(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, dropoutMask *Tensor, scale []float64)(retVal0 *Tensor, retVal1 *Tensor) { +func Must_ScaledDotProductAttentionMath(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, dropoutMask *Tensor, scale []float64, enableGqa bool)(retVal0 *Tensor, retVal1 *Tensor) { - retVal0, retVal1, err := _ScaledDotProductAttentionMath(query, key, value, attnMask, dropoutP, isCausal, dropoutMask, scale) + retVal0, retVal1, err := _ScaledDotProductAttentionMath(query, key, value, attnMask, dropoutP, isCausal, dropoutMask, scale, enableGqa) if err != nil { log.Fatal(err) } return retVal0, retVal1 } +func Must_ScaledDotProductAttentionMathForMps(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, dropoutMask *Tensor, scale []float64)(retVal0 *Tensor, retVal1 *Tensor) { + + retVal0, retVal1, err := _ScaledDotProductAttentionMathForMps(query, key, value, attnMask, dropoutP, isCausal, dropoutMask, scale) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1 +} + +func Must_ScaledDotProductCudnnAttentionBackward(gradOut *Tensor, query *Tensor, key *Tensor, value *Tensor, out *Tensor, logsumexp *Tensor, philoxSeed *Tensor, philoxOffset *Tensor, attnBias *Tensor, cumSeqQ *Tensor, cumSeqK *Tensor, maxQ int64, maxK int64, dropoutP float64, isCausal bool, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor) { + + retVal0, retVal1, retVal2, err := _ScaledDotProductCudnnAttentionBackward(gradOut, query, key, value, out, logsumexp, philoxSeed, philoxOffset, attnBias, cumSeqQ, cumSeqK, maxQ, maxK, dropoutP, isCausal, scale) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1, retVal2 +} + func Must_ScaledDotProductEfficientAttention(query *Tensor, key *Tensor, value *Tensor, attnBias *Tensor, computeLogSumexp bool, dropoutP float64, isCausal bool, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor) { retVal0, retVal1, retVal2, retVal3, err := _ScaledDotProductEfficientAttention(query, key, value, attnBias, computeLogSumexp, dropoutP, isCausal, scale) @@ -2225,20 +2497,68 @@ func Must_ScaledDotProductFlashAttentionBackward(gradOut *Tensor, query *Tensor, return retVal0, retVal1, retVal2 } -func(ts *Tensor) Must_ScaledMm(mat2 *Tensor, bias *Tensor, outDtype gotch.DType, scaleA *Tensor, scaleB *Tensor, scaleResult *Tensor, del bool)(retVal0 *Tensor, retVal1 *Tensor) { +func Must_ScaledDotProductFlashAttentionForCpu(query *Tensor, key *Tensor, value *Tensor, dropoutP float64, isCausal bool, attnMask *Tensor, scale []float64)(retVal0 *Tensor, retVal1 *Tensor) { - retVal0, retVal1, err := ts._ScaledMm(mat2, bias, outDtype, scaleA, scaleB, scaleResult, del) + retVal0, retVal1, err := _ScaledDotProductFlashAttentionForCpu(query, key, value, dropoutP, isCausal, attnMask, scale) if err != nil { log.Fatal(err) } return retVal0, retVal1 } -func(ts *Tensor) Must_ScaledMmOut(out *Tensor, outAmax *Tensor, mat2 *Tensor, bias *Tensor, outDtype gotch.DType, scaleA *Tensor, scaleB *Tensor, scaleResult *Tensor, del bool)(retVal0 *Tensor, retVal1 *Tensor) { +func Must_ScaledDotProductFlashAttentionForCpuBackward(gradOut *Tensor, query *Tensor, key *Tensor, value *Tensor, out *Tensor, logsumexp *Tensor, dropoutP float64, isCausal bool, attnMask *Tensor, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor) { - retVal0, retVal1, err := ts._ScaledMmOut(out, outAmax, mat2, bias, outDtype, scaleA, scaleB, scaleResult, del) + retVal0, retVal1, retVal2, err := _ScaledDotProductFlashAttentionForCpuBackward(gradOut, query, key, value, out, logsumexp, dropoutP, isCausal, attnMask, scale) if err != nil { log.Fatal(err) } - return retVal0, retVal1 + return retVal0, retVal1, retVal2 +} + +func(ts *Tensor) Must_ScaledGroupedMm(mat2 *Tensor, scaleA *Tensor, scaleB *Tensor, offs *Tensor, bias *Tensor, scaleResult *Tensor, outDtype gotch.DType, useFastAccum bool, del bool)(retVal *Tensor) { + + retVal, err := ts._ScaledGroupedMm(mat2, scaleA, scaleB, offs, bias, scaleResult, outDtype, useFastAccum, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_ScaledGroupedMmV2(mat2 *Tensor, scaleA []*Tensor, recipeA []int64, swizzleA []int64, scaleB []*Tensor, recipeB []int64, swizzleB []int64, offs *Tensor, bias *Tensor, outDtype gotch.DType, contractionDim []int64, useFastAccum bool, del bool)(retVal *Tensor) { + + retVal, err := ts._ScaledGroupedMmV2(mat2, scaleA, recipeA, swizzleA, scaleB, recipeB, swizzleB, offs, bias, outDtype, contractionDim, useFastAccum, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_ScaledMm(mat2 *Tensor, scaleA *Tensor, scaleB *Tensor, bias *Tensor, scaleResult *Tensor, outDtype gotch.DType, useFastAccum bool, del bool)(retVal *Tensor) { + + retVal, err := ts._ScaledMm(mat2, scaleA, scaleB, bias, scaleResult, outDtype, useFastAccum, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_ScaledMmOut(out *Tensor, mat2 *Tensor, scaleA *Tensor, scaleB *Tensor, bias *Tensor, scaleResult *Tensor, outDtype gotch.DType, useFastAccum bool, del bool)(retVal *Tensor) { + + retVal, err := ts._ScaledMmOut(out, mat2, scaleA, scaleB, bias, scaleResult, outDtype, useFastAccum, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_ScaledMmV2(mat2 *Tensor, scaleA []*Tensor, recipeA []int64, swizzleA []int64, scaleB []*Tensor, recipeB []int64, swizzleB []int64, bias *Tensor, outDtype gotch.DType, contractionDim []int64, useFastAccum bool, del bool)(retVal *Tensor) { + + retVal, err := ts._ScaledMmV2(mat2, scaleA, recipeA, swizzleA, scaleB, recipeB, swizzleB, bias, outDtype, contractionDim, useFastAccum, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_ScaledMmV2Out(out *Tensor, mat2 *Tensor, scaleA []*Tensor, recipeA []int64, swizzleA []int64, scaleB []*Tensor, recipeB []int64, swizzleB []int64, bias *Tensor, outDtype gotch.DType, contractionDim []int64, useFastAccum bool, del bool)(retVal *Tensor) { + + retVal, err := ts._ScaledMmV2Out(out, mat2, scaleA, recipeA, swizzleA, scaleB, recipeB, swizzleB, bias, outDtype, contractionDim, useFastAccum, del) + if err != nil { log.Fatal(err) } + + return retVal } func(ts *Tensor) Must_ScatterReduce(dim int64, index *Tensor, src *Tensor, reduce string, includeSelf bool, del bool)(retVal *Tensor) { @@ -2425,6 +2745,14 @@ func Must_SparseCompressedTensorUnsafe(compressedIndices *Tensor, plainIndices * return retVal } +func Must_SparseCompressedTensorWithDims(nnz int64, denseDim int64, size []int64, blocksize []int64, indexDtype gotch.DType, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor) { + + retVal, err := _SparseCompressedTensorWithDims(nnz, denseDim, size, blocksize, indexDtype, optionsKind, optionsDevice) + if err != nil { log.Fatal(err) } + + return retVal +} + func Must_SparseCooTensorUnsafe(indices *Tensor, values *Tensor, size []int64, optionsKind gotch.DType, optionsDevice gotch.Device, isCoalesced bool)(retVal *Tensor) { retVal, err := _SparseCooTensorUnsafe(indices, values, size, optionsKind, optionsDevice, isCoalesced) @@ -2593,14 +2921,46 @@ func(ts *Tensor) Must_SparseMmReduceImpl(other *Tensor, reduce string, del bool) return retVal0, retVal1 } -func Must_SparseSemiStructuredLinear(input *Tensor, weight *Tensor, meta *Tensor, bias *Tensor, activation string)(retVal *Tensor) { +func Must_SparseSemiStructuredApply(input *Tensor, threadMasks *Tensor)(retVal0 *Tensor, retVal1 *Tensor) { + + retVal0, retVal1, err := _SparseSemiStructuredApply(input, threadMasks) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1 +} + +func Must_SparseSemiStructuredApplyDense(input *Tensor, threadMasks *Tensor)(retVal *Tensor) { - retVal, err := _SparseSemiStructuredLinear(input, weight, meta, bias, activation) + retVal, err := _SparseSemiStructuredApplyDense(input, threadMasks) if err != nil { log.Fatal(err) } return retVal } +func Must_SparseSemiStructuredLinear(input *Tensor, weight *Tensor, meta *Tensor, bias *Tensor, activation string, outDtype gotch.DType)(retVal *Tensor) { + + retVal, err := _SparseSemiStructuredLinear(input, weight, meta, bias, activation, outDtype) + if err != nil { log.Fatal(err) } + + return retVal +} + +func Must_SparseSemiStructuredMm(mat1 *Tensor, mat1Meta *Tensor, mat2 *Tensor, outDtype gotch.DType)(retVal *Tensor) { + + retVal, err := _SparseSemiStructuredMm(mat1, mat1Meta, mat2, outDtype) + if err != nil { log.Fatal(err) } + + return retVal +} + +func Must_SparseSemiStructuredTile(input *Tensor, algorithm string, useCutlass bool)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, retVal4 *Tensor) { + + retVal0, retVal1, retVal2, retVal3, retVal4, err := _SparseSemiStructuredTile(input, algorithm, useCutlass) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1, retVal2, retVal3, retVal4 +} + func(ts *Tensor) Must_SparseSoftmax(dim int64, halfToFloat bool, del bool)(retVal *Tensor) { retVal, err := ts._SparseSoftmax(dim, halfToFloat, del) @@ -2729,6 +3089,14 @@ func Must_SpdiagsOut(out *Tensor, diagonals *Tensor, offsets *Tensor, shape []in return retVal } +func Must_Spsolve(a *Tensor, b *Tensor, left bool)(retVal *Tensor) { + + retVal, err := _Spsolve(a, b, left) + if err != nil { log.Fatal(err) } + + return retVal +} + func Must_Stack(tensors []*Tensor, dim int64)(retVal *Tensor) { retVal, err := _Stack(tensors, dim) @@ -2913,6 +3281,14 @@ func Must_TestOptionalIntlistOut(out *Tensor, values *Tensor, addends []int64)(r return retVal } +func(ts *Tensor) Must_TestParallelMaterialize(numParallel int64, skipFirst bool, del bool)(retVal *Tensor) { + + retVal, err := ts._TestParallelMaterialize(numParallel, skipFirst, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) Must_TestSerializationSubcmul(other *Tensor, del bool)(retVal *Tensor) { retVal, err := ts._TestSerializationSubcmul(other, del) @@ -3201,22 +3577,6 @@ func Must_UnpackDual(dual *Tensor, level int64)(retVal0 *Tensor, retVal1 *Tensor return retVal0, retVal1 } -func(ts *Tensor) Must_UnsafeIndex(indices []*Tensor, del bool)(retVal *Tensor) { - - retVal, err := ts._UnsafeIndex(indices, del) - if err != nil { log.Fatal(err) } - - return retVal -} - -func(ts *Tensor) Must_UnsafeIndexPut(indices []*Tensor, values *Tensor, accumulate bool, del bool)(retVal *Tensor) { - - retVal, err := ts._UnsafeIndexPut(indices, values, accumulate, del) - if err != nil { log.Fatal(err) } - - return retVal -} - func(ts *Tensor) Must_UnsafeView(size []int64, del bool)(retVal *Tensor) { retVal, err := ts._UnsafeView(size, del) @@ -3489,6 +3849,38 @@ func(ts *Tensor) Must_Version(del bool)(retVal int64) { return retVal } +func(ts *Tensor) Must_WeightInt4packMm(mat2 *Tensor, qGroupSize int64, qScaleAndZeros *Tensor, del bool)(retVal *Tensor) { + + retVal, err := ts._WeightInt4packMm(mat2, qGroupSize, qScaleAndZeros, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_WeightInt4packMmForCpu(mat2 *Tensor, qGroupSize int64, qScaleAndZeros *Tensor, del bool)(retVal *Tensor) { + + retVal, err := ts._WeightInt4packMmForCpu(mat2, qGroupSize, qScaleAndZeros, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_WeightInt4packMmWithScalesAndZeros(mat2 *Tensor, qGroupSize int64, qScale *Tensor, qZeros *Tensor, del bool)(retVal *Tensor) { + + retVal, err := ts._WeightInt4packMmWithScalesAndZeros(mat2, qGroupSize, qScale, qZeros, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) Must_WeightInt8packMm(mat2 *Tensor, scales *Tensor, del bool)(retVal *Tensor) { + + retVal, err := ts._WeightInt8packMm(mat2, scales, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func Must_WeightNorm(v *Tensor, g *Tensor, dim int64)(retVal *Tensor) { retVal, err := _WeightNorm(v, g, dim) @@ -3529,12 +3921,28 @@ func Must_WeightNormInterfaceBackwardOut(out0 *Tensor, out1 *Tensor, gradW *Tens return retVal0, retVal1 } -func Must_WeightNormInterfaceOut(out0 *Tensor, out1 *Tensor, v *Tensor, g *Tensor, dim int64)(retVal0 *Tensor, retVal1 *Tensor) { +func Must_WeightNormInterfaceOut(out0 *Tensor, out1 *Tensor, v *Tensor, g *Tensor, dim int64)(retVal0 *Tensor, retVal1 *Tensor) { + + retVal0, retVal1, err := _WeightNormInterfaceOut(out0, out1, v, g, dim) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1 +} + +func Must_WrappedLinearPrepack(weight *Tensor, weightScale *Tensor, weightZeroPoint *Tensor, bias *Tensor)(retVal *Tensor) { + + retVal, err := _WrappedLinearPrepack(weight, weightScale, weightZeroPoint, bias) + if err != nil { log.Fatal(err) } + + return retVal +} + +func Must_WrappedQuantizedLinearPrepacked(input *Tensor, inputScale *Tensor, inputZeroPoint *Tensor, packedWeight *Tensor, outputScale *Tensor, outputZeroPoint *Tensor, outChannel int64)(retVal *Tensor) { - retVal0, retVal1, err := _WeightNormInterfaceOut(out0, out1, v, g, dim) + retVal, err := _WrappedQuantizedLinearPrepacked(input, inputScale, inputZeroPoint, packedWeight, outputScale, outputZeroPoint, outChannel) if err != nil { log.Fatal(err) } - return retVal0, retVal1 + return retVal } func(ts *Tensor) MustAbs(del bool)(retVal *Tensor) { @@ -3641,6 +4049,14 @@ func(ts *Tensor) MustAdaptiveAvgPool1d(outputSize []int64, del bool)(retVal *Ten return retVal } +func(ts *Tensor) MustAdaptiveAvgPool1dOut(out *Tensor, outputSize []int64, del bool)(retVal *Tensor) { + + retVal, err := ts.AdaptiveAvgPool1dOut(out, outputSize, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustAdaptiveAvgPool2d(outputSize []int64, del bool)(retVal *Tensor) { retVal, err := ts.AdaptiveAvgPool2d(outputSize, del) @@ -3889,6 +4305,22 @@ func(ts *Tensor) MustAddmm_(mat1 *Tensor, mat2 *Tensor)() { return } +func(ts *Tensor) MustAddmmDtype(mat1 *Tensor, mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor) { + + retVal, err := ts.AddmmDtype(mat1, mat2, outDtype, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) MustAddmmDtypeOut(out *Tensor, mat1 *Tensor, mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor) { + + retVal, err := ts.AddmmDtypeOut(out, mat1, mat2, outDtype, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustAddmmOut(out *Tensor, mat1 *Tensor, mat2 *Tensor, del bool)(retVal *Tensor) { retVal, err := ts.AddmmOut(out, mat1, mat2, del) @@ -4033,6 +4465,22 @@ func(ts *Tensor) MustAllDim(dim int64, keepdim bool, del bool)(retVal *Tensor) { return retVal } +func(ts *Tensor) MustAllDims(dim []int64, keepdim bool, del bool)(retVal *Tensor) { + + retVal, err := ts.AllDims(dim, keepdim, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) MustAllDimsOut(out *Tensor, dim []int64, keepdim bool, del bool)(retVal *Tensor) { + + retVal, err := ts.AllDimsOut(out, dim, keepdim, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustAllOut(out *Tensor, dim int64, keepdim bool, del bool)(retVal *Tensor) { retVal, err := ts.AllOut(out, dim, keepdim, del) @@ -4153,6 +4601,22 @@ func(ts *Tensor) MustAnyDim(dim int64, keepdim bool, del bool)(retVal *Tensor) { return retVal } +func(ts *Tensor) MustAnyDims(dim []int64, keepdim bool, del bool)(retVal *Tensor) { + + retVal, err := ts.AnyDims(dim, keepdim, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) MustAnyDimsOut(out *Tensor, dim []int64, keepdim bool, del bool)(retVal *Tensor) { + + retVal, err := ts.AnyDimsOut(out, dim, keepdim, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustAnyOut(out *Tensor, dim int64, keepdim bool, del bool)(retVal *Tensor) { retVal, err := ts.AnyOut(out, dim, keepdim, del) @@ -4617,6 +5081,14 @@ func(ts *Tensor) MustAvgPool1d(kernelSize []int64, stride []int64, padding []int return retVal } +func(ts *Tensor) MustAvgPool1dOut(out *Tensor, kernelSize []int64, stride []int64, padding []int64, ceilMode bool, countIncludePad bool, del bool)(retVal *Tensor) { + + retVal, err := ts.AvgPool1dOut(out, kernelSize, stride, padding, ceilMode, countIncludePad, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustAvgPool2d(kernelSize []int64, stride []int64, padding []int64, ceilMode bool, countIncludePad bool, divisorOverride []int64, del bool)(retVal *Tensor) { retVal, err := ts.AvgPool2d(kernelSize, stride, padding, ceilMode, countIncludePad, divisorOverride, del) @@ -4697,6 +5169,22 @@ func(ts *Tensor) MustBaddbmm_(batch1 *Tensor, batch2 *Tensor)() { return } +func(ts *Tensor) MustBaddbmmDtype(batch1 *Tensor, batch2 *Tensor, outDtype gotch.DType, beta *Scalar, alpha *Scalar, del bool)(retVal *Tensor) { + + retVal, err := ts.BaddbmmDtype(batch1, batch2, outDtype, beta, alpha, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) MustBaddbmmDtypeOut(out *Tensor, batch1 *Tensor, batch2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor) { + + retVal, err := ts.BaddbmmDtypeOut(out, batch1, batch2, outDtype, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustBaddbmmOut(out *Tensor, batch1 *Tensor, batch2 *Tensor, del bool)(retVal *Tensor) { retVal, err := ts.BaddbmmOut(out, batch1, batch2, del) @@ -5385,6 +5873,22 @@ func(ts *Tensor) MustBmm(mat2 *Tensor, del bool)(retVal *Tensor) { return retVal } +func(ts *Tensor) MustBmmDtype(mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor) { + + retVal, err := ts.BmmDtype(mat2, outDtype, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) MustBmmDtypeOut(out *Tensor, mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor) { + + retVal, err := ts.BmmDtypeOut(out, mat2, outDtype, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustBmmOut(out *Tensor, mat2 *Tensor, del bool)(retVal *Tensor) { retVal, err := ts.BmmOut(out, mat2, del) @@ -5433,9 +5937,9 @@ func(ts *Tensor) MustBucketizeTensorOut(out *Tensor, boundaries *Tensor, outInt3 return retVal } -func MustCanCast(from gotch.DType, to gotch.DType)(retVal bool) { +func MustCanCast(from_ gotch.DType, to gotch.DType)(retVal bool) { - retVal, err := CanCast(from, to) + retVal, err := CanCast(from_, to) if err != nil { log.Fatal(err) } return retVal @@ -7785,6 +8289,22 @@ func MustFbgemmLinearFp16WeightFp32Activation(input *Tensor, packedWeight *Tenso return retVal } +func MustFbgemmLinearFp16WeightFp32ActivationOut(input *Tensor, packedWeight *Tensor, bias *Tensor, output *Tensor)(retVal *Tensor) { + + retVal, err := FbgemmLinearFp16WeightFp32ActivationOut(input, packedWeight, bias, output) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustFbgemmLinearFp16WeightOut(input *Tensor, packedWeight *Tensor, bias *Tensor, output *Tensor)(retVal *Tensor) { + + retVal, err := FbgemmLinearFp16WeightOut(input, packedWeight, bias, output) + if err != nil { log.Fatal(err) } + + return retVal +} + func MustFbgemmLinearInt8Weight(input *Tensor, weight *Tensor, packed *Tensor, colOffsets *Tensor, weightScale *Scalar, weightZeroPoint *Scalar, bias *Tensor)(retVal *Tensor) { retVal, err := FbgemmLinearInt8Weight(input, weight, packed, colOffsets, weightScale, weightZeroPoint, bias) @@ -8441,6 +8961,14 @@ func(ts *Tensor) MustFloorDivideScalar_(other *Scalar)() { return } +func(ts *Tensor) MustFloorDivideScalarOut(out *Tensor, other *Scalar, del bool)(retVal *Tensor) { + + retVal, err := ts.FloorDivideScalarOut(out, other, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustFloorOut(out *Tensor, del bool)(retVal *Tensor) { retVal, err := ts.FloorOut(out, del) @@ -9433,6 +9961,22 @@ func(ts *Tensor) MustHardtanhOut(out *Tensor, del bool)(retVal *Tensor) { return retVal } +func(ts *Tensor) MustHashTensor(dim []int64, keepdim bool, mode int64, del bool)(retVal *Tensor) { + + retVal, err := ts.HashTensor(dim, keepdim, mode, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) MustHashTensorOut(out *Tensor, dim []int64, keepdim bool, mode int64, del bool)(retVal *Tensor) { + + retVal, err := ts.HashTensorOut(out, dim, keepdim, mode, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustHeaviside(values *Tensor, del bool)(retVal *Tensor) { retVal, err := ts.Heaviside(values, del) @@ -9761,14 +10305,6 @@ func(ts *Tensor) MustIndexFillIntTensorOut(out *Tensor, dim int64, index *Tensor return retVal } -func(ts *Tensor) MustIndexPutOut(out *Tensor, indices []*Tensor, values *Tensor, accumulate bool, del bool)(retVal *Tensor) { - - retVal, err := ts.IndexPutOut(out, indices, values, accumulate, del) - if err != nil { log.Fatal(err) } - - return retVal -} - func(ts *Tensor) MustIndexReduce(dim int64, index *Tensor, source *Tensor, reduce string, includeSelf bool, del bool)(retVal *Tensor) { retVal, err := ts.IndexReduce(dim, index, source, reduce, includeSelf, del) @@ -9817,14 +10353,6 @@ func(ts *Tensor) MustIndexSelectOut(out *Tensor, dim int64, index *Tensor, del b return retVal } -func(ts *Tensor) MustIndexTensorOut(out *Tensor, indices []*Tensor, del bool)(retVal *Tensor) { - - retVal, err := ts.IndexTensorOut(out, indices, del) - if err != nil { log.Fatal(err) } - - return retVal -} - func(ts *Tensor) MustIndices(del bool)(retVal *Tensor) { retVal, err := ts.Indices(del) @@ -11385,6 +11913,54 @@ func MustLinspaceOut(out *Tensor, start *Scalar, end *Scalar, steps int64)(retVa return retVal } +func MustLinspaceScalarTensor(start *Scalar, end *Tensor, steps int64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor) { + + retVal, err := LinspaceScalarTensor(start, end, steps, optionsKind, optionsDevice) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustLinspaceScalarTensorOut(out *Tensor, start *Scalar, end *Tensor, steps int64)(retVal *Tensor) { + + retVal, err := LinspaceScalarTensorOut(out, start, end, steps) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustLinspaceTensorScalar(start *Tensor, end *Scalar, steps int64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor) { + + retVal, err := LinspaceTensorScalar(start, end, steps, optionsKind, optionsDevice) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustLinspaceTensorScalarOut(out *Tensor, start *Tensor, end *Scalar, steps int64)(retVal *Tensor) { + + retVal, err := LinspaceTensorScalarOut(out, start, end, steps) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustLinspaceTensorTensor(start *Tensor, end *Tensor, steps int64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor) { + + retVal, err := LinspaceTensorTensor(start, end, steps, optionsKind, optionsDevice) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustLinspaceTensorTensorOut(out *Tensor, start *Tensor, end *Tensor, steps int64)(retVal *Tensor) { + + retVal, err := LinspaceTensorTensorOut(out, start, end, steps) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustLog(del bool)(retVal *Tensor) { retVal, err := ts.Log(del) @@ -11761,6 +12337,54 @@ func MustLogspaceOut(out *Tensor, start *Scalar, end *Scalar, steps int64, base return retVal } +func MustLogspaceScalarTensor(start *Scalar, end *Tensor, steps int64, base float64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor) { + + retVal, err := LogspaceScalarTensor(start, end, steps, base, optionsKind, optionsDevice) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustLogspaceScalarTensorOut(out *Tensor, start *Scalar, end *Tensor, steps int64, base float64)(retVal *Tensor) { + + retVal, err := LogspaceScalarTensorOut(out, start, end, steps, base) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustLogspaceTensorScalar(start *Tensor, end *Scalar, steps int64, base float64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor) { + + retVal, err := LogspaceTensorScalar(start, end, steps, base, optionsKind, optionsDevice) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustLogspaceTensorScalarOut(out *Tensor, start *Tensor, end *Scalar, steps int64, base float64)(retVal *Tensor) { + + retVal, err := LogspaceTensorScalarOut(out, start, end, steps, base) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustLogspaceTensorTensor(start *Tensor, end *Tensor, steps int64, base float64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor) { + + retVal, err := LogspaceTensorTensor(start, end, steps, base, optionsKind, optionsDevice) + if err != nil { log.Fatal(err) } + + return retVal +} + +func MustLogspaceTensorTensorOut(out *Tensor, start *Tensor, end *Tensor, steps int64, base float64)(retVal *Tensor) { + + retVal, err := LogspaceTensorTensorOut(out, start, end, steps, base) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustLogsumexp(dim []int64, keepdim bool, del bool)(retVal *Tensor) { retVal, err := ts.Logsumexp(dim, keepdim, del) @@ -11953,6 +12577,14 @@ func(ts *Tensor) MustMaskedScatter_(mask *Tensor, source *Tensor)() { return } +func MustMaskedScatterBackward(gradOutput *Tensor, mask *Tensor, sizes []int64)(retVal *Tensor) { + + retVal, err := MaskedScatterBackward(gradOutput, mask, sizes) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustMaskedScatterOut(out *Tensor, mask *Tensor, source *Tensor, del bool)(retVal *Tensor) { retVal, err := ts.MaskedScatterOut(out, mask, source, del) @@ -12265,6 +12897,14 @@ func(ts *Tensor) MustMeanDim(dim []int64, keepdim bool, dtype gotch.DType, del b return retVal } +func(ts *Tensor) MustMeanDtypeOut(out *Tensor, dtype gotch.DType, del bool)(retVal *Tensor) { + + retVal, err := ts.MeanDtypeOut(out, dtype, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustMeanOut(out *Tensor, dim []int64, keepdim bool, dtype gotch.DType, del bool)(retVal *Tensor) { retVal, err := ts.MeanOut(out, dim, keepdim, dtype, del) @@ -12697,17 +13337,17 @@ func(ts *Tensor) MustMkldnnReorderConv2dWeightOut(out *Tensor, padding []int64, return retVal } -func(ts *Tensor) MustMkldnnReorderConv3dWeight(padding []int64, stride []int64, dilation []int64, groups int64, del bool)(retVal *Tensor) { +func(ts *Tensor) MustMkldnnReorderConv3dWeight(padding []int64, stride []int64, dilation []int64, groups int64, inputSize []int64, del bool)(retVal *Tensor) { - retVal, err := ts.MkldnnReorderConv3dWeight(padding, stride, dilation, groups, del) + retVal, err := ts.MkldnnReorderConv3dWeight(padding, stride, dilation, groups, inputSize, del) if err != nil { log.Fatal(err) } return retVal } -func(ts *Tensor) MustMkldnnReorderConv3dWeightOut(out *Tensor, padding []int64, stride []int64, dilation []int64, groups int64, del bool)(retVal *Tensor) { +func(ts *Tensor) MustMkldnnReorderConv3dWeightOut(out *Tensor, padding []int64, stride []int64, dilation []int64, groups int64, inputSize []int64, del bool)(retVal *Tensor) { - retVal, err := ts.MkldnnReorderConv3dWeightOut(out, padding, stride, dilation, groups, del) + retVal, err := ts.MkldnnReorderConv3dWeightOut(out, padding, stride, dilation, groups, inputSize, del) if err != nil { log.Fatal(err) } return retVal @@ -12753,6 +13393,22 @@ func(ts *Tensor) MustMm(mat2 *Tensor, del bool)(retVal *Tensor) { return retVal } +func(ts *Tensor) MustMmDtype(mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor) { + + retVal, err := ts.MmDtype(mat2, outDtype, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) MustMmDtypeOut(out *Tensor, mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor) { + + retVal, err := ts.MmDtypeOut(out, mat2, outDtype, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustMmOut(out *Tensor, mat2 *Tensor, del bool)(retVal *Tensor) { retVal, err := ts.MmOut(out, mat2, del) @@ -13921,9 +14577,9 @@ func(ts *Tensor) MustPad(pad []int64, mode string, value []float64, del bool)(re return retVal } -func MustPadSequence(sequences []*Tensor, batchFirst bool, paddingValue float64)(retVal *Tensor) { +func MustPadSequence(sequences []*Tensor, batchFirst bool, paddingValue float64, paddingSide string)(retVal *Tensor) { - retVal, err := PadSequence(sequences, batchFirst, paddingValue) + retVal, err := PadSequence(sequences, batchFirst, paddingValue, paddingSide) if err != nil { log.Fatal(err) } return retVal @@ -14561,6 +15217,22 @@ func(ts *Tensor) MustRandintLikeOut(out *Tensor, high int64, del bool)(retVal *T return retVal } +func(ts *Tensor) MustRandintLikeTensor(high *Tensor, del bool)(retVal *Tensor) { + + retVal, err := ts.RandintLikeTensor(high, del) + if err != nil { log.Fatal(err) } + + return retVal +} + +func(ts *Tensor) MustRandintLikeTensorOut(out *Tensor, high *Tensor, del bool)(retVal *Tensor) { + + retVal, err := ts.RandintLikeTensorOut(out, high, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func MustRandintLow(low int64, high int64, size []int64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor) { retVal, err := RandintLow(low, high, size, optionsKind, optionsDevice) @@ -15265,6 +15937,14 @@ func(ts *Tensor) MustRetainsGrad(del bool)(retVal bool) { return retVal } +func MustRmsNorm(input *Tensor, normalizedShape []int64, weight *Tensor, eps []float64)(retVal *Tensor) { + + retVal, err := RmsNorm(input, normalizedShape, weight, eps) + if err != nil { log.Fatal(err) } + + return retVal +} + func MustRnnRelu(input *Tensor, hx *Tensor, params []*Tensor, hasBiases bool, numLayers int64, dropout float64, train bool, bidirectional bool, batchFirst bool)(retVal0 *Tensor, retVal1 *Tensor) { retVal0, retVal1, err := RnnRelu(input, hx, params, hasBiases, numLayers, dropout, train, bidirectional, batchFirst) @@ -15481,6 +16161,14 @@ func(ts *Tensor) MustRreluWithNoiseBackwardOut(out *Tensor, gradOutput *Tensor, return retVal } +func(ts *Tensor) MustRreluWithNoiseFunctional(noise *Tensor, training bool, del bool)(retVal0 *Tensor, retVal1 *Tensor) { + + retVal0, retVal1, err := ts.RreluWithNoiseFunctional(noise, training, del) + if err != nil { log.Fatal(err) } + + return retVal0, retVal1 +} + func(ts *Tensor) MustRreluWithNoiseOut(out *Tensor, noise *Tensor, training bool, del bool)(retVal *Tensor) { retVal, err := ts.RreluWithNoiseOut(out, noise, training, del) @@ -15561,9 +16249,9 @@ func MustScalarTensorOut(out *Tensor, s *Scalar)(retVal *Tensor) { return retVal } -func MustScaledDotProductAttention(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, scale []float64)(retVal *Tensor) { +func MustScaledDotProductAttention(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, scale []float64, enableGqa bool)(retVal *Tensor) { - retVal, err := ScaledDotProductAttention(query, key, value, attnMask, dropoutP, isCausal, scale) + retVal, err := ScaledDotProductAttention(query, key, value, attnMask, dropoutP, isCausal, scale, enableGqa) if err != nil { log.Fatal(err) } return retVal @@ -16129,6 +16817,14 @@ func(ts *Tensor) MustSliceCopyTensorOut(out *Tensor, dim int64, start []int64, e return retVal } +func(ts *Tensor) MustSliceInverse(src *Tensor, dim int64, start []int64, end []int64, step int64, del bool)(retVal *Tensor) { + + retVal, err := ts.SliceInverse(src, dim, start, end, step, del) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustSliceScatter(src *Tensor, dim int64, start []int64, end []int64, step int64, del bool)(retVal *Tensor) { retVal, err := ts.SliceScatter(src, dim, start, end, step, del) @@ -18233,17 +18929,17 @@ func(ts *Tensor) MustStdOut(out *Tensor, dim []int64, unbiased bool, keepdim boo return retVal } -func(ts *Tensor) MustStft(nFft int64, hopLength []int64, winLength []int64, window *Tensor, normalized bool, onesided bool, returnComplex bool, del bool)(retVal *Tensor) { +func(ts *Tensor) MustStft(nFft int64, hopLength []int64, winLength []int64, window *Tensor, normalized bool, onesided bool, returnComplex bool, alignToWindow bool, del bool)(retVal *Tensor) { - retVal, err := ts.Stft(nFft, hopLength, winLength, window, normalized, onesided, returnComplex, del) + retVal, err := ts.Stft(nFft, hopLength, winLength, window, normalized, onesided, returnComplex, alignToWindow, del) if err != nil { log.Fatal(err) } return retVal } -func(ts *Tensor) MustStftCenter(nFft int64, hopLength []int64, winLength []int64, window *Tensor, center bool, padMode string, normalized bool, onesided bool, returnComplex bool, del bool)(retVal *Tensor) { +func(ts *Tensor) MustStftCenter(nFft int64, hopLength []int64, winLength []int64, window *Tensor, center bool, padMode string, normalized bool, onesided bool, returnComplex bool, alignToWindow bool, del bool)(retVal *Tensor) { - retVal, err := ts.StftCenter(nFft, hopLength, winLength, window, center, padMode, normalized, onesided, returnComplex, del) + retVal, err := ts.StftCenter(nFft, hopLength, winLength, window, center, padMode, normalized, onesided, returnComplex, alignToWindow, del) if err != nil { log.Fatal(err) } return retVal @@ -19281,6 +19977,14 @@ func MustUpsampleBilinear2dVec(input *Tensor, outputSize []int64, alignCorners b return retVal } +func MustUpsampleBilinear2dVecOut(out *Tensor, input *Tensor, outputSize []int64, alignCorners bool, scaleFactors []float64)(retVal *Tensor) { + + retVal, err := UpsampleBilinear2dVecOut(out, input, outputSize, alignCorners, scaleFactors) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustUpsampleLinear1d(outputSize []int64, alignCorners bool, scales []float64, del bool)(retVal *Tensor) { retVal, err := ts.UpsampleLinear1d(outputSize, alignCorners, scales, del) @@ -19401,6 +20105,14 @@ func MustUpsampleNearest2dVec(input *Tensor, outputSize []int64, scaleFactors [] return retVal } +func MustUpsampleNearest2dVecOut(out *Tensor, input *Tensor, outputSize []int64, scaleFactors []float64)(retVal *Tensor) { + + retVal, err := UpsampleNearest2dVecOut(out, input, outputSize, scaleFactors) + if err != nil { log.Fatal(err) } + + return retVal +} + func(ts *Tensor) MustUpsampleNearest3d(outputSize []int64, scalesD []float64, scalesH []float64, scalesW []float64, del bool)(retVal *Tensor) { retVal, err := ts.UpsampleNearest3d(outputSize, scalesD, scalesH, scalesW, del) diff --git a/ts/tensor-generated.go b/ts/tensor-generated.go index 0077169..0fa7a5c 100644 --- a/ts/tensor-generated.go +++ b/ts/tensor-generated.go @@ -866,6 +866,115 @@ ccpuEnabled := int32(0) return retVal, err } +// func.returns = `fixed ntensors`: +// --------------------------------- +func _BatchNormNoUpdate(input *Tensor, weight *Tensor, bias *Tensor, runningMean *Tensor, runningVar *Tensor, momentum float64, eps float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr3 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr2)) + unsafe.Sizeof(ctensorPtr0))) + + lib.Atg_BatchNormNoUpdate(ctensorPtr0, input.ctensor, weight.ctensor, bias.ctensor, runningMean.ctensor, runningVar.ctensor, momentum, eps) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_BatchNormNoUpdate() failed: %w", err) + return retVal0, retVal1, retVal2, retVal3, err + } + retVal0 = newTensor(*ctensorPtr0, "_BatchNormNoUpdate_0") + retVal1 = newTensor(*ctensorPtr1, "_BatchNormNoUpdate_1") + retVal2 = newTensor(*ctensorPtr2, "_BatchNormNoUpdate_2") + retVal3 = newTensor(*ctensorPtr3, "_BatchNormNoUpdate_3") + + return retVal0, retVal1, retVal2, retVal3, err +} + +// func.returns = `fixed ntensors`: +// --------------------------------- +func _BatchNormNoUpdateOut(out0 *Tensor, out1 *Tensor, out2 *Tensor, out3 *Tensor, input *Tensor, weight *Tensor, bias *Tensor, runningMean *Tensor, runningVar *Tensor, momentum float64, eps float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr3 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr2)) + unsafe.Sizeof(ctensorPtr0))) + + lib.Atg_BatchNormNoUpdateOut(ctensorPtr0, out0.ctensor, out1.ctensor, out2.ctensor, out3.ctensor, input.ctensor, weight.ctensor, bias.ctensor, runningMean.ctensor, runningVar.ctensor, momentum, eps) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_BatchNormNoUpdateOut() failed: %w", err) + return retVal0, retVal1, retVal2, retVal3, err + } + retVal0 = newTensor(*ctensorPtr0, "_BatchNormNoUpdateOut_0") + retVal1 = newTensor(*ctensorPtr1, "_BatchNormNoUpdateOut_1") + retVal2 = newTensor(*ctensorPtr2, "_BatchNormNoUpdateOut_2") + retVal3 = newTensor(*ctensorPtr3, "_BatchNormNoUpdateOut_3") + + return retVal0, retVal1, retVal2, retVal3, err +} + +// func.returns = `fixed ntensors`: +// --------------------------------- +func _BatchNormWithUpdate(input *Tensor, weight *Tensor, bias *Tensor, runningMean *Tensor, runningVar *Tensor, momentum float64, eps float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr3 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr2)) + unsafe.Sizeof(ctensorPtr0))) + + lib.Atg_BatchNormWithUpdate(ctensorPtr0, input.ctensor, weight.ctensor, bias.ctensor, runningMean.ctensor, runningVar.ctensor, momentum, eps) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_BatchNormWithUpdate() failed: %w", err) + return retVal0, retVal1, retVal2, retVal3, err + } + retVal0 = newTensor(*ctensorPtr0, "_BatchNormWithUpdate_0") + retVal1 = newTensor(*ctensorPtr1, "_BatchNormWithUpdate_1") + retVal2 = newTensor(*ctensorPtr2, "_BatchNormWithUpdate_2") + retVal3 = newTensor(*ctensorPtr3, "_BatchNormWithUpdate_3") + + return retVal0, retVal1, retVal2, retVal3, err +} + +// func.returns = `fixed ntensors`: +// --------------------------------- +func _BatchNormWithUpdateFunctional(input *Tensor, weight *Tensor, bias *Tensor, runningMean *Tensor, runningVar *Tensor, momentum float64, eps float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, retVal4 *Tensor, retVal5 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr3 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr2)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr4 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr3)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr5 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr4)) + unsafe.Sizeof(ctensorPtr0))) + + lib.Atg_BatchNormWithUpdateFunctional(ctensorPtr0, input.ctensor, weight.ctensor, bias.ctensor, runningMean.ctensor, runningVar.ctensor, momentum, eps) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_BatchNormWithUpdateFunctional() failed: %w", err) + return retVal0, retVal1, retVal2, retVal3, retVal4, retVal5, err + } + retVal0 = newTensor(*ctensorPtr0, "_BatchNormWithUpdateFunctional_0") + retVal1 = newTensor(*ctensorPtr1, "_BatchNormWithUpdateFunctional_1") + retVal2 = newTensor(*ctensorPtr2, "_BatchNormWithUpdateFunctional_2") + retVal3 = newTensor(*ctensorPtr3, "_BatchNormWithUpdateFunctional_3") + retVal4 = newTensor(*ctensorPtr4, "_BatchNormWithUpdateFunctional_4") + retVal5 = newTensor(*ctensorPtr5, "_BatchNormWithUpdateFunctional_5") + + return retVal0, retVal1, retVal2, retVal3, retVal4, retVal5, err +} + +// func.returns = `fixed ntensors`: +// --------------------------------- +func _BatchNormWithUpdateOut(out *Tensor, saveMean *Tensor, saveInvstd *Tensor, reserve *Tensor, input *Tensor, weight *Tensor, bias *Tensor, runningMean *Tensor, runningVar *Tensor, momentum float64, eps float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr3 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr2)) + unsafe.Sizeof(ctensorPtr0))) + + lib.Atg_BatchNormWithUpdateOut(ctensorPtr0, out.ctensor, saveMean.ctensor, saveInvstd.ctensor, reserve.ctensor, input.ctensor, weight.ctensor, bias.ctensor, runningMean.ctensor, runningVar.ctensor, momentum, eps) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_BatchNormWithUpdateOut() failed: %w", err) + return retVal0, retVal1, retVal2, retVal3, err + } + retVal0 = newTensor(*ctensorPtr0, "_BatchNormWithUpdateOut_0") + retVal1 = newTensor(*ctensorPtr1, "_BatchNormWithUpdateOut_1") + retVal2 = newTensor(*ctensorPtr2, "_BatchNormWithUpdateOut_2") + retVal3 = newTensor(*ctensorPtr3, "_BatchNormWithUpdateOut_3") + + return retVal0, retVal1, retVal2, retVal3, err +} + // func.returns = `fixed 1`: // -------------------------- @@ -1091,6 +1200,42 @@ func(ts *Tensor) _CholeskySolveHelperOut(out *Tensor, a *Tensor, upper bool, del // func.returns = `fixed 1`: // -------------------------- +func _ChunkCat(tensors []*Tensor, dim int64, numChunks int64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + var ctensors []lib.Ctensor + for _, t := range tensors {ctensors = append(ctensors, t.ctensor)} + lib.Atg_ChunkCat(ptr, ctensors, len(ctensors), dim, numChunks) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ChunkCat() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_ChunkCat") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func _ChunkCatOut(out *Tensor, tensors []*Tensor, dim int64, numChunks int64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + var ctensors []lib.Ctensor + for _, t := range tensors {ctensors = append(ctensors, t.ctensor)} + lib.Atg_ChunkCatOut(ptr, out.ctensor, ctensors, len(ctensors), dim, numChunks) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ChunkCatOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_ChunkCatOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) _Coalesce(del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -1418,6 +1563,40 @@ ctranspose := int32(0) // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) _ConvertWeightToInt4pack(innerKTiles int64, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_ConvertWeightToInt4pack(ptr, ts.ctensor, innerKTiles) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ConvertWeightToInt4pack() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_ConvertWeightToInt4pack") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _ConvertWeightToInt4packForCpu(innerKTiles int64, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_ConvertWeightToInt4packForCpu(ptr, ts.ctensor, innerKTiles) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ConvertWeightToInt4packForCpu() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_ConvertWeightToInt4packForCpu") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func _Convolution(input *Tensor, weight *Tensor, bias *Tensor, stride []int64, padding []int64, dilation []int64, transposed bool, outputPadding []int64, groups int64, benchmark bool, deterministic bool, cudnnEnabled bool, allowTf32 bool)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -1612,12 +1791,12 @@ func _CsltCompress(input *Tensor)(retVal *Tensor, err error) { // func.returns = `fixed 1`: // -------------------------- -func _CsltSparseMm(compressedA *Tensor, denseB *Tensor, bias *Tensor, transposeResult bool)(retVal *Tensor, err error) { +func _CsltSparseMm(compressedA *Tensor, denseB *Tensor, bias *Tensor, alpha *Tensor, outDtype gotch.DType, transposeResult bool, algId int64, splitK int64, splitKMode int64)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) ctransposeResult := int32(0) if transposeResult { ctransposeResult = int32(1) } - lib.Atg_CsltSparseMm(ptr, compressedA.ctensor, denseB.ctensor, bias.ctensor, ctransposeResult) + lib.Atg_CsltSparseMm(ptr, compressedA.ctensor, denseB.ctensor, bias.ctensor, alpha.ctensor, outDtype.CInt(), ctransposeResult, algId, splitK, splitKMode) if err = TorchErr(); err != nil { err = fmt.Errorf("_CsltSparseMm() failed: %w", err) return retVal, err @@ -1627,6 +1806,20 @@ func _CsltSparseMm(compressedA *Tensor, denseB *Tensor, bias *Tensor, transposeR return retVal, err } +// func.returns = `int64`: +// -------------------------- +func _CsltSparseMmSearch(compressedA *Tensor, denseB *Tensor, bias *Tensor, alpha *Tensor, outDtype gotch.DType, transposeResult bool)(retVal int64, err error) { + + ctransposeResult := int32(0) + if transposeResult { ctransposeResult = int32(1) } + retVal = lib.Atg_CsltSparseMmSearch(compressedA.ctensor, denseB.ctensor, bias.ctensor, alpha.ctensor, outDtype.CInt(), ctransposeResult) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_CsltSparseMmSearch() failed: %w", err) + return retVal, err + } + return retVal, err +} + // func.returns = `fixed ntensors`: // --------------------------------- func _CtcLoss(logProbs *Tensor, targets *Tensor, inputLengths []int64, targetLengths []int64, blank int64, zeroInfinity bool)(retVal0 *Tensor, retVal1 *Tensor, err error) { @@ -1765,6 +1958,33 @@ func _CtcLossTensorOut(out0 *Tensor, out1 *Tensor, logProbs *Tensor, targets *Te return retVal0, retVal1, err } +// func.returns = `fixed ntensors`: +// --------------------------------- +func _CudnnAttentionBackward(gradOut *Tensor, query *Tensor, key *Tensor, value *Tensor, out *Tensor, logsumexp *Tensor, philoxSeed *Tensor, philoxOffset *Tensor, attnBias *Tensor, cumSeqQ *Tensor, cumSeqK *Tensor, maxQ int64, maxK int64, dropoutP float64, isCausal bool, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) + + cisCausal := int32(0) + if isCausal { cisCausal = int32(1) } +var cscaleVal float64 = 0.0 + var cscaleNull int = 1 + if len(scale) > 0 { + cscaleVal = scale[0] + cscaleNull = 0 + } + lib.Atg_CudnnAttentionBackward(ctensorPtr0, gradOut.ctensor, query.ctensor, key.ctensor, value.ctensor, out.ctensor, logsumexp.ctensor, philoxSeed.ctensor, philoxOffset.ctensor, attnBias.ctensor, cumSeqQ.ctensor, cumSeqK.ctensor, maxQ, maxK, dropoutP, cisCausal, cscaleVal, cscaleNull) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_CudnnAttentionBackward() failed: %w", err) + return retVal0, retVal1, retVal2, err + } + retVal0 = newTensor(*ctensorPtr0, "_CudnnAttentionBackward_0") + retVal1 = newTensor(*ctensorPtr1, "_CudnnAttentionBackward_1") + retVal2 = newTensor(*ctensorPtr2, "_CudnnAttentionBackward_2") + + return retVal0, retVal1, retVal2, err +} + // func.returns = `fixed ntensors`: // --------------------------------- func _CudnnCtcLoss(logProbs *Tensor, targets *Tensor, inputLengths []int64, targetLengths []int64, blank int64, deterministic bool, zeroInfinity bool)(retVal0 *Tensor, retVal1 *Tensor, err error) { @@ -2063,9 +2283,41 @@ func _DirichletGradOut(out *Tensor, x *Tensor, alpha *Tensor, total *Tensor)(ret return retVal, err } +// func.returns = `fixed 1`: +// -------------------------- + +func _DynQuantMatmul4bit(inp *Tensor, packedWeights *Tensor, blockSize int64, inFeatures int64, outFeatures int64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_DynQuantMatmul4bit(ptr, inp.ctensor, packedWeights.ctensor, blockSize, inFeatures, outFeatures) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_DynQuantMatmul4bit() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_DynQuantMatmul4bit") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func _DynQuantPack4bitWeight(weights *Tensor, scalesZeros *Tensor, bias *Tensor, blockSize int64, inFeatures int64, outFeatures int64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_DynQuantPack4bitWeight(ptr, weights.ctensor, scalesZeros.ctensor, bias.ctensor, blockSize, inFeatures, outFeatures) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_DynQuantPack4bitWeight() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_DynQuantPack4bitWeight") + + return retVal, err +} + // func.returns = `fixed ntensors`: // --------------------------------- -func _EfficientAttentionBackward(gradOut_ *Tensor, query *Tensor, key *Tensor, value *Tensor, bias *Tensor, out *Tensor, cuSeqlensQ *Tensor, cuSeqlensK *Tensor, maxSeqlenK int64, maxSeqlenQ int64, logsumexp *Tensor, dropoutP float64, philoxSeed *Tensor, philoxOffset *Tensor, customMaskType int64, biasRequiresGrad bool, scale []float64, numSplitsKey []int64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, err error) { +func _EfficientAttentionBackward(gradOut_ *Tensor, query *Tensor, key *Tensor, value *Tensor, bias *Tensor, out *Tensor, cuSeqlensQ *Tensor, cuSeqlensK *Tensor, maxSeqlenQ int64, maxSeqlenK int64, logsumexp *Tensor, dropoutP float64, philoxSeed *Tensor, philoxOffset *Tensor, customMaskType int64, biasRequiresGrad bool, scale []float64, numSplitsKey []int64, windowSize []int64, sharedStorageDqdkdv bool)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, err error) { ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) @@ -2085,7 +2337,15 @@ var cnumSplitsKeyVal int64 = 0 cnumSplitsKeyVal = numSplitsKey[0] cnumSplitsKeyNull = 0 } - lib.Atg_EfficientAttentionBackward(ctensorPtr0, gradOut_.ctensor, query.ctensor, key.ctensor, value.ctensor, bias.ctensor, out.ctensor, cuSeqlensQ.ctensor, cuSeqlensK.ctensor, maxSeqlenK, maxSeqlenQ, logsumexp.ctensor, dropoutP, philoxSeed.ctensor, philoxOffset.ctensor, customMaskType, cbiasRequiresGrad, cscaleVal, cscaleNull, cnumSplitsKeyVal, cnumSplitsKeyNull) +var cwindowSizeVal int64 = 0 + var cwindowSizeNull int = 1 + if len(windowSize) > 0 { + cwindowSizeVal = windowSize[0] + cwindowSizeNull = 0 + } +csharedStorageDqdkdv := int32(0) + if sharedStorageDqdkdv { csharedStorageDqdkdv = int32(1) } + lib.Atg_EfficientAttentionBackward(ctensorPtr0, gradOut_.ctensor, query.ctensor, key.ctensor, value.ctensor, bias.ctensor, out.ctensor, cuSeqlensQ.ctensor, cuSeqlensK.ctensor, maxSeqlenQ, maxSeqlenK, logsumexp.ctensor, dropoutP, philoxSeed.ctensor, philoxOffset.ctensor, customMaskType, cbiasRequiresGrad, cscaleVal, cscaleNull, cnumSplitsKeyVal, cnumSplitsKeyNull, cwindowSizeVal, cwindowSizeNull, csharedStorageDqdkdv) if err = TorchErr(); err != nil { err = fmt.Errorf("_EfficientAttentionBackward() failed: %w", err) return retVal0, retVal1, retVal2, retVal3, err @@ -2724,7 +2984,7 @@ func(ts *Tensor) _FillMemEffDropoutMask_(dropoutP float64, seed int64, offset in // func.returns = `fixed ntensors`: // --------------------------------- -func _FlashAttentionBackward(gradOut *Tensor, query *Tensor, key *Tensor, value *Tensor, out *Tensor, logsumexp *Tensor, cumSeqQ *Tensor, cumSeqK *Tensor, maxQ int64, maxK int64, dropoutP float64, isCausal bool, philoxSeed *Tensor, philoxOffset *Tensor, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, err error) { +func _FlashAttentionBackward(gradOut *Tensor, query *Tensor, key *Tensor, value *Tensor, out *Tensor, logsumexp *Tensor, cumSeqQ *Tensor, cumSeqK *Tensor, maxQ int64, maxK int64, dropoutP float64, isCausal bool, rngState *Tensor, unused *Tensor, scale []float64, windowSizeLeft []int64, windowSizeRight []int64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, err error) { ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) @@ -2737,7 +2997,19 @@ var cscaleVal float64 = 0.0 cscaleVal = scale[0] cscaleNull = 0 } - lib.Atg_FlashAttentionBackward(ctensorPtr0, gradOut.ctensor, query.ctensor, key.ctensor, value.ctensor, out.ctensor, logsumexp.ctensor, cumSeqQ.ctensor, cumSeqK.ctensor, maxQ, maxK, dropoutP, cisCausal, philoxSeed.ctensor, philoxOffset.ctensor, cscaleVal, cscaleNull) +var cwindowSizeLeftVal int64 = 0 + var cwindowSizeLeftNull int = 1 + if len(windowSizeLeft) > 0 { + cwindowSizeLeftVal = windowSizeLeft[0] + cwindowSizeLeftNull = 0 + } +var cwindowSizeRightVal int64 = 0 + var cwindowSizeRightNull int = 1 + if len(windowSizeRight) > 0 { + cwindowSizeRightVal = windowSizeRight[0] + cwindowSizeRightNull = 0 + } + lib.Atg_FlashAttentionBackward(ctensorPtr0, gradOut.ctensor, query.ctensor, key.ctensor, value.ctensor, out.ctensor, logsumexp.ctensor, cumSeqQ.ctensor, cumSeqK.ctensor, maxQ, maxK, dropoutP, cisCausal, rngState.ctensor, unused.ctensor, cscaleVal, cscaleNull, cwindowSizeLeftVal, cwindowSizeLeftNull, cwindowSizeRightVal, cwindowSizeRightNull) if err = TorchErr(); err != nil { err = fmt.Errorf("_FlashAttentionBackward() failed: %w", err) return retVal0, retVal1, retVal2, err @@ -2815,6 +3087,22 @@ func(ts *Tensor) _FunctionalAssertAsync(assertMsg string, depToken *Tensor, del // func.returns = `fixed 1`: // -------------------------- +func _FunctionalAssertScalar(selfScalar *Scalar, assertMsg string, depToken *Tensor)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_FunctionalAssertScalar(ptr, selfScalar.cscalar, assertMsg, depToken.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_FunctionalAssertScalar() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_FunctionalAssertScalar") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func _FunctionalSymConstrainRange(size *Scalar, min []int64, max []int64, depToken *Tensor)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -2978,9 +3266,33 @@ csymmetricQuant := int32(0) return retVal0, retVal1, err } +// func.returns = `fixed ntensors`: +// --------------------------------- +func _FusedRmsNorm(input *Tensor, normalizedShape []int64, weight *Tensor, eps []float64)(retVal0 *Tensor, retVal1 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + + normalizedShapeLen := len(normalizedShape) +var cepsVal float64 = 0.0 + var cepsNull int = 1 + if len(eps) > 0 { + cepsVal = eps[0] + cepsNull = 0 + } + lib.Atg_FusedRmsNorm(ctensorPtr0, input.ctensor, normalizedShape, normalizedShapeLen, weight.ctensor, cepsVal, cepsNull) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_FusedRmsNorm() failed: %w", err) + return retVal0, retVal1, err + } + retVal0 = newTensor(*ctensorPtr0, "_FusedRmsNorm_0") + retVal1 = newTensor(*ctensorPtr1, "_FusedRmsNorm_1") + + return retVal0, retVal1, err +} + // func.returns = `int64`: // -------------------------- -func _FusedSdpChoice(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, scale []float64)(retVal int64, err error) { +func _FusedSdpChoice(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, scale []float64, enableGqa bool)(retVal int64, err error) { cisCausal := int32(0) if isCausal { cisCausal = int32(1) } @@ -2990,7 +3302,9 @@ var cscaleVal float64 = 0.0 cscaleVal = scale[0] cscaleNull = 0 } - retVal = lib.Atg_FusedSdpChoice(query.ctensor, key.ctensor, value.ctensor, attnMask.ctensor, dropoutP, cisCausal, cscaleVal, cscaleNull) +cenableGqa := int32(0) + if enableGqa { cenableGqa = int32(1) } + retVal = lib.Atg_FusedSdpChoice(query.ctensor, key.ctensor, value.ctensor, attnMask.ctensor, dropoutP, cisCausal, cscaleVal, cscaleNull, cenableGqa) if err = TorchErr(); err != nil { err = fmt.Errorf("_FusedSdpChoice() failed: %w", err) return retVal, err @@ -3121,6 +3435,23 @@ func _GridSampler2dCpuFallbackOut(out *Tensor, input *Tensor, grid *Tensor, inte return retVal, err } +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _GroupedMm(mat2 *Tensor, offs *Tensor, bias *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_GroupedMm(ptr, ts.ctensor, mat2.ctensor, offs.ctensor, bias.ctensor, outDtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_GroupedMm() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_GroupedMm") + + return retVal, err +} + // func.returns = `bool`: // -------------------------- func(ts *Tensor) _HasCompatibleShallowCopyType(from *Tensor, del bool)(retVal bool, err error) { @@ -3213,52 +3544,6 @@ cdensity := int32(0) // func.returns = `fixed 1`: // -------------------------- -func(ts *Tensor) _IndexPutImpl(indices []*Tensor, values *Tensor, accumulate bool, unsafety bool, del bool)(retVal *Tensor, err error) { - if del { defer ts.MustDrop() } - ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) - - var cindices []lib.Ctensor - for _, t := range indices {cindices = append(cindices, t.ctensor)} -caccumulate := int32(0) - if accumulate { caccumulate = int32(1) } -cunsafety := int32(0) - if unsafety { cunsafety = int32(1) } - lib.Atg_IndexPutImpl(ptr, ts.ctensor, cindices, len(cindices), values.ctensor, caccumulate, cunsafety) - if err = TorchErr(); err != nil { - err = fmt.Errorf("_IndexPutImpl() failed: %w", err) - return retVal, err - } - retVal = newTensor(*ptr, "_IndexPutImpl") - - return retVal, err -} - -// func.returns = `fixed 1`: -// -------------------------- - -func(ts *Tensor) _IndexPutImplOut(out *Tensor, indices []*Tensor, values *Tensor, accumulate bool, unsafety bool, del bool)(retVal *Tensor, err error) { - if del { defer ts.MustDrop() } - ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) - - var cindices []lib.Ctensor - for _, t := range indices {cindices = append(cindices, t.ctensor)} -caccumulate := int32(0) - if accumulate { caccumulate = int32(1) } -cunsafety := int32(0) - if unsafety { cunsafety = int32(1) } - lib.Atg_IndexPutImplOut(ptr, out.ctensor, ts.ctensor, cindices, len(cindices), values.ctensor, caccumulate, cunsafety) - if err = TorchErr(); err != nil { - err = fmt.Errorf("_IndexPutImplOut() failed: %w", err) - return retVal, err - } - retVal = newTensor(*ptr, "_IndexPutImplOut") - - return retVal, err -} - -// func.returns = `fixed 1`: -// -------------------------- - func(ts *Tensor) _Indices(del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -3388,6 +3673,23 @@ func(ts *Tensor) _IsZerotensor(del bool)(retVal bool, err error) { return retVal, err } +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _LazyClone(del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_LazyClone(ptr, ts.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_LazyClone() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_LazyClone") + + return retVal, err +} + // func.returns = `fixed ntensors`: // --------------------------------- func _LinalgDet(a *Tensor)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, err error) { @@ -3464,6 +3766,23 @@ func _LinalgEighEigenvalues(eigenvalues *Tensor, eigenvectors *Tensor, a *Tensor return retVal0, retVal1, err } +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _LinalgEigvals(del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_LinalgEigvals(ptr, ts.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_LinalgEigvals() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_LinalgEigvals") + + return retVal, err +} + // func.returns = `fixed ntensors`: // --------------------------------- func _LinalgSlogdet(a *Tensor)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, err error) { @@ -4075,6 +4394,22 @@ var cmaskTypeVal int64 = 0 // func.returns = `fixed 1`: // -------------------------- +func _MixedDtypesLinear(input *Tensor, weight *Tensor, scale *Tensor, bias *Tensor, activation string)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_MixedDtypesLinear(ptr, input.ctensor, weight.ctensor, scale.ctensor, bias.ctensor, activation) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_MixedDtypesLinear() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_MixedDtypesLinear") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) _MkldnnReshape(shape []int64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -4492,6 +4827,23 @@ func(ts *Tensor) _NegViewCopyOut(out *Tensor, del bool)(retVal *Tensor, err erro return retVal, err } +// func.returns = `fixed ntensors`: +// --------------------------------- +func _NestedComputeContiguousStridesOffsets(nestedSize *Tensor)(retVal0 *Tensor, retVal1 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + + lib.Atg_NestedComputeContiguousStridesOffsets(ctensorPtr0, nestedSize.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedComputeContiguousStridesOffsets() failed: %w", err) + return retVal0, retVal1, err + } + retVal0 = newTensor(*ctensorPtr0, "_NestedComputeContiguousStridesOffsets_0") + retVal1 = newTensor(*ctensorPtr1, "_NestedComputeContiguousStridesOffsets_1") + + return retVal0, retVal1, err +} + // func.returns = `fixed 1`: // -------------------------- @@ -4563,6 +4915,176 @@ func _NestedFromPaddedOut(out *Tensor, padded *Tensor, cpuNestedShapeExample *Te // func.returns = `fixed 1`: // -------------------------- +func _NestedFromPaddedTensor(padded *Tensor, offsets *Tensor, dummy *Tensor, raggedIdx int64, minSeqlen *Tensor, maxSeqlen *Tensor, sumS []int64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + var csumSVal int64 = 0 + var csumSNull int = 1 + if len(sumS) > 0 { + csumSVal = sumS[0] + csumSNull = 0 + } + lib.Atg_NestedFromPaddedTensor(ptr, padded.ctensor, offsets.ctensor, dummy.ctensor, raggedIdx, minSeqlen.ctensor, maxSeqlen.ctensor, csumSVal, csumSNull) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedFromPaddedTensor() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedFromPaddedTensor") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func _NestedGetJaggedDummy(any *Tensor)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedGetJaggedDummy(ptr, any.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedGetJaggedDummy() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedGetJaggedDummy") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _NestedGetLengths(del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedGetLengths(ptr, ts.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedGetLengths() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedGetLengths") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _NestedGetMaxSeqlen(del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedGetMaxSeqlen(ptr, ts.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedGetMaxSeqlen() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedGetMaxSeqlen") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _NestedGetMinSeqlen(del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedGetMinSeqlen(ptr, ts.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedGetMinSeqlen() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedGetMinSeqlen") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _NestedGetOffsets(del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedGetOffsets(ptr, ts.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedGetOffsets() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedGetOffsets") + + return retVal, err +} + +// func.returns = `int64`: +// -------------------------- +func(ts *Tensor) _NestedGetRaggedIdx(del bool)(retVal int64, err error) { + if del { defer ts.MustDrop() } + + retVal = lib.Atg_NestedGetRaggedIdx(ts.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedGetRaggedIdx() failed: %w", err) + return retVal, err + } + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _NestedGetValues(del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedGetValues(ptr, ts.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedGetValues() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedGetValues") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _NestedGetValuesCopy(del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedGetValuesCopy(ptr, ts.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedGetValuesCopy() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedGetValuesCopy") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _NestedGetValuesCopyOut(out *Tensor, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedGetValuesCopyOut(ptr, out.ctensor, ts.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedGetValuesCopyOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedGetValuesCopyOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) _NestedSelectBackward(gradOutput *Tensor, dim int64, index int64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -4651,6 +5173,57 @@ func(ts *Tensor) _NestedViewFromBufferCopyOut(out *Tensor, nestedSize *Tensor, n // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) _NestedViewFromJagged(offsets *Tensor, dummy *Tensor, lengths *Tensor, raggedIdx int64, minSeqlen *Tensor, maxSeqlen *Tensor, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedViewFromJagged(ptr, ts.ctensor, offsets.ctensor, dummy.ctensor, lengths.ctensor, raggedIdx, minSeqlen.ctensor, maxSeqlen.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedViewFromJagged() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedViewFromJagged") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _NestedViewFromJaggedCopy(offsets *Tensor, dummy *Tensor, lengths *Tensor, raggedIdx int64, minSeqlen *Tensor, maxSeqlen *Tensor, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedViewFromJaggedCopy(ptr, ts.ctensor, offsets.ctensor, dummy.ctensor, lengths.ctensor, raggedIdx, minSeqlen.ctensor, maxSeqlen.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedViewFromJaggedCopy() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedViewFromJaggedCopy") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _NestedViewFromJaggedCopyOut(out *Tensor, offsets *Tensor, dummy *Tensor, lengths *Tensor, raggedIdx int64, minSeqlen *Tensor, maxSeqlen *Tensor, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_NestedViewFromJaggedCopyOut(ptr, out.ctensor, ts.ctensor, offsets.ctensor, dummy.ctensor, lengths.ctensor, raggedIdx, minSeqlen.ctensor, maxSeqlen.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_NestedViewFromJaggedCopyOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_NestedViewFromJaggedCopyOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) _NewZerosWithSameFeatureMeta(other *Tensor, selfNumBatchDims int64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -5146,6 +5719,23 @@ func _RowwisePrune(weight *Tensor, mask *Tensor, compressedIndicesDtype gotch.DT // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) _SafeSoftmax(dim int64, dtype gotch.DType, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_SafeSoftmax(ptr, ts.ctensor, dim, dtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_SafeSoftmax() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_SafeSoftmax") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) _SampleDirichlet(del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -5195,7 +5785,7 @@ func _SaturateWeightToFp16(weight *Tensor)(retVal *Tensor, err error) { // func.returns = `fixed ntensors`: // --------------------------------- -func _ScaledDotProductAttentionMath(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, dropoutMask *Tensor, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, err error) { +func _ScaledDotProductAttentionMath(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, dropoutMask *Tensor, scale []float64, enableGqa bool)(retVal0 *Tensor, retVal1 *Tensor, err error) { ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) @@ -5207,7 +5797,9 @@ var cscaleVal float64 = 0.0 cscaleVal = scale[0] cscaleNull = 0 } - lib.Atg_ScaledDotProductAttentionMath(ctensorPtr0, query.ctensor, key.ctensor, value.ctensor, attnMask.ctensor, dropoutP, cisCausal, dropoutMask.ctensor, cscaleVal, cscaleNull) +cenableGqa := int32(0) + if enableGqa { cenableGqa = int32(1) } + lib.Atg_ScaledDotProductAttentionMath(ctensorPtr0, query.ctensor, key.ctensor, value.ctensor, attnMask.ctensor, dropoutP, cisCausal, dropoutMask.ctensor, cscaleVal, cscaleNull, cenableGqa) if err = TorchErr(); err != nil { err = fmt.Errorf("_ScaledDotProductAttentionMath() failed: %w", err) return retVal0, retVal1, err @@ -5218,6 +5810,58 @@ var cscaleVal float64 = 0.0 return retVal0, retVal1, err } +// func.returns = `fixed ntensors`: +// --------------------------------- +func _ScaledDotProductAttentionMathForMps(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, dropoutMask *Tensor, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + + cisCausal := int32(0) + if isCausal { cisCausal = int32(1) } +var cscaleVal float64 = 0.0 + var cscaleNull int = 1 + if len(scale) > 0 { + cscaleVal = scale[0] + cscaleNull = 0 + } + lib.Atg_ScaledDotProductAttentionMathForMps(ctensorPtr0, query.ctensor, key.ctensor, value.ctensor, attnMask.ctensor, dropoutP, cisCausal, dropoutMask.ctensor, cscaleVal, cscaleNull) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ScaledDotProductAttentionMathForMps() failed: %w", err) + return retVal0, retVal1, err + } + retVal0 = newTensor(*ctensorPtr0, "_ScaledDotProductAttentionMathForMps_0") + retVal1 = newTensor(*ctensorPtr1, "_ScaledDotProductAttentionMathForMps_1") + + return retVal0, retVal1, err +} + +// func.returns = `fixed ntensors`: +// --------------------------------- +func _ScaledDotProductCudnnAttentionBackward(gradOut *Tensor, query *Tensor, key *Tensor, value *Tensor, out *Tensor, logsumexp *Tensor, philoxSeed *Tensor, philoxOffset *Tensor, attnBias *Tensor, cumSeqQ *Tensor, cumSeqK *Tensor, maxQ int64, maxK int64, dropoutP float64, isCausal bool, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) + + cisCausal := int32(0) + if isCausal { cisCausal = int32(1) } +var cscaleVal float64 = 0.0 + var cscaleNull int = 1 + if len(scale) > 0 { + cscaleVal = scale[0] + cscaleNull = 0 + } + lib.Atg_ScaledDotProductCudnnAttentionBackward(ctensorPtr0, gradOut.ctensor, query.ctensor, key.ctensor, value.ctensor, out.ctensor, logsumexp.ctensor, philoxSeed.ctensor, philoxOffset.ctensor, attnBias.ctensor, cumSeqQ.ctensor, cumSeqK.ctensor, maxQ, maxK, dropoutP, cisCausal, cscaleVal, cscaleNull) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ScaledDotProductCudnnAttentionBackward() failed: %w", err) + return retVal0, retVal1, retVal2, err + } + retVal0 = newTensor(*ctensorPtr0, "_ScaledDotProductCudnnAttentionBackward_0") + retVal1 = newTensor(*ctensorPtr1, "_ScaledDotProductCudnnAttentionBackward_1") + retVal2 = newTensor(*ctensorPtr2, "_ScaledDotProductCudnnAttentionBackward_2") + + return retVal0, retVal1, retVal2, err +} + // func.returns = `fixed ntensors`: // --------------------------------- func _ScaledDotProductEfficientAttention(query *Tensor, key *Tensor, value *Tensor, attnBias *Tensor, computeLogSumexp bool, dropoutP float64, isCausal bool, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, err error) { @@ -5278,38 +5922,195 @@ var cscaleVal float64 = 0.0 // func.returns = `fixed ntensors`: // --------------------------------- -func(ts *Tensor) _ScaledMm(mat2 *Tensor, bias *Tensor, outDtype gotch.DType, scaleA *Tensor, scaleB *Tensor, scaleResult *Tensor, del bool)(retVal0 *Tensor, retVal1 *Tensor, err error) { - if del { defer ts.MustDrop() } +func _ScaledDotProductFlashAttentionForCpu(query *Tensor, key *Tensor, value *Tensor, dropoutP float64, isCausal bool, attnMask *Tensor, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, err error) { ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) - lib.Atg_ScaledMm(ctensorPtr0, ts.ctensor, mat2.ctensor, bias.ctensor, outDtype.CInt(), scaleA.ctensor, scaleB.ctensor, scaleResult.ctensor) + cisCausal := int32(0) + if isCausal { cisCausal = int32(1) } +var cscaleVal float64 = 0.0 + var cscaleNull int = 1 + if len(scale) > 0 { + cscaleVal = scale[0] + cscaleNull = 0 + } + lib.Atg_ScaledDotProductFlashAttentionForCpu(ctensorPtr0, query.ctensor, key.ctensor, value.ctensor, dropoutP, cisCausal, attnMask.ctensor, cscaleVal, cscaleNull) if err = TorchErr(); err != nil { - err = fmt.Errorf("_ScaledMm() failed: %w", err) + err = fmt.Errorf("_ScaledDotProductFlashAttentionForCpu() failed: %w", err) return retVal0, retVal1, err } - retVal0 = newTensor(*ctensorPtr0, "_ScaledMm_0") - retVal1 = newTensor(*ctensorPtr1, "_ScaledMm_1") + retVal0 = newTensor(*ctensorPtr0, "_ScaledDotProductFlashAttentionForCpu_0") + retVal1 = newTensor(*ctensorPtr1, "_ScaledDotProductFlashAttentionForCpu_1") return retVal0, retVal1, err } // func.returns = `fixed ntensors`: // --------------------------------- -func(ts *Tensor) _ScaledMmOut(out *Tensor, outAmax *Tensor, mat2 *Tensor, bias *Tensor, outDtype gotch.DType, scaleA *Tensor, scaleB *Tensor, scaleResult *Tensor, del bool)(retVal0 *Tensor, retVal1 *Tensor, err error) { - if del { defer ts.MustDrop() } +func _ScaledDotProductFlashAttentionForCpuBackward(gradOut *Tensor, query *Tensor, key *Tensor, value *Tensor, out *Tensor, logsumexp *Tensor, dropoutP float64, isCausal bool, attnMask *Tensor, scale []float64)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, err error) { ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) - lib.Atg_ScaledMmOut(ctensorPtr0, out.ctensor, outAmax.ctensor, ts.ctensor, mat2.ctensor, bias.ctensor, outDtype.CInt(), scaleA.ctensor, scaleB.ctensor, scaleResult.ctensor) + cisCausal := int32(0) + if isCausal { cisCausal = int32(1) } +var cscaleVal float64 = 0.0 + var cscaleNull int = 1 + if len(scale) > 0 { + cscaleVal = scale[0] + cscaleNull = 0 + } + lib.Atg_ScaledDotProductFlashAttentionForCpuBackward(ctensorPtr0, gradOut.ctensor, query.ctensor, key.ctensor, value.ctensor, out.ctensor, logsumexp.ctensor, dropoutP, cisCausal, attnMask.ctensor, cscaleVal, cscaleNull) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ScaledDotProductFlashAttentionForCpuBackward() failed: %w", err) + return retVal0, retVal1, retVal2, err + } + retVal0 = newTensor(*ctensorPtr0, "_ScaledDotProductFlashAttentionForCpuBackward_0") + retVal1 = newTensor(*ctensorPtr1, "_ScaledDotProductFlashAttentionForCpuBackward_1") + retVal2 = newTensor(*ctensorPtr2, "_ScaledDotProductFlashAttentionForCpuBackward_2") + + return retVal0, retVal1, retVal2, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _ScaledGroupedMm(mat2 *Tensor, scaleA *Tensor, scaleB *Tensor, offs *Tensor, bias *Tensor, scaleResult *Tensor, outDtype gotch.DType, useFastAccum bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + cuseFastAccum := int32(0) + if useFastAccum { cuseFastAccum = int32(1) } + lib.Atg_ScaledGroupedMm(ptr, ts.ctensor, mat2.ctensor, scaleA.ctensor, scaleB.ctensor, offs.ctensor, bias.ctensor, scaleResult.ctensor, outDtype.CInt(), cuseFastAccum) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ScaledGroupedMm() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_ScaledGroupedMm") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _ScaledGroupedMmV2(mat2 *Tensor, scaleA []*Tensor, recipeA []int64, swizzleA []int64, scaleB []*Tensor, recipeB []int64, swizzleB []int64, offs *Tensor, bias *Tensor, outDtype gotch.DType, contractionDim []int64, useFastAccum bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + var cscaleA []lib.Ctensor + for _, t := range scaleA {cscaleA = append(cscaleA, t.ctensor)} +recipeALen := len(recipeA) +swizzleALen := len(swizzleA) +var cscaleB []lib.Ctensor + for _, t := range scaleB {cscaleB = append(cscaleB, t.ctensor)} +recipeBLen := len(recipeB) +swizzleBLen := len(swizzleB) +contractionDimLen := len(contractionDim) +cuseFastAccum := int32(0) + if useFastAccum { cuseFastAccum = int32(1) } + lib.Atg_ScaledGroupedMmV2(ptr, ts.ctensor, mat2.ctensor, cscaleA, len(cscaleA), recipeA, recipeALen, swizzleA, swizzleALen, cscaleB, len(cscaleB), recipeB, recipeBLen, swizzleB, swizzleBLen, offs.ctensor, bias.ctensor, outDtype.CInt(), contractionDim, contractionDimLen, cuseFastAccum) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ScaledGroupedMmV2() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_ScaledGroupedMmV2") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _ScaledMm(mat2 *Tensor, scaleA *Tensor, scaleB *Tensor, bias *Tensor, scaleResult *Tensor, outDtype gotch.DType, useFastAccum bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + cuseFastAccum := int32(0) + if useFastAccum { cuseFastAccum = int32(1) } + lib.Atg_ScaledMm(ptr, ts.ctensor, mat2.ctensor, scaleA.ctensor, scaleB.ctensor, bias.ctensor, scaleResult.ctensor, outDtype.CInt(), cuseFastAccum) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ScaledMm() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_ScaledMm") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _ScaledMmOut(out *Tensor, mat2 *Tensor, scaleA *Tensor, scaleB *Tensor, bias *Tensor, scaleResult *Tensor, outDtype gotch.DType, useFastAccum bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + cuseFastAccum := int32(0) + if useFastAccum { cuseFastAccum = int32(1) } + lib.Atg_ScaledMmOut(ptr, out.ctensor, ts.ctensor, mat2.ctensor, scaleA.ctensor, scaleB.ctensor, bias.ctensor, scaleResult.ctensor, outDtype.CInt(), cuseFastAccum) if err = TorchErr(); err != nil { err = fmt.Errorf("_ScaledMmOut() failed: %w", err) - return retVal0, retVal1, err + return retVal, err } - retVal0 = newTensor(*ctensorPtr0, "_ScaledMmOut_0") - retVal1 = newTensor(*ctensorPtr1, "_ScaledMmOut_1") + retVal = newTensor(*ptr, "_ScaledMmOut") - return retVal0, retVal1, err + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _ScaledMmV2(mat2 *Tensor, scaleA []*Tensor, recipeA []int64, swizzleA []int64, scaleB []*Tensor, recipeB []int64, swizzleB []int64, bias *Tensor, outDtype gotch.DType, contractionDim []int64, useFastAccum bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + var cscaleA []lib.Ctensor + for _, t := range scaleA {cscaleA = append(cscaleA, t.ctensor)} +recipeALen := len(recipeA) +swizzleALen := len(swizzleA) +var cscaleB []lib.Ctensor + for _, t := range scaleB {cscaleB = append(cscaleB, t.ctensor)} +recipeBLen := len(recipeB) +swizzleBLen := len(swizzleB) +contractionDimLen := len(contractionDim) +cuseFastAccum := int32(0) + if useFastAccum { cuseFastAccum = int32(1) } + lib.Atg_ScaledMmV2(ptr, ts.ctensor, mat2.ctensor, cscaleA, len(cscaleA), recipeA, recipeALen, swizzleA, swizzleALen, cscaleB, len(cscaleB), recipeB, recipeBLen, swizzleB, swizzleBLen, bias.ctensor, outDtype.CInt(), contractionDim, contractionDimLen, cuseFastAccum) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ScaledMmV2() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_ScaledMmV2") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _ScaledMmV2Out(out *Tensor, mat2 *Tensor, scaleA []*Tensor, recipeA []int64, swizzleA []int64, scaleB []*Tensor, recipeB []int64, swizzleB []int64, bias *Tensor, outDtype gotch.DType, contractionDim []int64, useFastAccum bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + var cscaleA []lib.Ctensor + for _, t := range scaleA {cscaleA = append(cscaleA, t.ctensor)} +recipeALen := len(recipeA) +swizzleALen := len(swizzleA) +var cscaleB []lib.Ctensor + for _, t := range scaleB {cscaleB = append(cscaleB, t.ctensor)} +recipeBLen := len(recipeB) +swizzleBLen := len(swizzleB) +contractionDimLen := len(contractionDim) +cuseFastAccum := int32(0) + if useFastAccum { cuseFastAccum = int32(1) } + lib.Atg_ScaledMmV2Out(ptr, out.ctensor, ts.ctensor, mat2.ctensor, cscaleA, len(cscaleA), recipeA, recipeALen, swizzleA, swizzleALen, cscaleB, len(cscaleB), recipeB, recipeBLen, swizzleB, swizzleBLen, bias.ctensor, outDtype.CInt(), contractionDim, contractionDimLen, cuseFastAccum) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_ScaledMmV2Out() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_ScaledMmV2Out") + + return retVal, err } // func.returns = `fixed 1`: @@ -5717,6 +6518,24 @@ func _SparseCompressedTensorUnsafe(compressedIndices *Tensor, plainIndices *Tens // func.returns = `fixed 1`: // -------------------------- +func _SparseCompressedTensorWithDims(nnz int64, denseDim int64, size []int64, blocksize []int64, indexDtype gotch.DType, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + sizeLen := len(size) +blocksizeLen := len(blocksize) + lib.Atg_SparseCompressedTensorWithDims(ptr, nnz, denseDim, size, sizeLen, blocksize, blocksizeLen, indexDtype.CInt(), optionsKind.CInt(), optionsDevice.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_SparseCompressedTensorWithDims() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_SparseCompressedTensorWithDims") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func _SparseCooTensorUnsafe(indices *Tensor, values *Tensor, size []int64, optionsKind gotch.DType, optionsDevice gotch.Device, isCoalesced bool)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -6096,13 +6915,46 @@ func(ts *Tensor) _SparseMmReduceImpl(other *Tensor, reduce string, del bool)(ret return retVal0, retVal1, err } +// func.returns = `fixed ntensors`: +// --------------------------------- +func _SparseSemiStructuredApply(input *Tensor, threadMasks *Tensor)(retVal0 *Tensor, retVal1 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + + lib.Atg_SparseSemiStructuredApply(ctensorPtr0, input.ctensor, threadMasks.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_SparseSemiStructuredApply() failed: %w", err) + return retVal0, retVal1, err + } + retVal0 = newTensor(*ctensorPtr0, "_SparseSemiStructuredApply_0") + retVal1 = newTensor(*ctensorPtr1, "_SparseSemiStructuredApply_1") + + return retVal0, retVal1, err +} + // func.returns = `fixed 1`: // -------------------------- -func _SparseSemiStructuredLinear(input *Tensor, weight *Tensor, meta *Tensor, bias *Tensor, activation string)(retVal *Tensor, err error) { +func _SparseSemiStructuredApplyDense(input *Tensor, threadMasks *Tensor)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) - lib.Atg_SparseSemiStructuredLinear(ptr, input.ctensor, weight.ctensor, meta.ctensor, bias.ctensor, activation) + lib.Atg_SparseSemiStructuredApplyDense(ptr, input.ctensor, threadMasks.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_SparseSemiStructuredApplyDense() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_SparseSemiStructuredApplyDense") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func _SparseSemiStructuredLinear(input *Tensor, weight *Tensor, meta *Tensor, bias *Tensor, activation string, outDtype gotch.DType)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_SparseSemiStructuredLinear(ptr, input.ctensor, weight.ctensor, meta.ctensor, bias.ctensor, activation, outDtype.CInt()) if err = TorchErr(); err != nil { err = fmt.Errorf("_SparseSemiStructuredLinear() failed: %w", err) return retVal, err @@ -6115,6 +6967,47 @@ func _SparseSemiStructuredLinear(input *Tensor, weight *Tensor, meta *Tensor, bi // func.returns = `fixed 1`: // -------------------------- +func _SparseSemiStructuredMm(mat1 *Tensor, mat1Meta *Tensor, mat2 *Tensor, outDtype gotch.DType)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_SparseSemiStructuredMm(ptr, mat1.ctensor, mat1Meta.ctensor, mat2.ctensor, outDtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_SparseSemiStructuredMm() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_SparseSemiStructuredMm") + + return retVal, err +} + +// func.returns = `fixed ntensors`: +// --------------------------------- +func _SparseSemiStructuredTile(input *Tensor, algorithm string, useCutlass bool)(retVal0 *Tensor, retVal1 *Tensor, retVal2 *Tensor, retVal3 *Tensor, retVal4 *Tensor, err error) { + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr3 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr2)) + unsafe.Sizeof(ctensorPtr0))) + ctensorPtr4 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr3)) + unsafe.Sizeof(ctensorPtr0))) + + cuseCutlass := int32(0) + if useCutlass { cuseCutlass = int32(1) } + lib.Atg_SparseSemiStructuredTile(ctensorPtr0, input.ctensor, algorithm, cuseCutlass) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_SparseSemiStructuredTile() failed: %w", err) + return retVal0, retVal1, retVal2, retVal3, retVal4, err + } + retVal0 = newTensor(*ctensorPtr0, "_SparseSemiStructuredTile_0") + retVal1 = newTensor(*ctensorPtr1, "_SparseSemiStructuredTile_1") + retVal2 = newTensor(*ctensorPtr2, "_SparseSemiStructuredTile_2") + retVal3 = newTensor(*ctensorPtr3, "_SparseSemiStructuredTile_3") + retVal4 = newTensor(*ctensorPtr4, "_SparseSemiStructuredTile_4") + + return retVal0, retVal1, retVal2, retVal3, retVal4, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) _SparseSoftmax(dim int64, halfToFloat bool, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -6396,6 +7289,24 @@ func _SpdiagsOut(out *Tensor, diagonals *Tensor, offsets *Tensor, shape []int64, // func.returns = `fixed 1`: // -------------------------- +func _Spsolve(a *Tensor, b *Tensor, left bool)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + cleft := int32(0) + if left { cleft = int32(1) } + lib.Atg_Spsolve(ptr, a.ctensor, b.ctensor, cleft) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_Spsolve() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_Spsolve") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func _Stack(tensors []*Tensor, dim int64)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -6789,6 +7700,25 @@ func _TestOptionalIntlistOut(out *Tensor, values *Tensor, addends []int64)(retVa // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) _TestParallelMaterialize(numParallel int64, skipFirst bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + cskipFirst := int32(0) + if skipFirst { cskipFirst = int32(1) } + lib.Atg_TestParallelMaterialize(ptr, ts.ctensor, numParallel, cskipFirst) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_TestParallelMaterialize() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_TestParallelMaterialize") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) _TestSerializationSubcmul(other *Tensor, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -7526,46 +8456,6 @@ func _UnpackDual(dual *Tensor, level int64)(retVal0 *Tensor, retVal1 *Tensor, er // func.returns = `fixed 1`: // -------------------------- -func(ts *Tensor) _UnsafeIndex(indices []*Tensor, del bool)(retVal *Tensor, err error) { - if del { defer ts.MustDrop() } - ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) - - var cindices []lib.Ctensor - for _, t := range indices {cindices = append(cindices, t.ctensor)} - lib.Atg_UnsafeIndex(ptr, ts.ctensor, cindices, len(cindices)) - if err = TorchErr(); err != nil { - err = fmt.Errorf("_UnsafeIndex() failed: %w", err) - return retVal, err - } - retVal = newTensor(*ptr, "_UnsafeIndex") - - return retVal, err -} - -// func.returns = `fixed 1`: -// -------------------------- - -func(ts *Tensor) _UnsafeIndexPut(indices []*Tensor, values *Tensor, accumulate bool, del bool)(retVal *Tensor, err error) { - if del { defer ts.MustDrop() } - ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) - - var cindices []lib.Ctensor - for _, t := range indices {cindices = append(cindices, t.ctensor)} -caccumulate := int32(0) - if accumulate { caccumulate = int32(1) } - lib.Atg_UnsafeIndexPut(ptr, ts.ctensor, cindices, len(cindices), values.ctensor, caccumulate) - if err = TorchErr(); err != nil { - err = fmt.Errorf("_UnsafeIndexPut() failed: %w", err) - return retVal, err - } - retVal = newTensor(*ptr, "_UnsafeIndexPut") - - return retVal, err -} - -// func.returns = `fixed 1`: -// -------------------------- - func(ts *Tensor) _UnsafeView(size []int64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -8414,6 +9304,74 @@ func(ts *Tensor) _Version(del bool)(retVal int64, err error) { // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) _WeightInt4packMm(mat2 *Tensor, qGroupSize int64, qScaleAndZeros *Tensor, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_WeightInt4packMm(ptr, ts.ctensor, mat2.ctensor, qGroupSize, qScaleAndZeros.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_WeightInt4packMm() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_WeightInt4packMm") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _WeightInt4packMmForCpu(mat2 *Tensor, qGroupSize int64, qScaleAndZeros *Tensor, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_WeightInt4packMmForCpu(ptr, ts.ctensor, mat2.ctensor, qGroupSize, qScaleAndZeros.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_WeightInt4packMmForCpu() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_WeightInt4packMmForCpu") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _WeightInt4packMmWithScalesAndZeros(mat2 *Tensor, qGroupSize int64, qScale *Tensor, qZeros *Tensor, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_WeightInt4packMmWithScalesAndZeros(ptr, ts.ctensor, mat2.ctensor, qGroupSize, qScale.ctensor, qZeros.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_WeightInt4packMmWithScalesAndZeros() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_WeightInt4packMmWithScalesAndZeros") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) _WeightInt8packMm(mat2 *Tensor, scales *Tensor, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_WeightInt8packMm(ptr, ts.ctensor, mat2.ctensor, scales.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_WeightInt8packMm() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_WeightInt8packMm") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func _WeightNorm(v *Tensor, g *Tensor, dim int64)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -8515,6 +9473,38 @@ func _WeightNormInterfaceOut(out0 *Tensor, out1 *Tensor, v *Tensor, g *Tensor, d // func.returns = `fixed 1`: // -------------------------- +func _WrappedLinearPrepack(weight *Tensor, weightScale *Tensor, weightZeroPoint *Tensor, bias *Tensor)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_WrappedLinearPrepack(ptr, weight.ctensor, weightScale.ctensor, weightZeroPoint.ctensor, bias.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_WrappedLinearPrepack() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_WrappedLinearPrepack") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func _WrappedQuantizedLinearPrepacked(input *Tensor, inputScale *Tensor, inputZeroPoint *Tensor, packedWeight *Tensor, outputScale *Tensor, outputZeroPoint *Tensor, outChannel int64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.Atg_WrappedQuantizedLinearPrepacked(ptr, input.ctensor, inputScale.ctensor, inputZeroPoint.ctensor, packedWeight.ctensor, outputScale.ctensor, outputZeroPoint.ctensor, outChannel) + if err = TorchErr(); err != nil { + err = fmt.Errorf("_WrappedQuantizedLinearPrepacked() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "_WrappedQuantizedLinearPrepacked") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) Abs(del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -8733,6 +9723,24 @@ func(ts *Tensor) AdaptiveAvgPool1d(outputSize []int64, del bool)(retVal *Tensor, // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) AdaptiveAvgPool1dOut(out *Tensor, outputSize []int64, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + outputSizeLen := len(outputSize) + lib.AtgAdaptiveAvgPool1dOut(ptr, out.ctensor, ts.ctensor, outputSize, outputSizeLen) + if err = TorchErr(); err != nil { + err = fmt.Errorf("AdaptiveAvgPool1dOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "AdaptiveAvgPool1dOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) AdaptiveAvgPool2d(outputSize []int64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -9268,6 +10276,40 @@ func(ts *Tensor) Addmm_(mat1 *Tensor, mat2 *Tensor)(err error) { // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) AddmmDtype(mat1 *Tensor, mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgAddmmDtype(ptr, ts.ctensor, mat1.ctensor, mat2.ctensor, outDtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("AddmmDtype() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "AddmmDtype") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) AddmmDtypeOut(out *Tensor, mat1 *Tensor, mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgAddmmDtypeOut(ptr, out.ctensor, ts.ctensor, mat1.ctensor, mat2.ctensor, outDtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("AddmmDtypeOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "AddmmDtypeOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) AddmmOut(out *Tensor, mat1 *Tensor, mat2 *Tensor, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -9580,6 +10622,46 @@ func(ts *Tensor) AllDim(dim int64, keepdim bool, del bool)(retVal *Tensor, err e // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) AllDims(dim []int64, keepdim bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + dimLen := len(dim) +ckeepdim := int32(0) + if keepdim { ckeepdim = int32(1) } + lib.AtgAllDims(ptr, ts.ctensor, dim, dimLen, ckeepdim) + if err = TorchErr(); err != nil { + err = fmt.Errorf("AllDims() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "AllDims") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) AllDimsOut(out *Tensor, dim []int64, keepdim bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + dimLen := len(dim) +ckeepdim := int32(0) + if keepdim { ckeepdim = int32(1) } + lib.AtgAllDimsOut(ptr, out.ctensor, ts.ctensor, dim, dimLen, ckeepdim) + if err = TorchErr(); err != nil { + err = fmt.Errorf("AllDimsOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "AllDimsOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) AllOut(out *Tensor, dim int64, keepdim bool, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -9869,6 +10951,46 @@ func(ts *Tensor) AnyDim(dim int64, keepdim bool, del bool)(retVal *Tensor, err e // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) AnyDims(dim []int64, keepdim bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + dimLen := len(dim) +ckeepdim := int32(0) + if keepdim { ckeepdim = int32(1) } + lib.AtgAnyDims(ptr, ts.ctensor, dim, dimLen, ckeepdim) + if err = TorchErr(); err != nil { + err = fmt.Errorf("AnyDims() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "AnyDims") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) AnyDimsOut(out *Tensor, dim []int64, keepdim bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + dimLen := len(dim) +ckeepdim := int32(0) + if keepdim { ckeepdim = int32(1) } + lib.AtgAnyDimsOut(ptr, out.ctensor, ts.ctensor, dim, dimLen, ckeepdim) + if err = TorchErr(); err != nil { + err = fmt.Errorf("AnyDimsOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "AnyDimsOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) AnyOut(out *Tensor, dim int64, keepdim bool, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -10938,6 +12060,30 @@ ccountIncludePad := int32(0) // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) AvgPool1dOut(out *Tensor, kernelSize []int64, stride []int64, padding []int64, ceilMode bool, countIncludePad bool, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + kernelSizeLen := len(kernelSize) +strideLen := len(stride) +paddingLen := len(padding) +cceilMode := int32(0) + if ceilMode { cceilMode = int32(1) } +ccountIncludePad := int32(0) + if countIncludePad { ccountIncludePad = int32(1) } + lib.AtgAvgPool1dOut(ptr, out.ctensor, ts.ctensor, kernelSize, kernelSizeLen, stride, strideLen, padding, paddingLen, cceilMode, ccountIncludePad) + if err = TorchErr(); err != nil { + err = fmt.Errorf("AvgPool1dOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "AvgPool1dOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) AvgPool2d(kernelSize []int64, stride []int64, padding []int64, ceilMode bool, countIncludePad bool, divisorOverride []int64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -11211,6 +12357,40 @@ func(ts *Tensor) Baddbmm_(batch1 *Tensor, batch2 *Tensor)(err error) { // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) BaddbmmDtype(batch1 *Tensor, batch2 *Tensor, outDtype gotch.DType, beta *Scalar, alpha *Scalar, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgBaddbmmDtype(ptr, ts.ctensor, batch1.ctensor, batch2.ctensor, outDtype.CInt(), beta.cscalar, alpha.cscalar) + if err = TorchErr(); err != nil { + err = fmt.Errorf("BaddbmmDtype() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "BaddbmmDtype") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) BaddbmmDtypeOut(out *Tensor, batch1 *Tensor, batch2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgBaddbmmDtypeOut(ptr, out.ctensor, ts.ctensor, batch1.ctensor, batch2.ctensor, outDtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("BaddbmmDtypeOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "BaddbmmDtypeOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) BaddbmmOut(out *Tensor, batch1 *Tensor, batch2 *Tensor, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -12668,6 +13848,40 @@ func(ts *Tensor) Bmm(mat2 *Tensor, del bool)(retVal *Tensor, err error) { // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) BmmDtype(mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgBmmDtype(ptr, ts.ctensor, mat2.ctensor, outDtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("BmmDtype() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "BmmDtype") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) BmmDtypeOut(out *Tensor, mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgBmmDtypeOut(ptr, out.ctensor, ts.ctensor, mat2.ctensor, outDtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("BmmDtypeOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "BmmDtypeOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) BmmOut(out *Tensor, mat2 *Tensor, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -12784,9 +13998,9 @@ cright := int32(0) // func.returns = `bool`: // -------------------------- -func CanCast(from gotch.DType, to gotch.DType)(retVal bool, err error) { +func CanCast(from_ gotch.DType, to gotch.DType)(retVal bool, err error) { - retVal = lib.AtgCanCast(from.CInt(), to.CInt()) + retVal = lib.AtgCanCast(from_.CInt(), to.CInt()) if err = TorchErr(); err != nil { err = fmt.Errorf("CanCast() failed: %w", err) return retVal, err @@ -17971,6 +19185,38 @@ func FbgemmLinearFp16WeightFp32Activation(input *Tensor, packedWeight *Tensor, b // func.returns = `fixed 1`: // -------------------------- +func FbgemmLinearFp16WeightFp32ActivationOut(input *Tensor, packedWeight *Tensor, bias *Tensor, output *Tensor)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgFbgemmLinearFp16WeightFp32ActivationOut(ptr, input.ctensor, packedWeight.ctensor, bias.ctensor, output.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("FbgemmLinearFp16WeightFp32ActivationOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "FbgemmLinearFp16WeightFp32ActivationOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func FbgemmLinearFp16WeightOut(input *Tensor, packedWeight *Tensor, bias *Tensor, output *Tensor)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgFbgemmLinearFp16WeightOut(ptr, input.ctensor, packedWeight.ctensor, bias.ctensor, output.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("FbgemmLinearFp16WeightOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "FbgemmLinearFp16WeightOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func FbgemmLinearInt8Weight(input *Tensor, weight *Tensor, packed *Tensor, colOffsets *Tensor, weightScale *Scalar, weightZeroPoint *Scalar, bias *Tensor)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -19476,6 +20722,23 @@ func(ts *Tensor) FloorDivideScalar_(other *Scalar)(err error) { // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) FloorDivideScalarOut(out *Tensor, other *Scalar, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgFloorDivideScalarOut(ptr, out.ctensor, ts.ctensor, other.cscalar) + if err = TorchErr(); err != nil { + err = fmt.Errorf("FloorDivideScalarOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "FloorDivideScalarOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) FloorOut(out *Tensor, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -21644,6 +22907,46 @@ func(ts *Tensor) HardtanhOut(out *Tensor, del bool)(retVal *Tensor, err error) { // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) HashTensor(dim []int64, keepdim bool, mode int64, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + dimLen := len(dim) +ckeepdim := int32(0) + if keepdim { ckeepdim = int32(1) } + lib.AtgHashTensor(ptr, ts.ctensor, dim, dimLen, ckeepdim, mode) + if err = TorchErr(); err != nil { + err = fmt.Errorf("HashTensor() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "HashTensor") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) HashTensorOut(out *Tensor, dim []int64, keepdim bool, mode int64, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + dimLen := len(dim) +ckeepdim := int32(0) + if keepdim { ckeepdim = int32(1) } + lib.AtgHashTensorOut(ptr, out.ctensor, ts.ctensor, dim, dimLen, ckeepdim, mode) + if err = TorchErr(); err != nil { + err = fmt.Errorf("HashTensorOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "HashTensorOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) Heaviside(values *Tensor, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -22340,27 +23643,6 @@ func(ts *Tensor) IndexFillIntTensorOut(out *Tensor, dim int64, index *Tensor, va // func.returns = `fixed 1`: // -------------------------- -func(ts *Tensor) IndexPutOut(out *Tensor, indices []*Tensor, values *Tensor, accumulate bool, del bool)(retVal *Tensor, err error) { - if del { defer ts.MustDrop() } - ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) - - var cindices []lib.Ctensor - for _, t := range indices {cindices = append(cindices, t.ctensor)} -caccumulate := int32(0) - if accumulate { caccumulate = int32(1) } - lib.AtgIndexPutOut(ptr, out.ctensor, ts.ctensor, cindices, len(cindices), values.ctensor, caccumulate) - if err = TorchErr(); err != nil { - err = fmt.Errorf("IndexPutOut() failed: %w", err) - return retVal, err - } - retVal = newTensor(*ptr, "IndexPutOut") - - return retVal, err -} - -// func.returns = `fixed 1`: -// -------------------------- - func(ts *Tensor) IndexReduce(dim int64, index *Tensor, source *Tensor, reduce string, includeSelf bool, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -22468,25 +23750,6 @@ func(ts *Tensor) IndexSelectOut(out *Tensor, dim int64, index *Tensor, del bool) // func.returns = `fixed 1`: // -------------------------- -func(ts *Tensor) IndexTensorOut(out *Tensor, indices []*Tensor, del bool)(retVal *Tensor, err error) { - if del { defer ts.MustDrop() } - ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) - - var cindices []lib.Ctensor - for _, t := range indices {cindices = append(cindices, t.ctensor)} - lib.AtgIndexTensorOut(ptr, out.ctensor, ts.ctensor, cindices, len(cindices)) - if err = TorchErr(); err != nil { - err = fmt.Errorf("IndexTensorOut() failed: %w", err) - return retVal, err - } - retVal = newTensor(*ptr, "IndexTensorOut") - - return retVal, err -} - -// func.returns = `fixed 1`: -// -------------------------- - func(ts *Tensor) Indices(del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -26006,6 +27269,102 @@ func LinspaceOut(out *Tensor, start *Scalar, end *Scalar, steps int64)(retVal *T // func.returns = `fixed 1`: // -------------------------- +func LinspaceScalarTensor(start *Scalar, end *Tensor, steps int64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLinspaceScalarTensor(ptr, start.cscalar, end.ctensor, steps, optionsKind.CInt(), optionsDevice.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LinspaceScalarTensor() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LinspaceScalarTensor") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func LinspaceScalarTensorOut(out *Tensor, start *Scalar, end *Tensor, steps int64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLinspaceScalarTensorOut(ptr, out.ctensor, start.cscalar, end.ctensor, steps) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LinspaceScalarTensorOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LinspaceScalarTensorOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func LinspaceTensorScalar(start *Tensor, end *Scalar, steps int64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLinspaceTensorScalar(ptr, start.ctensor, end.cscalar, steps, optionsKind.CInt(), optionsDevice.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LinspaceTensorScalar() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LinspaceTensorScalar") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func LinspaceTensorScalarOut(out *Tensor, start *Tensor, end *Scalar, steps int64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLinspaceTensorScalarOut(ptr, out.ctensor, start.ctensor, end.cscalar, steps) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LinspaceTensorScalarOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LinspaceTensorScalarOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func LinspaceTensorTensor(start *Tensor, end *Tensor, steps int64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLinspaceTensorTensor(ptr, start.ctensor, end.ctensor, steps, optionsKind.CInt(), optionsDevice.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LinspaceTensorTensor() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LinspaceTensorTensor") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func LinspaceTensorTensorOut(out *Tensor, start *Tensor, end *Tensor, steps int64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLinspaceTensorTensorOut(ptr, out.ctensor, start.ctensor, end.ctensor, steps) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LinspaceTensorTensorOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LinspaceTensorTensorOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) Log(del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -26823,6 +28182,102 @@ func LogspaceOut(out *Tensor, start *Scalar, end *Scalar, steps int64, base floa // func.returns = `fixed 1`: // -------------------------- +func LogspaceScalarTensor(start *Scalar, end *Tensor, steps int64, base float64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLogspaceScalarTensor(ptr, start.cscalar, end.ctensor, steps, base, optionsKind.CInt(), optionsDevice.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LogspaceScalarTensor() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LogspaceScalarTensor") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func LogspaceScalarTensorOut(out *Tensor, start *Scalar, end *Tensor, steps int64, base float64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLogspaceScalarTensorOut(ptr, out.ctensor, start.cscalar, end.ctensor, steps, base) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LogspaceScalarTensorOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LogspaceScalarTensorOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func LogspaceTensorScalar(start *Tensor, end *Scalar, steps int64, base float64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLogspaceTensorScalar(ptr, start.ctensor, end.cscalar, steps, base, optionsKind.CInt(), optionsDevice.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LogspaceTensorScalar() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LogspaceTensorScalar") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func LogspaceTensorScalarOut(out *Tensor, start *Tensor, end *Scalar, steps int64, base float64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLogspaceTensorScalarOut(ptr, out.ctensor, start.ctensor, end.cscalar, steps, base) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LogspaceTensorScalarOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LogspaceTensorScalarOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func LogspaceTensorTensor(start *Tensor, end *Tensor, steps int64, base float64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLogspaceTensorTensor(ptr, start.ctensor, end.ctensor, steps, base, optionsKind.CInt(), optionsDevice.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LogspaceTensorTensor() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LogspaceTensorTensor") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func LogspaceTensorTensorOut(out *Tensor, start *Tensor, end *Tensor, steps int64, base float64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgLogspaceTensorTensorOut(ptr, out.ctensor, start.ctensor, end.ctensor, steps, base) + if err = TorchErr(); err != nil { + err = fmt.Errorf("LogspaceTensorTensorOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "LogspaceTensorTensorOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) Logsumexp(dim []int64, keepdim bool, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -27271,6 +28726,23 @@ func(ts *Tensor) MaskedScatter_(mask *Tensor, source *Tensor)(err error) { // func.returns = `fixed 1`: // -------------------------- +func MaskedScatterBackward(gradOutput *Tensor, mask *Tensor, sizes []int64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + sizesLen := len(sizes) + lib.AtgMaskedScatterBackward(ptr, gradOutput.ctensor, mask.ctensor, sizes, sizesLen) + if err = TorchErr(); err != nil { + err = fmt.Errorf("MaskedScatterBackward() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "MaskedScatterBackward") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) MaskedScatterOut(out *Tensor, mask *Tensor, source *Tensor, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -28039,6 +29511,23 @@ ckeepdim := int32(0) // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) MeanDtypeOut(out *Tensor, dtype gotch.DType, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgMeanDtypeOut(ptr, out.ctensor, ts.ctensor, dtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("MeanDtypeOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "MeanDtypeOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) MeanOut(out *Tensor, dim []int64, keepdim bool, dtype gotch.DType, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -29127,14 +30616,15 @@ inputSizeLen := len(inputSize) // func.returns = `fixed 1`: // -------------------------- -func(ts *Tensor) MkldnnReorderConv3dWeight(padding []int64, stride []int64, dilation []int64, groups int64, del bool)(retVal *Tensor, err error) { +func(ts *Tensor) MkldnnReorderConv3dWeight(padding []int64, stride []int64, dilation []int64, groups int64, inputSize []int64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) paddingLen := len(padding) strideLen := len(stride) dilationLen := len(dilation) - lib.AtgMkldnnReorderConv3dWeight(ptr, ts.ctensor, padding, paddingLen, stride, strideLen, dilation, dilationLen, groups) +inputSizeLen := len(inputSize) + lib.AtgMkldnnReorderConv3dWeight(ptr, ts.ctensor, padding, paddingLen, stride, strideLen, dilation, dilationLen, groups, inputSize, inputSizeLen) if err = TorchErr(); err != nil { err = fmt.Errorf("MkldnnReorderConv3dWeight() failed: %w", err) return retVal, err @@ -29147,14 +30637,15 @@ dilationLen := len(dilation) // func.returns = `fixed 1`: // -------------------------- -func(ts *Tensor) MkldnnReorderConv3dWeightOut(out *Tensor, padding []int64, stride []int64, dilation []int64, groups int64, del bool)(retVal *Tensor, err error) { +func(ts *Tensor) MkldnnReorderConv3dWeightOut(out *Tensor, padding []int64, stride []int64, dilation []int64, groups int64, inputSize []int64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) paddingLen := len(padding) strideLen := len(stride) dilationLen := len(dilation) - lib.AtgMkldnnReorderConv3dWeightOut(ptr, out.ctensor, ts.ctensor, padding, paddingLen, stride, strideLen, dilation, dilationLen, groups) +inputSizeLen := len(inputSize) + lib.AtgMkldnnReorderConv3dWeightOut(ptr, out.ctensor, ts.ctensor, padding, paddingLen, stride, strideLen, dilation, dilationLen, groups, inputSize, inputSizeLen) if err = TorchErr(); err != nil { err = fmt.Errorf("MkldnnReorderConv3dWeightOut() failed: %w", err) return retVal, err @@ -29324,6 +30815,40 @@ func(ts *Tensor) Mm(mat2 *Tensor, del bool)(retVal *Tensor, err error) { // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) MmDtype(mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgMmDtype(ptr, ts.ctensor, mat2.ctensor, outDtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("MmDtype() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "MmDtype") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) MmDtypeOut(out *Tensor, mat2 *Tensor, outDtype gotch.DType, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgMmDtypeOut(ptr, out.ctensor, ts.ctensor, mat2.ctensor, outDtype.CInt()) + if err = TorchErr(); err != nil { + err = fmt.Errorf("MmDtypeOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "MmDtypeOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) MmOut(out *Tensor, mat2 *Tensor, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -31981,14 +33506,14 @@ var cvalueVal float64 = 0.0 // func.returns = `fixed 1`: // -------------------------- -func PadSequence(sequences []*Tensor, batchFirst bool, paddingValue float64)(retVal *Tensor, err error) { +func PadSequence(sequences []*Tensor, batchFirst bool, paddingValue float64, paddingSide string)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) var csequences []lib.Ctensor for _, t := range sequences {csequences = append(csequences, t.ctensor)} cbatchFirst := int32(0) if batchFirst { cbatchFirst = int32(1) } - lib.AtgPadSequence(ptr, csequences, len(csequences), cbatchFirst, paddingValue) + lib.AtgPadSequence(ptr, csequences, len(csequences), cbatchFirst, paddingValue, paddingSide) if err = TorchErr(); err != nil { err = fmt.Errorf("PadSequence() failed: %w", err) return retVal, err @@ -33415,6 +34940,40 @@ func(ts *Tensor) RandintLikeOut(out *Tensor, high int64, del bool)(retVal *Tenso // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) RandintLikeTensor(high *Tensor, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgRandintLikeTensor(ptr, ts.ctensor, high.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("RandintLikeTensor() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "RandintLikeTensor") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + +func(ts *Tensor) RandintLikeTensorOut(out *Tensor, high *Tensor, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + lib.AtgRandintLikeTensorOut(ptr, out.ctensor, ts.ctensor, high.ctensor) + if err = TorchErr(); err != nil { + err = fmt.Errorf("RandintLikeTensorOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "RandintLikeTensorOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func RandintLow(low int64, high int64, size []int64, optionsKind gotch.DType, optionsDevice gotch.Device)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -34967,6 +36526,29 @@ func(ts *Tensor) RetainsGrad(del bool)(retVal bool, err error) { return retVal, err } +// func.returns = `fixed 1`: +// -------------------------- + +func RmsNorm(input *Tensor, normalizedShape []int64, weight *Tensor, eps []float64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + normalizedShapeLen := len(normalizedShape) +var cepsVal float64 = 0.0 + var cepsNull int = 1 + if len(eps) > 0 { + cepsVal = eps[0] + cepsNull = 0 + } + lib.AtgRmsNorm(ptr, input.ctensor, normalizedShape, normalizedShapeLen, weight.ctensor, cepsVal, cepsNull) + if err = TorchErr(); err != nil { + err = fmt.Errorf("RmsNorm() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "RmsNorm") + + return retVal, err +} + // func.returns = `fixed ntensors`: // --------------------------------- func RnnRelu(input *Tensor, hx *Tensor, params []*Tensor, hasBiases bool, numLayers int64, dropout float64, train bool, bidirectional bool, batchFirst bool)(retVal0 *Tensor, retVal1 *Tensor, err error) { @@ -35480,6 +37062,26 @@ cselfIsResult := int32(0) return retVal, err } +// func.returns = `fixed ntensors`: +// --------------------------------- +func(ts *Tensor) RreluWithNoiseFunctional(noise *Tensor, training bool, del bool)(retVal0 *Tensor, retVal1 *Tensor, err error) { + if del { defer ts.MustDrop() } + ctensorPtr0 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr0)) + unsafe.Sizeof(ctensorPtr0))) + + ctraining := int32(0) + if training { ctraining = int32(1) } + lib.AtgRreluWithNoiseFunctional(ctensorPtr0, ts.ctensor, noise.ctensor, ctraining) + if err = TorchErr(); err != nil { + err = fmt.Errorf("RreluWithNoiseFunctional() failed: %w", err) + return retVal0, retVal1, err + } + retVal0 = newTensor(*ctensorPtr0, "RreluWithNoiseFunctional_0") + retVal1 = newTensor(*ctensorPtr1, "RreluWithNoiseFunctional_1") + + return retVal0, retVal1, err +} + // func.returns = `fixed 1`: // -------------------------- @@ -35652,7 +37254,7 @@ func ScalarTensorOut(out *Tensor, s *Scalar)(retVal *Tensor, err error) { // func.returns = `fixed 1`: // -------------------------- -func ScaledDotProductAttention(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, scale []float64)(retVal *Tensor, err error) { +func ScaledDotProductAttention(query *Tensor, key *Tensor, value *Tensor, attnMask *Tensor, dropoutP float64, isCausal bool, scale []float64, enableGqa bool)(retVal *Tensor, err error) { ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) cisCausal := int32(0) @@ -35663,7 +37265,9 @@ var cscaleVal float64 = 0.0 cscaleVal = scale[0] cscaleNull = 0 } - lib.AtgScaledDotProductAttention(ptr, query.ctensor, key.ctensor, value.ctensor, attnMask.ctensor, dropoutP, cisCausal, cscaleVal, cscaleNull) +cenableGqa := int32(0) + if enableGqa { cenableGqa = int32(1) } + lib.AtgScaledDotProductAttention(ptr, query.ctensor, key.ctensor, value.ctensor, attnMask.ctensor, dropoutP, cisCausal, cscaleVal, cscaleNull, cenableGqa) if err = TorchErr(); err != nil { err = fmt.Errorf("ScaledDotProductAttention() failed: %w", err) return retVal, err @@ -36904,6 +38508,35 @@ var cendVal int64 = 0 // func.returns = `fixed 1`: // -------------------------- +func(ts *Tensor) SliceInverse(src *Tensor, dim int64, start []int64, end []int64, step int64, del bool)(retVal *Tensor, err error) { + if del { defer ts.MustDrop() } + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + var cstartVal int64 = 0 + var cstartNull int = 1 + if len(start) > 0 { + cstartVal = start[0] + cstartNull = 0 + } +var cendVal int64 = 0 + var cendNull int = 1 + if len(end) > 0 { + cendVal = end[0] + cendNull = 0 + } + lib.AtgSliceInverse(ptr, ts.ctensor, src.ctensor, dim, cstartVal, cstartNull, cendVal, cendNull, step) + if err = TorchErr(); err != nil { + err = fmt.Errorf("SliceInverse() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "SliceInverse") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) SliceScatter(src *Tensor, dim int64, start []int64, end []int64, step int64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -41425,7 +43058,7 @@ ckeepdim := int32(0) // func.returns = `fixed 1`: // -------------------------- -func(ts *Tensor) Stft(nFft int64, hopLength []int64, winLength []int64, window *Tensor, normalized bool, onesided bool, returnComplex bool, del bool)(retVal *Tensor, err error) { +func(ts *Tensor) Stft(nFft int64, hopLength []int64, winLength []int64, window *Tensor, normalized bool, onesided bool, returnComplex bool, alignToWindow bool, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -41447,7 +43080,9 @@ conesided := int32(0) if onesided { conesided = int32(1) } creturnComplex := int32(0) if returnComplex { creturnComplex = int32(1) } - lib.AtgStft(ptr, ts.ctensor, nFft, chopLengthVal, chopLengthNull, cwinLengthVal, cwinLengthNull, window.ctensor, cnormalized, conesided, creturnComplex) +calignToWindow := int32(0) + if alignToWindow { calignToWindow = int32(1) } + lib.AtgStft(ptr, ts.ctensor, nFft, chopLengthVal, chopLengthNull, cwinLengthVal, cwinLengthNull, window.ctensor, cnormalized, conesided, creturnComplex, calignToWindow) if err = TorchErr(); err != nil { err = fmt.Errorf("Stft() failed: %w", err) return retVal, err @@ -41460,7 +43095,7 @@ creturnComplex := int32(0) // func.returns = `fixed 1`: // -------------------------- -func(ts *Tensor) StftCenter(nFft int64, hopLength []int64, winLength []int64, window *Tensor, center bool, padMode string, normalized bool, onesided bool, returnComplex bool, del bool)(retVal *Tensor, err error) { +func(ts *Tensor) StftCenter(nFft int64, hopLength []int64, winLength []int64, window *Tensor, center bool, padMode string, normalized bool, onesided bool, returnComplex bool, alignToWindow bool, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -41484,7 +43119,9 @@ conesided := int32(0) if onesided { conesided = int32(1) } creturnComplex := int32(0) if returnComplex { creturnComplex = int32(1) } - lib.AtgStftCenter(ptr, ts.ctensor, nFft, chopLengthVal, chopLengthNull, cwinLengthVal, cwinLengthNull, window.ctensor, ccenter, padMode, cnormalized, conesided, creturnComplex) +calignToWindow := int32(0) + if alignToWindow { calignToWindow = int32(1) } + lib.AtgStftCenter(ptr, ts.ctensor, nFft, chopLengthVal, chopLengthNull, cwinLengthVal, cwinLengthNull, window.ctensor, ccenter, padMode, cnormalized, conesided, creturnComplex, calignToWindow) if err = TorchErr(); err != nil { err = fmt.Errorf("StftCenter() failed: %w", err) return retVal, err @@ -43963,6 +45600,26 @@ scaleFactorsLen := len(scaleFactors) // func.returns = `fixed 1`: // -------------------------- +func UpsampleBilinear2dVecOut(out *Tensor, input *Tensor, outputSize []int64, alignCorners bool, scaleFactors []float64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + outputSizeLen := len(outputSize) +calignCorners := int32(0) + if alignCorners { calignCorners = int32(1) } +scaleFactorsLen := len(scaleFactors) + lib.AtgUpsampleBilinear2dVecOut(ptr, out.ctensor, input.ctensor, outputSize, outputSizeLen, calignCorners, scaleFactors, scaleFactorsLen) + if err = TorchErr(); err != nil { + err = fmt.Errorf("UpsampleBilinear2dVecOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "UpsampleBilinear2dVecOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) UpsampleLinear1d(outputSize []int64, alignCorners bool, scales []float64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) @@ -44339,6 +45996,24 @@ scaleFactorsLen := len(scaleFactors) // func.returns = `fixed 1`: // -------------------------- +func UpsampleNearest2dVecOut(out *Tensor, input *Tensor, outputSize []int64, scaleFactors []float64)(retVal *Tensor, err error) { + ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) + + outputSizeLen := len(outputSize) +scaleFactorsLen := len(scaleFactors) + lib.AtgUpsampleNearest2dVecOut(ptr, out.ctensor, input.ctensor, outputSize, outputSizeLen, scaleFactors, scaleFactorsLen) + if err = TorchErr(); err != nil { + err = fmt.Errorf("UpsampleNearest2dVecOut() failed: %w", err) + return retVal, err + } + retVal = newTensor(*ptr, "UpsampleNearest2dVecOut") + + return retVal, err +} + +// func.returns = `fixed 1`: +// -------------------------- + func(ts *Tensor) UpsampleNearest3d(outputSize []int64, scalesD []float64, scalesH []float64, scalesW []float64, del bool)(retVal *Tensor, err error) { if del { defer ts.MustDrop() } ptr := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) From 76a878c3d1459cf17f36215603f7e18c64ddc848 Mon Sep 17 00:00:00 2001 From: tlc Date: Sat, 14 Feb 2026 18:10:09 -0800 Subject: [PATCH 2/7] Unit testing fallout. Added a make file as a way to say how to run tests. The pytorch 2.10.0 is local. --- Makefile | 116 +++++++++++++++++++++++++++++++++++++++ nn/init_test.go | 3 + nn/optimizer_test.go | 19 +++++-- ts/basic-example_test.go | 4 +- ts/patch-example_test.go | 12 ++-- 5 files changed, 141 insertions(+), 13 deletions(-) create mode 100644 Makefile diff --git a/Makefile b/Makefile new file mode 100644 index 0000000..212605b --- /dev/null +++ b/Makefile @@ -0,0 +1,116 @@ +# Makefile for gotch - PyTorch 2.10.0 Go bindings +# macOS with MPS support + +# Path to libtorch installation +LIBTORCH_PATH := /Users/tlc/src/gotch/libtorch-2.10.0-macos + +# Environment variables for building and testing +export CPATH := $(LIBTORCH_PATH)/include/torch/csrc/api/include:$(LIBTORCH_PATH)/include +export LIBRARY_PATH := $(LIBTORCH_PATH)/lib +export DYLD_LIBRARY_PATH := $(LIBTORCH_PATH)/lib + +# Go test flags +TEST_FLAGS := -v +TEST_TIMEOUT := 5m + +.PHONY: all build test test-nn test-ts test-all clean help + +# Default target +all: build + +# Build all core packages +build: + @echo "Building gotch core packages..." + @go build -v . ./ts ./nn ./vision + +# Run all tests +test-all: test-nn test-ts + @echo "All tests completed" + +# Run nn package tests +test-nn: + @echo "Running nn package tests..." + @go test $(TEST_FLAGS) -timeout $(TEST_TIMEOUT) ./nn + +# Run ts package tests +test-ts: + @echo "Running ts package tests..." + @go test $(TEST_FLAGS) -timeout $(TEST_TIMEOUT) ./ts + +# Run specific test in nn package +# Usage: make test-nn-specific TEST=TestInitTensor_Memcheck +test-nn-specific: + @echo "Running specific nn test: $(TEST)..." + @go test $(TEST_FLAGS) -timeout $(TEST_TIMEOUT) -run $(TEST) ./nn + +# Run specific test in ts package +# Usage: make test-ts-specific TEST=TestTensor +test-ts-specific: + @echo "Running specific ts test: $(TEST)..." + @go test $(TEST_FLAGS) -timeout $(TEST_TIMEOUT) -run $(TEST) ./ts + +# Run tests with coverage +test-nn-coverage: + @echo "Running nn tests with coverage..." + @go test -v -timeout $(TEST_TIMEOUT) -coverprofile=coverage-nn.out ./nn + @go tool cover -html=coverage-nn.out -o coverage-nn.html + @echo "Coverage report saved to coverage-nn.html" + +test-ts-coverage: + @echo "Running ts tests with coverage..." + @go test -v -timeout $(TEST_TIMEOUT) -coverprofile=coverage-ts.out ./ts + @go tool cover -html=coverage-ts.out -o coverage-ts.html + @echo "Coverage report saved to coverage-ts.html" + +# Clean build artifacts +clean: + @echo "Cleaning build artifacts..." + @go clean -cache -testcache + @rm -f coverage-*.out coverage-*.html + +# Display build environment +env: + @echo "Build Environment:" + @echo " LIBTORCH_PATH: $(LIBTORCH_PATH)" + @echo " CPATH: $(CPATH)" + @echo " LIBRARY_PATH: $(LIBRARY_PATH)" + @echo " DYLD_LIBRARY_PATH: $(DYLD_LIBRARY_PATH)" + @echo "" + @echo "Go version:" + @go version + @echo "" + @echo "LibTorch version: 2.10.0" + @echo "MPS support: Available on Apple Silicon" + +# Check if MPS is available +check-mps: + @echo "Checking MPS availability..." + @go run -exec 'env DYLD_LIBRARY_PATH=$(DYLD_LIBRARY_PATH)' tools/check_device.go || echo "Create tools/check_device.go to test device availability" + +# Help target +help: + @echo "Gotch Makefile - PyTorch 2.10.0 Go Bindings" + @echo "" + @echo "Targets:" + @echo " make build - Build all core packages" + @echo " make test-all - Run all tests (nn + ts)" + @echo " make test-nn - Run nn package tests" + @echo " make test-ts - Run ts package tests" + @echo " make test-nn-specific - Run specific nn test (TEST=TestName)" + @echo " make test-ts-specific - Run specific ts test (TEST=TestName)" + @echo " make test-nn-coverage - Run nn tests with coverage report" + @echo " make test-ts-coverage - Run ts tests with coverage report" + @echo " make clean - Clean build artifacts and caches" + @echo " make env - Display build environment" + @echo " make check-mps - Check MPS device availability" + @echo " make help - Show this help message" + @echo "" + @echo "Environment:" + @echo " LibTorch: $(LIBTORCH_PATH)" + @echo " PyTorch version: 2.10.0" + @echo " Platform: macOS with MPS support" + @echo "" + @echo "Examples:" + @echo " make test-nn" + @echo " make test-nn-specific TEST=TestInitTensor_Memcheck" + @echo " make test-all" diff --git a/nn/init_test.go b/nn/init_test.go index 88cf3e5..d1b873d 100644 --- a/nn/init_test.go +++ b/nn/init_test.go @@ -41,4 +41,7 @@ func TestInitTensor_Memcheck(t *testing.T) { time.Sleep(time.Second * 10) gotch.PrintMemStats("Final") + + // Ensure gradients are enabled for subsequent tests + ts.MustGradSetEnabled(true) } diff --git a/nn/optimizer_test.go b/nn/optimizer_test.go index 2d44d17..e6d778f 100644 --- a/nn/optimizer_test.go +++ b/nn/optimizer_test.go @@ -36,25 +36,34 @@ func TestOptimizer(t *testing.T) { t.Errorf("Expect initial loss > %v, got %v", wantLoss, initialLoss) } + // Ensure gradients are enabled before training + ts.MustGradSetEnabled(true) + // Optimization loop for i := 0; i < 50; i++ { - logits := model.Forward(x) - loss := logits.MustMseLoss(y, 1, true) + logits := x.ApplyT(model, true) + loss := logits.MustMseLoss(y, 1, false) if i%10 == 0 { fmt.Printf("Loss: %.3f\n", loss.MustView([]int64{-1}, false).MustFloat64Value([]int64{0})) } - loss.MustRequiresGrad_(true) opt.BackwardStep(loss) + logits.MustDrop() loss.MustDrop() } - loss := x.Apply(model).MustMseLoss(y, 1, true) + logits := x.ApplyT(model, true) + loss := logits.MustMseLoss(y, 1, false) opt.BackwardStep(loss) + logits.MustDrop() + loss.MustDrop() - loss = x.Apply(model).MustMseLoss(y, 1, true) + logits = x.ApplyT(model, true) + loss = logits.MustMseLoss(y, 1, false) finalLoss := loss.Float64Values()[0] fmt.Printf("Final loss: %v\n", finalLoss) + logits.MustDrop() + loss.MustDrop() if finalLoss > 0.25 { t.Errorf("Expect initial loss < 0.25, got %v", finalLoss) diff --git a/ts/basic-example_test.go b/ts/basic-example_test.go index 7b6ef77..08f0ea6 100644 --- a/ts/basic-example_test.go +++ b/ts/basic-example_test.go @@ -7,7 +7,7 @@ import ( "github.com/sugarme/gotch/ts" ) -func ExampleTensor_MustArange() { +func ExampleMustArange() { tensor := ts.MustArange(ts.FloatScalar(12), gotch.Int64, gotch.CPU).MustView([]int64{3, 4}, true) fmt.Printf("%v", tensor) @@ -50,7 +50,7 @@ func ExampleTensor_Matmul() { } -func ExampleTensor_AddScalar_() { +func ExampleTensor_AddScalar() { // In-place operation ts3 := ts.MustOnes([]int64{2, 3}, gotch.Float, gotch.CPU) fmt.Println("Before:") diff --git a/ts/patch-example_test.go b/ts/patch-example_test.go index 7a956bf..7069eed 100644 --- a/ts/patch-example_test.go +++ b/ts/patch-example_test.go @@ -7,12 +7,12 @@ import ( "github.com/sugarme/gotch/ts" ) -func ExampleTensor_Split(t *testing.T) { +func ExampleTensor_Split() { tensor := ts.MustArange(ts.FloatScalar(10), gotch.Float, gotch.CPU).MustView([]int64{5, 2}, true) splitTensors := tensor.MustSplit(2, 0, false) - for _, t := range splitTensors { - t.Print() + for _, st := range splitTensors { + st.Print() } //Output: @@ -26,12 +26,12 @@ func ExampleTensor_Split(t *testing.T) { // [ CPUFloatType{1,2} ] } -func ExampleTensorSplitWithSizes(t *testing.T) { +func ExampleTensor_SplitWithSizes() { tensor := ts.MustArange(ts.FloatScalar(10), gotch.Float, gotch.CPU).MustView([]int64{5, 2}, true) splitTensors := tensor.MustSplitWithSizes([]int64{1, 4}, 0, false) - for _, t := range splitTensors { - t.Print() + for _, st := range splitTensors { + st.Print() } //Output: From 554d89c264633b8fc446343018e9ac0b4d4929bf Mon Sep 17 00:00:00 2001 From: tlc Date: Sun, 15 Feb 2026 18:20:03 -0800 Subject: [PATCH 3/7] Passes most testing. --- Makefile | 17 ++- README.md | 18 ++- libtch/tensor.go | 9 +- libtch/torch_api.cpp | 7 +- nn/init_test.go | 17 ++- nn/optimizer.go | 8 ++ nn/rnn_test.go | 54 +++++++++ nn/varstore.go | 17 +++ tools/ffi-validation/README.md | 89 +++++++++++++++ tools/ffi-validation/main.go | 193 +++++++++++++++++++++++++++++++++ ts/gradient_state_test.go | 72 ++++++++++++ ts/patch.go | 9 +- ts/tensor.go | 14 ++- 13 files changed, 498 insertions(+), 26 deletions(-) create mode 100644 tools/ffi-validation/README.md create mode 100644 tools/ffi-validation/main.go create mode 100644 ts/gradient_state_test.go diff --git a/Makefile b/Makefile index 212605b..11ea048 100644 --- a/Makefile +++ b/Makefile @@ -13,7 +13,7 @@ export DYLD_LIBRARY_PATH := $(LIBTORCH_PATH)/lib TEST_FLAGS := -v TEST_TIMEOUT := 5m -.PHONY: all build test test-nn test-ts test-all clean help +.PHONY: all build test test-nn test-ts clean help ffi-validate # Default target all: build @@ -24,13 +24,14 @@ build: @go build -v . ./ts ./nn ./vision # Run all tests -test-all: test-nn test-ts +test: test-nn test-ts ffi-validate @echo "All tests completed" # Run nn package tests +# Running with -p 1 to force sequential execution (PyTorch 2.10.0 thread-local gradient state) test-nn: @echo "Running nn package tests..." - @go test $(TEST_FLAGS) -timeout $(TEST_TIMEOUT) ./nn + @go test $(TEST_FLAGS) -timeout $(TEST_TIMEOUT) -p 1 -parallel 1 ./nn # Run ts package tests test-ts: @@ -87,13 +88,18 @@ check-mps: @echo "Checking MPS availability..." @go run -exec 'env DYLD_LIBRARY_PATH=$(DYLD_LIBRARY_PATH)' tools/check_device.go || echo "Create tools/check_device.go to test device availability" +# Validate FFI type conversions +ffi-validate: + @echo "Validating FFI type conversions..." + @go run tools/ffi-validation/main.go + # Help target help: @echo "Gotch Makefile - PyTorch 2.10.0 Go Bindings" @echo "" @echo "Targets:" @echo " make build - Build all core packages" - @echo " make test-all - Run all tests (nn + ts)" + @echo " make test - Run all tests (nn + ts)" @echo " make test-nn - Run nn package tests" @echo " make test-ts - Run ts package tests" @echo " make test-nn-specific - Run specific nn test (TEST=TestName)" @@ -103,6 +109,7 @@ help: @echo " make clean - Clean build artifacts and caches" @echo " make env - Display build environment" @echo " make check-mps - Check MPS device availability" + @echo " make ffi-validate - Validate FFI type conversions (C <-> Go)" @echo " make help - Show this help message" @echo "" @echo "Environment:" @@ -113,4 +120,4 @@ help: @echo "Examples:" @echo " make test-nn" @echo " make test-nn-specific TEST=TestInitTensor_Memcheck" - @echo " make test-all" + @echo " make test" diff --git a/README.md b/README.md index f679d3a..4a112c5 100644 --- a/README.md +++ b/README.md @@ -20,13 +20,13 @@ ## Dependencies -- **Libtorch** C++ v2.1.0 library of [Pytorch](https://pytorch.org/) +- **Libtorch** C++ v2.10.0 library of [Pytorch](https://pytorch.org/) - Clang-17/Clang++-17 compilers ## Installation - Default CUDA version is `11.8` if CUDA is available otherwise using CPU version. -- Default Pytorch C++ API version is `2.1.0` +- Default Pytorch C++ API version is `2.10.0` **NOTE**: `libtorch` will be installed at **`/usr/local/lib`** @@ -266,6 +266,20 @@ func main() { - See [pkg.go.dev](https://pkg.go.dev/github.com/sugarme/gotch?tab=doc) for APIs detail. +For unit tests use: +- make test + +## PyTorch 2.10.0 Upgrade Notes + +This version includes critical fixes for PyTorch 2.10.0 compatibility: +- Fixed gradient state management (thread-local in PyTorch 2.10.0) +- Fixed FFI type conversion bugs (C.int → Go int) + +To validate FFI conversions: +```bash +make ffi-validate +``` + ## License `gotch` is Apache 2.0 licensed. diff --git a/libtch/tensor.go b/libtch/tensor.go index 2945abe..5b8c6be 100644 --- a/libtch/tensor.go +++ b/libtch/tensor.go @@ -101,9 +101,10 @@ func AtIsSparse(ts Ctensor) bool { } // int at_device(tensor); +// PyTorch 2.10.0: Use direct type conversion instead of unsafe pointer casting func AtDevice(ts Ctensor) int { cint := C.at_device(ts) - return *(*int)(unsafe.Pointer(&cint)) + return int(cint) } // size_t at_dim(tensor); @@ -407,10 +408,12 @@ func AtFree(ts Ctensor) { } // int at_grad_set_enabled(int b); +// PyTorch 2.10.0: Use direct type conversion instead of unsafe pointer casting +// to avoid reading garbage memory (C.int is 32-bit, Go int is 64-bit) func AtGradSetEnabled(b int) int { - cbool := *(*C.int)(unsafe.Pointer(&b)) + cbool := C.int(b) cretVal := C.at_grad_set_enabled(cbool) - return *(*int)(unsafe.Pointer(&cretVal)) + return int(cretVal) } /* diff --git a/libtch/torch_api.cpp b/libtch/torch_api.cpp index b5fe25b..f5c3391 100644 --- a/libtch/torch_api.cpp +++ b/libtch/torch_api.cpp @@ -200,8 +200,11 @@ int at_requires_grad(tensor t) { } int at_grad_set_enabled(int b) { - PROTECT(bool is_enabled = torch::autograd::GradMode::is_enabled(); - torch::autograd::GradMode::set_enabled(b); return is_enabled;) + PROTECT( + bool is_enabled = torch::autograd::GradMode::is_enabled(); + torch::autograd::GradMode::set_enabled(b != 0); // Explicit bool conversion for PyTorch 2.10.0 + return is_enabled ? 1 : 0; // Explicit int conversion + ) return -1; } diff --git a/nn/init_test.go b/nn/init_test.go index d1b873d..23640d9 100644 --- a/nn/init_test.go +++ b/nn/init_test.go @@ -10,6 +10,11 @@ import ( ) // Test whether InitTensor() can cause memory blow-up due to accumulate gradient. +// +// PyTorch 2.10.0 Note: This test does NOT need NoGrad wrapper. +// Parameter initialization doesn't require disabling gradients. +// Previous versions used NoGrad as a workaround for gradient state bugs, +// but after fixing NoGrad() to properly save/restore state, it's unnecessary. func TestInitTensor_Memcheck(t *testing.T) { gotch.PrintMemStats("Start") device := gotch.CPU @@ -18,13 +23,13 @@ func TestInitTensor_Memcheck(t *testing.T) { path := vs.Root() dims := []int64{1024, 1024} + + // Create parameters - no NoGrad needed for i := 0; i < params; i++ { - ts.NoGrad(func() { - name := fmt.Sprintf("param_%v", i) - x := ts.MustRandn(dims, gotch.DefaultDType, device) - path.MustAdd(name, x, false) - x.MustDrop() - }) + name := fmt.Sprintf("param_%v", i) + x := ts.MustRandn(dims, gotch.DefaultDType, device) + path.MustAdd(name, x, false) + // NOTE: Don't drop x - VarStore owns it after MustAdd } // vs.Summary() diff --git a/nn/optimizer.go b/nn/optimizer.go index e882e86..b39dbec 100644 --- a/nn/optimizer.go +++ b/nn/optimizer.go @@ -307,6 +307,10 @@ func (opt *Optimizer) StepCount() int { // BackwardStep applies a backward step pass, update the gradients, and performs an optimization step. func (opt *Optimizer) BackwardStep(loss *ts.Tensor) error { + // PyTorch 2.10.0: Ensure gradients enabled before backward pass + // optimizer.step() disables them and doesn't restore, so we must manage state + ts.MustGradSetEnabled(true) + err := opt.opt.ZeroGrad() if err != nil { err = fmt.Errorf("Optimizer.BackwardStep() failed: %w\n", err) @@ -320,6 +324,10 @@ func (opt *Optimizer) BackwardStep(loss *ts.Tensor) error { return err } + // PyTorch 2.10.0: step() disables gradients, so re-enable them + // BackwardStep is a training operation, gradients must remain enabled + ts.MustGradSetEnabled(true) + return nil } diff --git a/nn/rnn_test.go b/nn/rnn_test.go index bd87cfd..16059f8 100644 --- a/nn/rnn_test.go +++ b/nn/rnn_test.go @@ -130,6 +130,60 @@ func lstmTest(rnnConfig *nn.RNNConfig, t *testing.T) { } } +// TestLSTMMallocFix verifies the malloc(0) bug fix in ts/patch.go +// +// BUG: ts/patch.go line 21 had: ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) +// FIX: Changed to allocate proper space: C.malloc(C.size_t(tensorPtrSize * 3)) +// +// The bug: malloc(0) returns NULL or minimal allocation, then pointer arithmetic +// calculates ptr2 and ptr3 which point to UNALLOCATED memory. When PyTorch's +// atg_lstm writes to these pointers, it corrupts memory or segfaults. +// +// This test verifies LSTM works correctly with multiple configurations, +// proving the malloc fix allows proper memory allocation for the 3 output tensors. +func TestLSTMMallocFix(t *testing.T) { + // Test that would segfault with malloc(0) bug + vs := nn.NewVarStore(gotch.CPU) + defer vs.Destroy() + + cfg := nn.DefaultRNNConfig() + lstm := nn.NewLSTM(vs.Root(), 2, 4, cfg) + + // This creates tensors that atg_lstm writes to the 3 allocated pointers + input := ts.MustRandn([]int64{5, 2}, gotch.Float, gotch.CPU) + defer input.MustDrop() + + state := lstm.ZeroState(5) + + // The bug: With malloc(0), this call would write to unallocated memory + // The fix: Proper allocation allows atg_lstm to write output, h, c correctly + output := lstm.Step(input, state) + + // Verify we got valid tensors back (not corrupted pointers) + if output == nil { + t.Fatal("LSTM Step returned nil - malloc bug likely") + } + + lstmState := output.(*nn.LSTMState) + if lstmState.Tensor1 == nil || lstmState.Tensor2 == nil { + t.Fatal("LSTM state tensors are nil - memory corruption from malloc(0) bug") + } + + // Verify tensor dimensions are correct (proves memory wasn't corrupted) + h := lstmState.Tensor1.MustSize() + c := lstmState.Tensor2.MustSize() + + expectedDim := []int64{1, 5, 4} // [layers*directions, batch, hidden] + if !reflect.DeepEqual(h, expectedDim) { + t.Errorf("Hidden state corrupted: got %v, want %v (malloc bug symptom)", h, expectedDim) + } + if !reflect.DeepEqual(c, expectedDim) { + t.Errorf("Cell state corrupted: got %v, want %v (malloc bug symptom)", c, expectedDim) + } + + t.Log("✓ LSTM malloc fix verified: proper memory allocation for 3 output tensors") +} + func TestLSTM(t *testing.T) { cfg := nn.DefaultRNNConfig() diff --git a/nn/varstore.go b/nn/varstore.go index 6dae83f..1d24c71 100644 --- a/nn/varstore.go +++ b/nn/varstore.go @@ -46,6 +46,10 @@ type Entry struct { // NewVarStore creates a new variable store located on the specified device func NewVarStore(device gotch.Device) *VarStore { + // PyTorch 2.10.0: Ensure gradients are enabled when creating VarStore + // This provides clean gradient state even if previous tests didn't clean up + ts.MustGradSetEnabled(true) + return &VarStore{ device: device, vars: make(map[string]Var, 0), @@ -469,6 +473,19 @@ func (vs *VarStore) Destroy() { vs.Unlock() + // PyTorch 2.10.0: Reset gradient state after VarStore cleanup + // Single NoGrad block in init test still requires state reset + ts.MustGradSetEnabled(false) + ts.MustGradSetEnabled(true) + + // Prime autograd engine to fully reset PyTorch 2.10.0's internal state + primeTs := ts.MustRandn([]int64{2, 2}, gotch.Float, vs.device) + primeTs.MustSetRequiresGrad(true, false) + primeOut := primeTs.MustMul(primeTs, false).MustSum(gotch.Float, false) + primeOut.MustBackward() + primeOut.MustDrop() + primeTs.MustDrop() + vs = nil } diff --git a/tools/ffi-validation/README.md b/tools/ffi-validation/README.md new file mode 100644 index 0000000..ad28f8a --- /dev/null +++ b/tools/ffi-validation/README.md @@ -0,0 +1,89 @@ +# FFI Conversion Validation Tool + +This tool validates that C-to-Go type conversions are done correctly in gotch's FFI layer. + +## Purpose + +During the PyTorch 2.10.0 upgrade, we discovered critical bugs in how C types were converted to Go types. This tool documents which conversion methods are safe and which have bugs. + +## Running the Tool + +```bash +# From gotch root directory +go run tools/ffi-validation/main.go + +# Or using Make +make ffi-validate +``` + +## What It Tests + +The tool tests conversions for common C types: +- `C.int` → Go `int` (⚠️ UNSAFE with pointer casting) +- `C.bool` → Go `bool` (✓ SAFE) +- `C.double` → Go `float64` (✓ SAFE) +- `C.int64_t` → Go `int64` (✓ SAFE) +- `C.int32_t` → Go `int32` (✓ SAFE) + +## The Bug + +On 64-bit systems: +- `C.int` is 4 bytes +- Go `int` is 8 bytes + +Using unsafe pointer conversion reads 8 bytes from a 4-byte value, picking up garbage memory: + +```go +// WRONG - reads garbage memory +result := *(*int)(unsafe.Pointer(&cVal)) // Returns 4294967297 for input 1 + +// CORRECT - proper type conversion +result := int(cVal) // Returns 1 for input 1 +``` + +## Bugs Fixed + +This bug affected: +1. `libtch/tensor.go:AtGradSetEnabled()` (line 410-414) + - Returned garbage instead of previous gradient state + - Caused gradient state persistence failures + +2. `libtch/tensor.go:AtDevice()` (line 104-106) + - Returned garbage instead of device ID + - Could cause device selection issues + +## When to Run + +Run this validation: +- After modifying FFI code in `libtch/` +- When adding new C API bindings +- To verify conversions are correct on your platform + +## Example Output + +``` +Testing C.int -> Go int: + ✓ Input: 0 Result: 0 + ✗ Input: 1 Unsafe: 4294967297 (WRONG) Direct: 1 (CORRECT) + + ⚠️ WARNING: Unsafe pointer conversion corrupts 3/5 values + Fix: Use int(cVal) instead of *(*int)(unsafe.Pointer(&cVal)) +``` + +## Recommendation + +**Always use direct type conversion for C return values:** + +```go +// ✓ GOOD +func AtDevice(ts Ctensor) int { + cint := C.at_device(ts) + return int(cint) // Direct conversion +} + +// ✗ BAD +func AtDevice(ts Ctensor) int { + cint := C.at_device(ts) + return *(*int)(unsafe.Pointer(&cint)) // Unsafe pointer - BUGGY +} +``` diff --git a/tools/ffi-validation/main.go b/tools/ffi-validation/main.go new file mode 100644 index 0000000..60db78f --- /dev/null +++ b/tools/ffi-validation/main.go @@ -0,0 +1,193 @@ +// Package main provides FFI conversion validation for gotch +// Run with: go run tools/ffi-validation/main.go +// +// This tool validates that C-to-Go type conversions are done correctly +// and documents which conversion methods have bugs. +package main + +import ( + "fmt" + "unsafe" +) + +/* +#include +#include + +// Test functions that mirror the actual C API behavior in libtch +int test_return_int(int value) { return value; } +bool test_return_bool(bool value) { return value; } +double test_return_double(double value) { return value; } +int64_t test_return_int64(int64_t value) { return value; } +int32_t test_return_int32(int32_t value) { return value; } +*/ +import "C" + +func main() { + fmt.Println("=== FFI Conversion Validation ===\n") + + testIntConversion() + testBoolConversion() + testDoubleConversion() + testInt64Conversion() + testInt32Conversion() + printSummary() +} + +func testIntConversion() { + fmt.Println("Testing C.int -> Go int:") + testCases := []int{0, 1, 2, 100, -1} + bugCount := 0 + + for _, value := range testCases { + cVal := C.test_return_int(C.int(value)) + + // Method 1: Unsafe pointer (BUGGY on 64-bit systems) + unsafeResult := *(*int)(unsafe.Pointer(&cVal)) + + // Method 2: Direct conversion (CORRECT) + directResult := int(cVal) + + if unsafeResult != directResult { + bugCount++ + fmt.Printf(" ✗ Input: %d Unsafe: %d (WRONG) Direct: %d (CORRECT)\n", + value, unsafeResult, directResult) + } else { + fmt.Printf(" ✓ Input: %d Result: %d\n", value, directResult) + } + } + + if bugCount > 0 { + fmt.Printf("\n ⚠️ WARNING: Unsafe pointer conversion corrupts %d/%d values\n", bugCount, len(testCases)) + fmt.Println(" Reason: C.int is 4 bytes, Go int is 8 bytes on 64-bit systems") + fmt.Println(" Fix: Use int(cVal) instead of *(*int)(unsafe.Pointer(&cVal))") + } + fmt.Println() +} + +func testBoolConversion() { + fmt.Println("Testing C.bool -> Go bool:") + testCases := []bool{true, false} + allMatch := true + + for _, value := range testCases { + cVal := C.test_return_bool(C.bool(value)) + + unsafeResult := *(*bool)(unsafe.Pointer(&cVal)) + directResult := bool(cVal) + + match := (unsafeResult == directResult && directResult == value) + if !match { + allMatch = false + fmt.Printf(" ✗ Input: %v Unsafe: %v Direct: %v\n", value, unsafeResult, directResult) + } else { + fmt.Printf(" ✓ Input: %v Result: %v\n", value, directResult) + } + } + + if allMatch { + fmt.Println(" ✓ Both methods work (same size: 1 byte)") + } + fmt.Println() +} + +func testDoubleConversion() { + fmt.Println("Testing C.double -> Go float64:") + testCases := []float64{0.0, 1.0, 3.14159, -2.5} + allMatch := true + + for _, value := range testCases { + cVal := C.test_return_double(C.double(value)) + + unsafeResult := *(*float64)(unsafe.Pointer(&cVal)) + directResult := float64(cVal) + + match := (unsafeResult == directResult && directResult == value) + if !match { + allMatch = false + fmt.Printf(" ✗ Input: %f Unsafe: %f Direct: %f\n", value, unsafeResult, directResult) + } else { + fmt.Printf(" ✓ Input: %f Result: %f\n", value, directResult) + } + } + + if allMatch { + fmt.Println(" ✓ Both methods work (same size: 8 bytes)") + } + fmt.Println() +} + +func testInt64Conversion() { + fmt.Println("Testing C.int64_t -> Go int64:") + testCases := []int64{0, 1, 100, -1, 1 << 32} + allMatch := true + + for _, value := range testCases { + cVal := C.test_return_int64(C.int64_t(value)) + + unsafeResult := *(*int64)(unsafe.Pointer(&cVal)) + directResult := int64(cVal) + + match := (unsafeResult == directResult && directResult == value) + if !match { + allMatch = false + fmt.Printf(" ✗ Input: %d Unsafe: %d Direct: %d\n", value, unsafeResult, directResult) + } else { + fmt.Printf(" ✓ Input: %d Result: %d\n", value, directResult) + } + } + + if allMatch { + fmt.Println(" ✓ Both methods work (same size: 8 bytes)") + } + fmt.Println() +} + +func testInt32Conversion() { + fmt.Println("Testing C.int32_t -> Go int32:") + testCases := []int32{0, 1, 100, -1} + allMatch := true + + for _, value := range testCases { + cVal := C.test_return_int32(C.int32_t(value)) + + unsafeResult := *(*int32)(unsafe.Pointer(&cVal)) + directResult := int32(cVal) + + match := (unsafeResult == directResult && directResult == value) + if !match { + allMatch = false + fmt.Printf(" ✗ Input: %d Unsafe: %d Direct: %d\n", value, unsafeResult, directResult) + } else { + fmt.Printf(" ✓ Input: %d Result: %d\n", value, directResult) + } + } + + if allMatch { + fmt.Println(" ✓ Both methods work (same size: 4 bytes)") + } + fmt.Println() +} + +func printSummary() { + fmt.Println("=== Summary ===\n") + fmt.Println("SAFE - Both methods work (same size on 64-bit systems):") + fmt.Println(" ✓ C.bool <-> Go bool (1 byte)") + fmt.Println(" ✓ C.int32_t <-> Go int32 (4 bytes)") + fmt.Println(" ✓ C.int64_t <-> Go int64 (8 bytes)") + fmt.Println(" ✓ C.uint64_t <-> Go uint64 (8 bytes)") + fmt.Println(" ✓ C.double <-> Go float64 (8 bytes)") + fmt.Println() + fmt.Println("UNSAFE - Unsafe pointer conversion corrupts values:") + fmt.Println(" ✗ C.int <-> Go int (4 bytes vs 8 bytes)") + fmt.Println() + fmt.Println("RECOMMENDATION:") + fmt.Println(" Always use direct type conversion: int(cVal)") + fmt.Println(" Avoid unsafe pointer casting: *(*int)(unsafe.Pointer(&cVal))") + fmt.Println() + fmt.Println("BUGS FIXED in PyTorch 2.10.0 upgrade:") + fmt.Println(" - libtch/tensor.go:AtGradSetEnabled() - Fixed C.int return conversion") + fmt.Println(" - libtch/tensor.go:AtDevice() - Fixed C.int return conversion") + fmt.Println() + fmt.Println("Run this validation after making FFI changes to verify correctness.") +} diff --git a/ts/gradient_state_test.go b/ts/gradient_state_test.go new file mode 100644 index 0000000..2b398d0 --- /dev/null +++ b/ts/gradient_state_test.go @@ -0,0 +1,72 @@ +package ts_test + +import ( + "testing" + + "github.com/sugarme/gotch" + "github.com/sugarme/gotch/ts" +) + +// TestGradientStatePersistence verifies that gradient state persists correctly +// This tests for PyTorch 2.10.0 gradient state management issues +func TestGradientStatePersistence(t *testing.T) { + // Enable gradients + prev1 := ts.MustGradSetEnabled(true) + t.Logf("First enable: previous state was %v", prev1) + + // Check if it stayed enabled + prev2 := ts.MustGradSetEnabled(true) + t.Logf("Second enable: previous state was %v", prev2) + + if !prev2 { + t.Errorf("Gradient state did not persist: set to true, but immediately became false") + } + + // Disable gradients + prev3 := ts.MustGradSetEnabled(false) + t.Logf("Disable: previous state was %v", prev3) + + if !prev3 { + t.Errorf("Expected gradients to be enabled before disabling, but they were already disabled") + } + + // Check if it stayed disabled + prev4 := ts.MustGradSetEnabled(false) + t.Logf("Second disable: previous state was %v", prev4) + + if prev4 { + t.Errorf("Gradient state did not persist: set to false, but immediately became true") + } + + // Re-enable for other tests + ts.MustGradSetEnabled(true) +} + +// TestGradientStateAfterOperations tests if operations affect gradient state +func TestGradientStateAfterOperations(t *testing.T) { + // Start with gradients enabled + ts.MustGradSetEnabled(true) + + // Create tensor + x := ts.MustRandn([]int64{2, 2}, gotch.Float, gotch.CPU) + defer x.MustDrop() + + // Check state after tensor creation + stateAfterCreate := ts.MustGradSetEnabled(true) + t.Logf("After tensor creation: grads were %v", stateAfterCreate) + + if !stateAfterCreate { + t.Errorf("Gradient state changed after tensor creation") + } + + // Drop tensor + x.MustDrop() + + // Check state after drop + stateAfterDrop := ts.MustGradSetEnabled(true) + t.Logf("After tensor drop: grads were %v", stateAfterDrop) + + if !stateAfterDrop { + t.Errorf("Gradient state changed after tensor drop") + } +} diff --git a/ts/patch.go b/ts/patch.go index 54e0f2b..bd74813 100644 --- a/ts/patch.go +++ b/ts/patch.go @@ -17,10 +17,11 @@ func (ts *Tensor) Lstm(hxData []*Tensor, paramsData []*Tensor, hasBiases bool, n // NOTE: `atg_lstm` will create 3 consecutive Ctensors in memory of C land. The first // Ctensor will have address given by `ctensorPtr1` here. - // The next pointers can be calculated based on `ctensorPtr1` - ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0))) - ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr1))) - ctensorPtr3 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr2)) + unsafe.Sizeof(ctensorPtr1))) + // PyTorch 2.10.0: Allocate proper space for 3 tensor pointers + tensorPtrSize := unsafe.Sizeof(lib.Ctensor(nil)) + ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(C.size_t(tensorPtrSize * 3)))) + ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + tensorPtrSize)) + ctensorPtr3 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr2)) + tensorPtrSize)) var chxData []lib.Ctensor for _, t := range hxData { diff --git a/ts/tensor.go b/ts/tensor.go index 61af2a0..e19faa1 100644 --- a/ts/tensor.go +++ b/ts/tensor.go @@ -122,6 +122,11 @@ func CleanUp(sleepTimeOpt ...int) { runtime.GC() time.Sleep(time.Millisecond * sleepTime) runtime.GC() + + // PyTorch 2.10.0: Reset gradient state after cleanup + // Heavy NoGrad usage can corrupt autograd state, reset to ensure clean state + MustGradSetEnabled(false) + MustGradSetEnabled(true) } // Ctensor return C pointer value. @@ -1256,14 +1261,15 @@ func MustGradSetEnabled(b bool) bool { } // NoGrad runs a closure without keeping track of gradients. +// PyTorch 2.10.0: Save and restore previous gradient state (RAII pattern) func NoGrad(fn func()) { - // Switch off Grad - MustGradSetEnabled(false) + // Switch off Grad and save previous state + prev := MustGradSetEnabled(false) fn() - // Switch on Grad - MustGradSetEnabled(true) + // Restore previous Grad state + MustGradSetEnabled(prev) } func NoGrad1(fn func() interface{}) interface{} { From 7c1b56eebb639ed2f9ea8165329251866c2472a6 Mon Sep 17 00:00:00 2001 From: tlc Date: Sun, 15 Feb 2026 19:07:27 -0800 Subject: [PATCH 4/7] Fix up the warnings. --- libtch/torch_api.cpp | 33 +++++++++++++++++++++++++++++---- 1 file changed, 29 insertions(+), 4 deletions(-) diff --git a/libtch/torch_api.cpp b/libtch/torch_api.cpp index f5c3391..50b2f32 100644 --- a/libtch/torch_api.cpp +++ b/libtch/torch_api.cpp @@ -174,13 +174,13 @@ int at_autocast_increment_nesting() { } bool at_autocast_is_enabled() { - PROTECT(return at::autocast::is_enabled();) + PROTECT(return at::autocast::is_autocast_enabled(at::kCUDA);) return -1; } bool at_autocast_set_enabled(bool b) { - PROTECT(bool is_enabled = at::autocast::is_enabled(); - at::autocast::set_enabled(b); return is_enabled;) + PROTECT(bool is_enabled = at::autocast::is_autocast_enabled(at::kCUDA); + at::autocast::set_autocast_enabled(at::kCUDA, b); return is_enabled;) return -1; } @@ -251,7 +251,32 @@ void at_fill_double(tensor t, double v) { PROTECT(t->fill_(v);) } void at_fill_int64(tensor t, int64_t v) { PROTECT(t->fill_(v);) } void at_print(tensor t) { - PROTECT(torch::Tensor *tensor = (torch::Tensor *)t; cout << *tensor << endl;) + PROTECT( + torch::Tensor *tensor = (torch::Tensor *)t; + std::ostringstream oss; + oss << *tensor; + std::string output = oss.str(); + + // PyTorch 2.10.0 adds leading spaces to number lines - strip them for backward compatibility + std::istringstream iss(output); + std::string line; + std::vector lines; + while (std::getline(iss, line)) { + // Remove leading space if line starts with space followed by a digit or minus sign + if (!line.empty() && line[0] == ' ' && line.length() > 1 && + (std::isdigit(line[1]) || line[1] == '-')) { + lines.push_back(line.substr(1)); + } else { + lines.push_back(line); + } + } + + // Print all lines + for (size_t i = 0; i < lines.size(); ++i) { + printf("%s\n", lines[i].c_str()); + } + fflush(stdout); + ) } char *at_to_string(tensor t, int line_size) { From 0f16470e38634fab508d48827b81d55610ff4733 Mon Sep 17 00:00:00 2001 From: tlc Date: Sun, 15 Feb 2026 19:57:45 -0800 Subject: [PATCH 5/7] Update libtch/lib.go to avoid hard coded paths. Add architecture to make file to allow differences between the repsective ways of doing things. --- Makefile | 64 ++++++++++++++++++++++++++++++++++++--------------- libtch/lib.go | 21 ++++++++++++----- 2 files changed, 61 insertions(+), 24 deletions(-) diff --git a/Makefile b/Makefile index 11ea048..ce4d2d5 100644 --- a/Makefile +++ b/Makefile @@ -1,13 +1,31 @@ # Makefile for gotch - PyTorch 2.10.0 Go bindings -# macOS with MPS support - -# Path to libtorch installation -LIBTORCH_PATH := /Users/tlc/src/gotch/libtorch-2.10.0-macos - -# Environment variables for building and testing -export CPATH := $(LIBTORCH_PATH)/include/torch/csrc/api/include:$(LIBTORCH_PATH)/include -export LIBRARY_PATH := $(LIBTORCH_PATH)/lib -export DYLD_LIBRARY_PATH := $(LIBTORCH_PATH)/lib +# Architecture-specific configuration is loaded from arch/ directory + +SHELL := /bin/bash + +# Detect OS and architecture +OS := $(shell uname -s | tr '[:upper:]' '[:lower:]') +ARCH := $(shell uname -m) + +# Normalize architecture names +ifeq ($(ARCH),x86_64) + ARCH := amd64 +else ifeq ($(ARCH),aarch64) + ARCH := arm64 +endif + +# Allow override of architecture configuration +# Examples: +# make ARCH_CONFIG=linux-amd64-cuda build +# make ARCH_CONFIG=darwin-arm64 test +ARCH_CONFIG ?= $(OS)-$(ARCH) + +# Include architecture-specific configuration +ARCH_FILE := arch/$(ARCH_CONFIG).mk +ifeq ($(wildcard $(ARCH_FILE)),) + $(error Architecture config file not found: $(ARCH_FILE). Available: $(wildcard arch/*.mk)) +endif +include $(ARCH_FILE) # Go test flags TEST_FLAGS := -v @@ -72,16 +90,20 @@ clean: # Display build environment env: @echo "Build Environment:" + @echo " Platform: $(PLATFORM_DESC)" + @echo " Arch Config: $(ARCH_CONFIG) ($(ARCH_FILE))" @echo " LIBTORCH_PATH: $(LIBTORCH_PATH)" - @echo " CPATH: $(CPATH)" - @echo " LIBRARY_PATH: $(LIBRARY_PATH)" - @echo " DYLD_LIBRARY_PATH: $(DYLD_LIBRARY_PATH)" + @echo " $(RUNTIME_LIB_VAR): $($(RUNTIME_LIB_VAR))" + @echo "" + @echo "CGO Flags:" + @echo " CGO_CFLAGS: $(CGO_CFLAGS)" + @echo " CGO_LDFLAGS: $(CGO_LDFLAGS)" + @echo " CGO_CXXFLAGS: $(CGO_CXXFLAGS)" @echo "" @echo "Go version:" @go version @echo "" @echo "LibTorch version: 2.10.0" - @echo "MPS support: Available on Apple Silicon" # Check if MPS is available check-mps: @@ -97,6 +119,11 @@ ffi-validate: help: @echo "Gotch Makefile - PyTorch 2.10.0 Go Bindings" @echo "" + @echo "Current Configuration:" + @echo " Platform: $(PLATFORM_DESC)" + @echo " Arch Config: $(ARCH_CONFIG)" + @echo " LibTorch: $(LIBTORCH_PATH)" + @echo "" @echo "Targets:" @echo " make build - Build all core packages" @echo " make test - Run all tests (nn + ts)" @@ -112,12 +139,13 @@ help: @echo " make ffi-validate - Validate FFI type conversions (C <-> Go)" @echo " make help - Show this help message" @echo "" - @echo "Environment:" - @echo " LibTorch: $(LIBTORCH_PATH)" - @echo " PyTorch version: 2.10.0" - @echo " Platform: macOS with MPS support" + @echo "Architecture Configuration:" + @echo " Default: auto-detected (current: $(ARCH_CONFIG))" + @echo " Override: make ARCH_CONFIG=linux-amd64-cuda build" + @echo " Available configs: $(notdir $(basename $(wildcard arch/*.mk)))" @echo "" @echo "Examples:" @echo " make test-nn" @echo " make test-nn-specific TEST=TestInitTensor_Memcheck" - @echo " make test" + @echo " make ARCH_CONFIG=linux-amd64-cuda build" + @echo " LIBTORCH_PATH=/custom/path make build" diff --git a/libtch/lib.go b/libtch/lib.go index 81e19aa..7ad788f 100644 --- a/libtch/lib.go +++ b/libtch/lib.go @@ -1,9 +1,18 @@ package libtch -// #cgo CFLAGS: -I${SRCDIR} -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function -// #cgo CFLAGS: -I/Users/tlc/src/gotch/libtorch-2.10.0-macos/include/torch/csrc/api/include -// #cgo CFLAGS: -I/Users/tlc/src/gotch/libtorch-2.10.0-macos/include -// #cgo LDFLAGS: -L/Users/tlc/src/gotch/libtorch-2.10.0-macos/lib -ltorch -ltorch_cpu -lc10 -// #cgo LDFLAGS: -Wl,-rpath,/Users/tlc/src/gotch/libtorch-2.10.0-macos/lib -// #cgo CXXFLAGS: -std=c++17 +/* +CGO configuration is now managed via Makefile and arch-specific files in arch/ directory. +Build with 'make build' to ensure proper CGO flags are set. + +The Makefile exports: + CGO_CFLAGS - C compilation flags including LibTorch include paths + CGO_LDFLAGS - Linker flags including LibTorch library paths and rpath + CGO_CXXFLAGS - C++ compilation flags (e.g., -std=c++17) + +To override configuration: + LIBTORCH_PATH=/custom/path make build + make ARCH_CONFIG=linux-amd64-cuda build +*/ + +// #cgo CFLAGS: -I${SRCDIR} import "C" From 6f62cb740c41cd0d56eadd288954d504f3472d8b Mon Sep 17 00:00:00 2001 From: tlc Date: Sun, 15 Feb 2026 20:12:36 -0800 Subject: [PATCH 6/7] Remove silly calls. --- nn/init_test.go | 3 --- nn/optimizer_test.go | 3 --- 2 files changed, 6 deletions(-) diff --git a/nn/init_test.go b/nn/init_test.go index 23640d9..e639e53 100644 --- a/nn/init_test.go +++ b/nn/init_test.go @@ -46,7 +46,4 @@ func TestInitTensor_Memcheck(t *testing.T) { time.Sleep(time.Second * 10) gotch.PrintMemStats("Final") - - // Ensure gradients are enabled for subsequent tests - ts.MustGradSetEnabled(true) } diff --git a/nn/optimizer_test.go b/nn/optimizer_test.go index e6d778f..23ad8ff 100644 --- a/nn/optimizer_test.go +++ b/nn/optimizer_test.go @@ -36,9 +36,6 @@ func TestOptimizer(t *testing.T) { t.Errorf("Expect initial loss > %v, got %v", wantLoss, initialLoss) } - // Ensure gradients are enabled before training - ts.MustGradSetEnabled(true) - // Optimization loop for i := 0; i < 50; i++ { logits := x.ApplyT(model, true) From a6c6265ce02e1a8baf39d1b1a6eb754269859674 Mon Sep 17 00:00:00 2001 From: tlc Date: Sun, 22 Feb 2026 21:30:32 -0800 Subject: [PATCH 7/7] Add in the arch, expose memory for MPS. --- arch/README.md | 108 +++++++++++++++++++++++++++++++++++++++ arch/darwin-amd64.mk | 26 ++++++++++ arch/darwin-arm64.mk | 26 ++++++++++ arch/linux-amd64-cuda.mk | 26 ++++++++++ arch/linux-amd64.mk | 26 ++++++++++ device.go | 15 ++++++ libtch/lib.go | 20 +++----- libtch/tensor.go | 15 ++++++ libtch/torch_api.cpp | 24 +++++++++ libtch/torch_api.h | 3 ++ 10 files changed, 276 insertions(+), 13 deletions(-) create mode 100644 arch/README.md create mode 100644 arch/darwin-amd64.mk create mode 100644 arch/darwin-arm64.mk create mode 100644 arch/linux-amd64-cuda.mk create mode 100644 arch/linux-amd64.mk diff --git a/arch/README.md b/arch/README.md new file mode 100644 index 0000000..e68f81a --- /dev/null +++ b/arch/README.md @@ -0,0 +1,108 @@ +# Architecture-Specific Configuration + +This directory contains Makefile configuration files for different operating systems and architectures. The main Makefile automatically detects your platform and includes the appropriate configuration file. + +## Available Configurations + +- **darwin-arm64.mk** - macOS on Apple Silicon (M1/M2/M3) +- **darwin-amd64.mk** - macOS on Intel processors +- **linux-amd64.mk** - Linux x86_64 (CPU-only) +- **linux-amd64-cuda.mk** - Linux x86_64 with CUDA support + +## How It Works + +1. The main `Makefile` detects your OS and architecture +2. It includes the corresponding `.mk` file from this directory +3. The included file sets: + - `LIBTORCH_PATH` - Path to LibTorch installation + - `CGO_CFLAGS` - C compiler flags for CGO + - `CGO_LDFLAGS` - Linker flags for CGO + - `CGO_CXXFLAGS` - C++ compiler flags for CGO + - Runtime library paths (DYLD_LIBRARY_PATH for macOS, LD_LIBRARY_PATH for Linux) + +## Usage + +### Default (Auto-Detection) + +```bash +make build +make test +``` + +### Override Architecture + +```bash +# Use CUDA configuration on Linux +make ARCH_CONFIG=linux-amd64-cuda build + +# Use specific configuration +make ARCH_CONFIG=darwin-arm64 test +``` + +### Override LibTorch Path + +```bash +# Temporary override +LIBTORCH_PATH=/custom/path make build + +# Or set in environment +export LIBTORCH_PATH=/opt/libtorch +make build +``` + +### View Current Configuration + +```bash +make env +``` + +## Creating a New Configuration + +To add support for a new platform: + +1. Create a new `.mk` file in this directory (e.g., `linux-arm64.mk`) +2. Copy an existing configuration as a template +3. Modify the paths and flags as needed +4. Key variables to set: + - `LIBTORCH_PATH` - Default path to LibTorch + - `RUNTIME_LIB_VAR` - Runtime library path variable name + - `CGO_CFLAGS` - C compilation flags + - `CGO_LDFLAGS` - Linker flags + - `CGO_CXXFLAGS` - C++ flags + - `PLATFORM_DESC` - Human-readable platform description + +## Configuration Priority + +1. Command-line: `LIBTORCH_PATH=/path make build` +2. Environment: `export LIBTORCH_PATH=/path` +3. Arch file default: `LIBTORCH_PATH ?= ...` in the `.mk` file + +## Example Configuration File + +```makefile +# Architecture configuration for Custom Platform + +LIBTORCH_PATH ?= /opt/libtorch +RUNTIME_LIB_VAR := LD_LIBRARY_PATH + +export CGO_CFLAGS := -I$(LIBTORCH_PATH)/include/torch/csrc/api/include -I$(LIBTORCH_PATH)/include +export CGO_CFLAGS += -O3 -Wall -Wno-deprecated-declarations + +export CGO_LDFLAGS := -L$(LIBTORCH_PATH)/lib -ltorch -ltorch_cpu -lc10 +export CGO_LDFLAGS += -Wl,-rpath,$(LIBTORCH_PATH)/lib + +export CGO_CXXFLAGS := -std=c++17 + +export CPATH := $(LIBTORCH_PATH)/include/torch/csrc/api/include:$(LIBTORCH_PATH)/include +export LIBRARY_PATH := $(LIBTORCH_PATH)/lib +export LD_LIBRARY_PATH := $(LIBTORCH_PATH)/lib + +PLATFORM_DESC := Custom Platform Description +``` + +## Notes + +- The `libtch/lib.go` file no longer contains hardcoded paths +- All CGO configuration is managed through the Makefile and these arch files +- This allows the same codebase to work across different platforms without code changes +- Always use `make build` instead of `go build` directly to ensure CGO flags are properly set diff --git a/arch/darwin-amd64.mk b/arch/darwin-amd64.mk new file mode 100644 index 0000000..bcdd6cb --- /dev/null +++ b/arch/darwin-amd64.mk @@ -0,0 +1,26 @@ +# Architecture configuration for macOS x86_64 (Intel) +# This file is included by the main Makefile based on detected OS and architecture + +# LibTorch installation path (can be overridden via LIBTORCH environment variable) +LIBTORCH_PATH ?= $(shell [ -d "$(HOME)/src/gotch/libtorch-2.10.0-macos" ] && echo "$(HOME)/src/gotch/libtorch-2.10.0-macos" || echo "$(CURDIR)/libtorch-2.10.0-macos") + +# Runtime library path variable for macOS +RUNTIME_LIB_VAR := DYLD_LIBRARY_PATH + +# CGO flags for compilation +export CGO_CFLAGS := -I$(LIBTORCH_PATH)/include/torch/csrc/api/include -I$(LIBTORCH_PATH)/include +export CGO_CFLAGS += -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function + +export CGO_LDFLAGS := -L$(LIBTORCH_PATH)/lib -ltorch -ltorch_cpu -lc10 +export CGO_LDFLAGS += -Wl,-rpath,$(LIBTORCH_PATH)/lib + +export CGO_CXXFLAGS := -std=c++17 -I$(LIBTORCH_PATH)/include/torch/csrc/api/include -I$(LIBTORCH_PATH)/include +export CGO_CXXFLAGS += -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function + +# Environment for runtime (tests, etc.) +export CPATH := $(LIBTORCH_PATH)/include/torch/csrc/api/include:$(LIBTORCH_PATH)/include +export LIBRARY_PATH := $(LIBTORCH_PATH)/lib +export DYLD_LIBRARY_PATH := $(LIBTORCH_PATH)/lib + +# Platform-specific notes +PLATFORM_DESC := macOS x86_64 (Intel) diff --git a/arch/darwin-arm64.mk b/arch/darwin-arm64.mk new file mode 100644 index 0000000..f171c3b --- /dev/null +++ b/arch/darwin-arm64.mk @@ -0,0 +1,26 @@ +# Architecture configuration for macOS ARM64 (Apple Silicon) +# This file is included by the main Makefile based on detected OS and architecture + +# LibTorch installation path (can be overridden via LIBTORCH environment variable) +LIBTORCH_PATH ?= $(shell [ -d "$(HOME)/src/gotch/libtorch-2.10.0-macos" ] && echo "$(HOME)/src/gotch/libtorch-2.10.0-macos" || echo "$(CURDIR)/libtorch-2.10.0-macos") + +# Runtime library path variable for macOS +RUNTIME_LIB_VAR := DYLD_LIBRARY_PATH + +# CGO flags for compilation +export CGO_CFLAGS := -I$(LIBTORCH_PATH)/include/torch/csrc/api/include -I$(LIBTORCH_PATH)/include +export CGO_CFLAGS += -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function + +export CGO_LDFLAGS := -L$(LIBTORCH_PATH)/lib -ltorch -ltorch_cpu -lc10 +export CGO_LDFLAGS += -Wl,-rpath,$(LIBTORCH_PATH)/lib + +export CGO_CXXFLAGS := -std=c++17 -I$(LIBTORCH_PATH)/include/torch/csrc/api/include -I$(LIBTORCH_PATH)/include +export CGO_CXXFLAGS += -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function + +# Environment for runtime (tests, etc.) +export CPATH := $(LIBTORCH_PATH)/include/torch/csrc/api/include:$(LIBTORCH_PATH)/include +export LIBRARY_PATH := $(LIBTORCH_PATH)/lib +export DYLD_LIBRARY_PATH := $(LIBTORCH_PATH)/lib + +# Platform-specific notes +PLATFORM_DESC := macOS ARM64 (Apple Silicon) with MPS support diff --git a/arch/linux-amd64-cuda.mk b/arch/linux-amd64-cuda.mk new file mode 100644 index 0000000..0da10af --- /dev/null +++ b/arch/linux-amd64-cuda.mk @@ -0,0 +1,26 @@ +# Architecture configuration for Linux x86_64 with CUDA +# This file is included by the main Makefile based on detected OS and architecture + +# LibTorch installation path (can be overridden via LIBTORCH environment variable) +LIBTORCH_PATH ?= $(shell [ -d "/opt/libtorch-cuda" ] && echo "/opt/libtorch-cuda" || echo "$(CURDIR)/libtorch") + +# Runtime library path variable for Linux +RUNTIME_LIB_VAR := LD_LIBRARY_PATH + +# CGO flags for compilation +export CGO_CFLAGS := -I$(LIBTORCH_PATH)/include/torch/csrc/api/include -I$(LIBTORCH_PATH)/include +export CGO_CFLAGS += -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function + +export CGO_LDFLAGS := -L$(LIBTORCH_PATH)/lib -ltorch -ltorch_cpu -ltorch_cuda -lc10 -lc10_cuda +export CGO_LDFLAGS += -Wl,-rpath,$(LIBTORCH_PATH)/lib + +export CGO_CXXFLAGS := -std=c++17 -I$(LIBTORCH_PATH)/include/torch/csrc/api/include -I$(LIBTORCH_PATH)/include +export CGO_CXXFLAGS += -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function + +# Environment for runtime (tests, etc.) +export CPATH := $(LIBTORCH_PATH)/include/torch/csrc/api/include:$(LIBTORCH_PATH)/include +export LIBRARY_PATH := $(LIBTORCH_PATH)/lib +export LD_LIBRARY_PATH := $(LIBTORCH_PATH)/lib + +# Platform-specific notes +PLATFORM_DESC := Linux x86_64 with CUDA support diff --git a/arch/linux-amd64.mk b/arch/linux-amd64.mk new file mode 100644 index 0000000..e24ab32 --- /dev/null +++ b/arch/linux-amd64.mk @@ -0,0 +1,26 @@ +# Architecture configuration for Linux x86_64 +# This file is included by the main Makefile based on detected OS and architecture + +# LibTorch installation path (can be overridden via LIBTORCH environment variable) +LIBTORCH_PATH ?= $(shell [ -d "/opt/libtorch" ] && echo "/opt/libtorch" || echo "$(CURDIR)/libtorch") + +# Runtime library path variable for Linux +RUNTIME_LIB_VAR := LD_LIBRARY_PATH + +# CGO flags for compilation +export CGO_CFLAGS := -I$(LIBTORCH_PATH)/include/torch/csrc/api/include -I$(LIBTORCH_PATH)/include +export CGO_CFLAGS += -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function + +export CGO_LDFLAGS := -L$(LIBTORCH_PATH)/lib -ltorch -ltorch_cpu -lc10 +export CGO_LDFLAGS += -Wl,-rpath,$(LIBTORCH_PATH)/lib + +export CGO_CXXFLAGS := -std=c++17 -I$(LIBTORCH_PATH)/include/torch/csrc/api/include -I$(LIBTORCH_PATH)/include +export CGO_CXXFLAGS += -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function + +# Environment for runtime (tests, etc.) +export CPATH := $(LIBTORCH_PATH)/include/torch/csrc/api/include:$(LIBTORCH_PATH)/include +export LIBRARY_PATH := $(LIBTORCH_PATH)/lib +export LD_LIBRARY_PATH := $(LIBTORCH_PATH)/lib + +# Platform-specific notes +PLATFORM_DESC := Linux x86_64 (CPU) diff --git a/device.go b/device.go index 908d982..8b6c7ce 100644 --- a/device.go +++ b/device.go @@ -44,6 +44,21 @@ func (m Mps) IsAvailable() bool { return lib.AtcMpsIsAvailable() } +// CurrentAllocatedMemory returns the number of bytes currently allocated on the MPS device. +func (m Mps) CurrentAllocatedMemory() uint64 { + return lib.AtcMpsCurrentAllocatedMemory() +} + +// EmptyCache releases all cached MPS memory back to the system. +func (m Mps) EmptyCache() { + lib.AtcMpsEmptyCache() +} + +// Synchronize waits for all MPS operations to complete. +func (m Mps) Synchronize() { + lib.AtcMpsSynchronize() +} + // Cuda methods: // ============= diff --git a/libtch/lib.go b/libtch/lib.go index 7ad788f..7477b3e 100644 --- a/libtch/lib.go +++ b/libtch/lib.go @@ -1,18 +1,12 @@ package libtch -/* -CGO configuration is now managed via Makefile and arch-specific files in arch/ directory. -Build with 'make build' to ensure proper CGO flags are set. - -The Makefile exports: - CGO_CFLAGS - C compilation flags including LibTorch include paths - CGO_LDFLAGS - Linker flags including LibTorch library paths and rpath - CGO_CXXFLAGS - C++ compilation flags (e.g., -std=c++17) - -To override configuration: - LIBTORCH_PATH=/custom/path make build - make ARCH_CONFIG=linux-amd64-cuda build -*/ +// CGO flags are set via environment variables exported by the arch/*.mk +// Makefile includes (e.g., arch/darwin-arm64.mk). Run builds through +// `make` or export CGO_CFLAGS/CGO_LDFLAGS/CGO_CXXFLAGS manually. +// +// The only hardcoded flag is -I${SRCDIR} so CGO can find the local +// torch_api.h header next to the .cpp source. // #cgo CFLAGS: -I${SRCDIR} +// #cgo CXXFLAGS: -I${SRCDIR} import "C" diff --git a/libtch/tensor.go b/libtch/tensor.go index 5b8c6be..8b7547f 100644 --- a/libtch/tensor.go +++ b/libtch/tensor.go @@ -159,6 +159,21 @@ func AtcMpsIsAvailable() bool { return *(*bool)(unsafe.Pointer(&result)) } +// size_t atc_mps_current_allocated_memory(); +func AtcMpsCurrentAllocatedMemory() uint64 { + return uint64(C.atc_mps_current_allocated_memory()) +} + +// void atc_mps_empty_cache(); +func AtcMpsEmptyCache() { + C.atc_mps_empty_cache() +} + +// void atc_mps_synchronize(); +func AtcMpsSynchronize() { + C.atc_mps_synchronize() +} + // int atc_cudnn_is_available(); func AtcCudnnIsAvailable() bool { result := C.atc_cudnn_is_available() diff --git a/libtch/torch_api.cpp b/libtch/torch_api.cpp index 50b2f32..19e5830 100644 --- a/libtch/torch_api.cpp +++ b/libtch/torch_api.cpp @@ -8,6 +8,7 @@ #include #include #include +#include #include #define STB_IMAGE_IMPLEMENTATION @@ -186,6 +187,7 @@ bool at_autocast_set_enabled(bool b) { int at_device(tensor t) { PROTECT(auto device = t->device(); if (device.type() == at::kCPU) return -1; + if (device.type() == at::kMPS) return -2; if (device.type() == at::kCUDA) return device.index();) return -99; // error } @@ -790,6 +792,28 @@ int atc_mps_is_available() { return -1; } +size_t atc_mps_current_allocated_memory() { + PROTECT( + auto* allocator = at::mps::getIMPSAllocator(false); + if (allocator) return allocator->getCurrentAllocatedMemory(); + ) + return 0; +} + +void atc_mps_empty_cache() { + PROTECT( + auto* allocator = at::mps::getIMPSAllocator(false); + if (allocator) allocator->emptyCache(); + ) +} + +void atc_mps_synchronize() { + PROTECT( + auto guard = c10::impl::VirtualGuardImpl(at::DeviceType::MPS); + guard.synchronizeDevice(0); + ) +} + int atc_cudnn_is_available() { PROTECT(return torch::cuda::cudnn_is_available();) return -1; diff --git a/libtch/torch_api.h b/libtch/torch_api.h index 8465917..30b5fe6 100644 --- a/libtch/torch_api.h +++ b/libtch/torch_api.h @@ -156,6 +156,9 @@ void ats_free(scalar); int atc_cuda_device_count(); int atc_cuda_is_available(); int atc_mps_is_available(); +size_t atc_mps_current_allocated_memory(); +void atc_mps_empty_cache(); +void atc_mps_synchronize(); int atc_cudnn_is_available(); void atc_set_benchmark_cudnn(int b); void atc_synchronize(int64_t device_index);