torch 0.12 · 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_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_mean1 : 
  (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_median1 : 
  (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result

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

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

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

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

val stubs_min_out1 : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 ->
    int ->
    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_mkldnn_adaptive_avg_pool2d : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    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_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_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_out : 
  (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_moveaxis1 : 
  (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_movedim1 : 
  (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_out : 
  (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_mul1 : (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

package torch

Parameters

Signature