torch 0.11 · 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__1 : (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__1 : (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__1 : 
  (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__1 : (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__1 : 
  (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__1 : (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__1 : 
  (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__1 : (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__1 : 
  (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__1 : (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__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__addmv_impl_ : 
  (t Ctypes_static.ptr -> t -> t -> t -> t -> unit F.return) F.result

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

val stubs__aminmax1 : 
  (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__bmm : 
  (t Ctypes_static.ptr -> t -> t -> int -> unit F.return) F.result

val stubs__bmm_out : 
  (t Ctypes_static.ptr -> t -> t -> t -> int -> 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_helper : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs__cholesky_solve_helper : 
  (t Ctypes_static.ptr -> t -> t -> int -> 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__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 ->
    unit F.return)
    F.result

val stubs__convolution1 : 
  (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_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__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 ->
    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 ->
    int ->
    int ->
    unit F.return)
    F.result

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

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

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

val stubs__cumsum_out : 
  (t Ctypes_static.ptr -> t -> t -> int64 -> 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 ->
    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 ->
    unit F.return)
    F.result

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

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

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

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

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

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

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

val stubs__fused_dropout : 
  (t Ctypes_static.ptr -> t -> float -> 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

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

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

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

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

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

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

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

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

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

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

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

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

val stubs__mkldnn_reshape : 
  (t Ctypes_static.ptr ->
    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

package torch

Parameters

Signature