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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

package torch

Parameters

Signature