torch 0.17 · 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_cudnn_batch_norm_backward : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    t ->
    t ->
    t ->
    float ->
    t ->
    unit F.return)
    F.result

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

package torch

Parameters

Signature