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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

val stubs_slogdet_out : 
  (t Ctypes_static.ptr -> t -> t -> 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_dilated2d_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 ->
    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_dilated3d_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 ->
    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

package torch

Parameters

Signature