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_max_pool3d_with_indices_backward : 
  (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 ->
    t ->
    unit F.return)
    F.result

val stubs_max_pool3d_with_indices_backward_grad_input : 
  (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 ->
    int ->
    t ->
    unit F.return)
    F.result

val stubs_max_pool3d_with_indices_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 ->
    int ->
    unit F.return)
    F.result

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

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

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

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

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

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

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

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

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

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

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

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

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

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

val stubs_meshgrid_indexing : 
  (t Ctypes_static.ptr ->
    int ->
    string ->
    t Ctypes_static.ptr F.return)
    F.result

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

val stubs_mkldnn_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_mkldnn_max_pool2d_backward : 
  (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 ->
    int ->
    unit F.return)
    F.result

val stubs_mkldnn_max_pool3d : 
  (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_mkldnn_max_pool3d_backward : 
  (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 ->
    int ->
    unit F.return)
    F.result

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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