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

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

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

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

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

val stubs_slow_conv_dilated2d : 
  (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_slow_conv_dilated3d : 
  (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_slow_conv_transpose2d : 
  (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 ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs_slow_conv_transpose2d_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 ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

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

val stubs_slow_conv_transpose3d_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 ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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