package torch

  1. Overview
  2. Docs
package torch
Legend:
Library
Module
Module type
Parameter
Class
Class type

Parameters

module F : Cstubs.FOREIGN

Signature

type t = unit Ctypes.ptr
val t : t Ctypes.typ
type scalar = unit Ctypes.ptr
val scalar : scalar Ctypes.typ
val stubs__mkldnn_reshape_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__mkldnn_transpose : (t Ctypes_static.ptr -> t -> int64 -> int64 -> unit F.return) F.result
val stubs__mkldnn_transpose_ : (t Ctypes_static.ptr -> t -> int64 -> int64 -> unit F.return) F.result
val stubs__mkldnn_transpose_out : (t Ctypes_static.ptr -> t -> t -> int64 -> int64 -> unit F.return) F.result
val stubs__mps_convolution : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs__mps_convolution_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs__mps_convolution_transpose : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs__mps_convolution_transpose_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs__mps_max_pool2d : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs__mps_max_pool2d_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs__native_decoder_only_multi_head_attention : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> int64 -> t -> t -> t -> t -> t -> t -> t -> int -> int -> unit F.return) F.result
val stubs__native_decoder_only_multi_head_attention_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t -> int64 -> int64 -> t -> t -> t -> t -> t -> t -> t -> int -> int -> unit F.return) F.result
val stubs__native_multi_head_attention : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> int64 -> t -> t -> t -> t -> t -> int -> int -> int64 -> int -> unit F.return) F.result
val stubs__native_multi_head_attention_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> int64 -> int64 -> t -> t -> t -> t -> t -> int -> int -> int64 -> int -> unit F.return) F.result
val stubs__neg_view : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs__neg_view_copy : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs__neg_view_copy_out : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs__nested_from_padded : (t Ctypes_static.ptr -> t -> t -> int -> unit F.return) F.result
val stubs__nested_from_padded_and_nested_example : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs__nested_from_padded_and_nested_example_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs__nested_from_padded_out : (t Ctypes_static.ptr -> t -> t -> t -> int -> unit F.return) F.result
val stubs__nested_select_backward : (t Ctypes_static.ptr -> t -> t -> int64 -> int64 -> unit F.return) F.result
val stubs__nested_sum_backward : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs__nested_view_from_buffer : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__nested_view_from_buffer_copy : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__nested_view_from_buffer_copy_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__new_zeros_with_same_feature_meta : (t Ctypes_static.ptr -> t -> t -> int64 -> unit F.return) F.result
val stubs__new_zeros_with_same_feature_meta_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> unit F.return) F.result
val stubs__nnpack_spatial_convolution : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__nnpack_spatial_convolution_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__pack_padded_sequence : (t Ctypes_static.ptr -> t -> t -> int -> unit F.return) F.result
val stubs__pack_padded_sequence_backward : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> t -> int -> unit F.return) F.result
val stubs__pack_padded_sequence_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int -> unit F.return) F.result
val stubs__pad_circular : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__pad_enum : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 -> float -> int -> unit F.return) F.result
val stubs__pad_packed_sequence : (t Ctypes_static.ptr -> t -> t -> int -> scalar -> int64 -> unit F.return) F.result
val stubs__pdist_backward : (t Ctypes_static.ptr -> t -> t -> float -> t -> unit F.return) F.result
val stubs__pdist_backward_out : (t Ctypes_static.ptr -> t -> t -> t -> float -> t -> unit F.return) F.result
val stubs__pin_memory : (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result
val stubs__pin_memory_out : (t Ctypes_static.ptr -> t -> t -> int -> unit F.return) F.result
val stubs__remove_batch_dim : (t Ctypes_static.ptr -> t -> int64 -> int64 -> int64 -> unit F.return) F.result
val stubs__reshape_alias : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__reshape_alias_copy : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__reshape_alias_copy_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__reshape_from_tensor : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs__resize_output : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs__resize_output_ : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs__resize_output_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs__rowwise_prune : (t Ctypes_static.ptr -> t -> t -> int -> unit F.return) F.result
val stubs__sample_dirichlet : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs__sample_dirichlet_out : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs__saturate_weight_to_fp16 : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs__scaled_dot_product_attention : (t Ctypes_static.ptr -> t -> t -> t -> t -> float -> int -> int -> unit F.return) F.result
val stubs__scaled_dot_product_attention_math : (t Ctypes_static.ptr -> t -> t -> t -> t -> float -> int -> int -> unit F.return) F.result
val stubs__scatter_reduce : (t Ctypes_static.ptr -> t -> int64 -> t -> t -> string -> int -> unit F.return) F.result
val stubs__scatter_reduce_ : (t Ctypes_static.ptr -> t -> int64 -> t -> t -> string -> int -> unit F.return) F.result
val stubs__scatter_reduce_two_out : (t Ctypes_static.ptr -> t -> t -> int64 -> t -> t -> string -> int -> unit F.return) F.result
val stubs__segment_reduce_backward : (t Ctypes_static.ptr -> t -> t -> t -> string -> t -> t -> int64 -> scalar -> unit F.return) F.result
val stubs__segment_reduce_backward_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> string -> t -> t -> int64 -> scalar -> unit F.return) F.result
val stubs__shape_as_tensor : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs__slow_conv2d_backward : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__sobol_engine_draw : (t Ctypes_static.ptr -> t -> int64 -> t -> int64 -> int64 -> int -> unit F.return) F.result
val stubs__sobol_engine_ff_ : (t Ctypes_static.ptr -> t -> int64 -> t -> int64 -> int64 -> unit F.return) F.result
val stubs__sobol_engine_initialize_state_ : (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result
val stubs__sobol_engine_scramble_ : (t Ctypes_static.ptr -> t -> t -> int64 -> unit F.return) F.result
val stubs__softmax : (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result
val stubs__softmax_backward_data : (t Ctypes_static.ptr -> t -> t -> int64 -> int -> unit F.return) F.result
val stubs__softmax_backward_data_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> int -> unit F.return) F.result
val stubs__softmax_out : (t Ctypes_static.ptr -> t -> t -> int64 -> int -> unit F.return) F.result
val stubs__sparse_addmm : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs__sparse_addmm_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> unit F.return) F.result
val stubs__sparse_broadcast_to : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__sparse_broadcast_to_copy : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__sparse_broadcast_to_copy_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__sparse_bsc_tensor_unsafe : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_bsr_tensor_unsafe : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_compressed_tensor_unsafe : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_coo_tensor_unsafe : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_coo_tensor_with_dims : (t Ctypes_static.ptr -> int64 -> int64 -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_coo_tensor_with_dims_and_tensors : (t Ctypes_static.ptr -> int64 -> int64 -> int64 Ctypes_static.ptr -> int -> t -> t -> int -> int -> unit F.return) F.result
val stubs__sparse_coo_tensor_with_dims_and_tensors_out : (t Ctypes_static.ptr -> t -> int64 -> int64 -> int64 Ctypes_static.ptr -> int -> t -> t -> unit F.return) F.result
val stubs__sparse_coo_tensor_with_dims_out : (t Ctypes_static.ptr -> t -> int64 -> int64 -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__sparse_csc_tensor_unsafe : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_csr_prod : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_csr_prod_dim_dtype_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_csr_sum : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_csr_sum_dim_dtype_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_csr_tensor_unsafe : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__sparse_log_softmax : (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result
val stubs__sparse_log_softmax_backward_data : (t Ctypes_static.ptr -> t -> t -> int64 -> t -> unit F.return) F.result
val stubs__sparse_log_softmax_backward_data_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> t -> unit F.return) F.result
val stubs__sparse_log_softmax_int : (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result
val stubs__sparse_log_softmax_out : (t Ctypes_static.ptr -> t -> t -> int64 -> int -> unit F.return) F.result
val stubs__sparse_mask_helper : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs__sparse_mask_helper_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs__sparse_mm : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs__sparse_softmax : (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result
val stubs__sparse_softmax_backward_data : (t Ctypes_static.ptr -> t -> t -> int64 -> t -> unit F.return) F.result
val stubs__sparse_softmax_backward_data_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> t -> unit F.return) F.result
val stubs__sparse_softmax_int : (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result