torch 0.14 · OCaml Package

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___and__ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs___and__tensor_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs___iand__ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs___iand__tensor_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs___ilshift__ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs___ilshift__tensor_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs___ior__ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs___ior__tensor_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs___irshift__ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs___irshift__tensor_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs___ixor__ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs___ixor__tensor_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs___lshift__ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs___lshift__tensor_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs___or__ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs___or__tensor_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs___rshift__ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs___rshift__tensor_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs___xor__ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs___xor__tensor_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs__adaptive_avg_pool2d : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs__adaptive_avg_pool2d_backward : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs__adaptive_avg_pool3d : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs__adaptive_avg_pool3d_backward : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs__add_batch_dim : 
  (t Ctypes_static.ptr -> t -> int64 -> int64 -> unit F.return) F.result

val stubs__add_relu : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs__add_relu_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs__add_relu_out : 
  (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result

val stubs__add_relu_scalar : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs__add_relu_scalar_ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs__aminmax : (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs__aminmax_dim : 
  (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result

val stubs__amp_update_scale_ : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    float ->
    float ->
    int64 ->
    unit F.return)
    F.result

val stubs__baddbmm_mkl_ : 
  (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result

val stubs__cast_byte : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs__cast_char : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs__cast_double : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs__cast_float : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs__cast_half : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs__cast_int : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs__cast_long : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs__cast_short : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs__cat : 
  (t Ctypes_static.ptr ->
    t Ctypes_static.ptr ->
    int ->
    int64 ->
    unit F.return)
    F.result

val stubs__cat_out : 
  (t Ctypes_static.ptr ->
    t ->
    t Ctypes_static.ptr ->
    int ->
    int64 ->
    unit F.return)
    F.result

val stubs__cdist_backward : 
  (t Ctypes_static.ptr -> t -> t -> t -> float -> t -> unit F.return) F.result

val stubs__cholesky_solve_helper : 
  (t Ctypes_static.ptr -> t -> t -> int -> unit F.return) F.result

val stubs__coalesce : (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs__coalesced_ : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs__compute_linear_combination : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs__compute_linear_combination_out : 
  (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result

val stubs__conj : (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs__conj_physical : (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs__conv_depthwise2d : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs__conv_depthwise2d_backward : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs__conv_depthwise2d_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs__convert_indices_from_coo_to_csr : 
  (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result

val stubs__convert_indices_from_coo_to_csr_out : 
  (t Ctypes_static.ptr -> t -> t -> int64 -> int -> unit F.return) F.result

val stubs__convolution : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    int ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs__convolution_deprecated : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs__convolution_mode : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    string ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    unit F.return)
    F.result

val stubs__convolution_nogroup : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs__copy_from : 
  (t Ctypes_static.ptr -> t -> t -> int -> unit F.return) F.result

val stubs__copy_from_and_resize : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs__ctc_loss : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs__ctc_loss_backward : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    t ->
    t ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs__cudnn_ctc_loss : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs__cudnn_init_dropout_state : 
  (t Ctypes_static.ptr ->
    float ->
    int ->
    int64 ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs__cudnn_rnn : 
  (t Ctypes_static.ptr ->
    t ->
    t Ctypes_static.ptr ->
    int ->
    int64 ->
    t ->
    t ->
    t ->
    int64 ->
    int64 ->
    int64 ->
    int64 ->
    int ->
    float ->
    int ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    t ->
    unit F.return)
    F.result

val stubs__cudnn_rnn_flatten_weight : 
  (t Ctypes_static.ptr ->
    t Ctypes_static.ptr ->
    int ->
    int64 ->
    int64 ->
    int64 ->
    int64 ->
    int64 ->
    int64 ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs__det_lu_based_helper : 
  (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs__det_lu_based_helper_backward_helper : 
  (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> unit F.return) F.result

val stubs__dim_arange : 
  (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result

val stubs__dirichlet_grad : 
  (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result

val stubs__embedding_bag : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int ->
    int64 ->
    int ->
    t ->
    int ->
    int64 ->
    unit F.return)
    F.result

val stubs__embedding_bag_backward : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    t ->
    t ->
    int64 ->
    int ->
    int64 ->
    int ->
    t ->
    int64 ->
    unit F.return)
    F.result

val stubs__embedding_bag_dense_backward : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    t ->
    int64 ->
    int ->
    int64 ->
    t ->
    int64 ->
    unit F.return)
    F.result

val stubs__embedding_bag_forward_only : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int ->
    int64 ->
    int ->
    t ->
    int ->
    int64 ->
    unit F.return)
    F.result

val stubs__embedding_bag_per_sample_weights_backward : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    t ->
    int64 ->
    int64 ->
    unit F.return)
    F.result

val stubs__embedding_bag_sparse_backward : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    t ->
    int64 ->
    int ->
    int64 ->
    t ->
    int64 ->
    unit F.return)
    F.result

val stubs__empty_affine_quantized : 
  (t Ctypes_static.ptr ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    int ->
    float ->
    int64 ->
    unit F.return)
    F.result

val stubs__empty_per_channel_affine_quantized : 
  (t Ctypes_static.ptr ->
    int64 Ctypes_static.ptr ->
    int ->
    t ->
    t ->
    int64 ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs__euclidean_dist : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs__fake_quantize_learnable_per_channel_affine : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 ->
    int64 ->
    int64 ->
    float ->
    unit F.return)
    F.result

val stubs__fake_quantize_learnable_per_channel_affine_backward : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    int64 ->
    int64 ->
    int64 ->
    float ->
    unit F.return)
    F.result

val stubs__fake_quantize_learnable_per_tensor_affine : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 ->
    int64 ->
    float ->
    unit F.return)
    F.result

val stubs__fake_quantize_learnable_per_tensor_affine_backward : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    int64 ->
    int64 ->
    float ->
    unit F.return)
    F.result

val stubs__fake_quantize_per_tensor_affine_cachemask_tensor_qparams : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    int64 ->
    int64 ->
    unit F.return)
    F.result

val stubs__fft_c2c : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs__fft_c2c_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs__fft_c2r : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int64 ->
    unit F.return)
    F.result

val stubs__fft_c2r_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int64 ->
    unit F.return)
    F.result

val stubs__fft_r2c : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs__fft_r2c_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs__fused_dropout : 
  (t Ctypes_static.ptr -> t -> float -> unit F.return) F.result

val stubs__fused_moving_avg_obs_fq_helper : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    t ->
    t ->
    t ->
    float ->
    int64 ->
    int64 ->
    int64 ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs__fw_primal : 
  (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result

val stubs__gather_sparse_backward : 
  (t Ctypes_static.ptr -> t -> int64 -> t -> t -> unit F.return) F.result

val stubs__grid_sampler_2d_cpu_fallback : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs__grid_sampler_2d_cpu_fallback_backward : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs__index_copy_ : 
  (t Ctypes_static.ptr -> t -> int64 -> t -> t -> unit F.return) F.result