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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

val stubs__sparse_addmm : 
  (t Ctypes_static.ptr -> t -> t -> t -> 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_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__svd_helper : 
  (t Ctypes_static.ptr -> t -> int -> int -> 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_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__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__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__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__values : (t Ctypes_static.ptr -> 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_cuda_interface : 
  (t Ctypes_static.ptr -> t -> t -> int64 -> unit F.return) F.result

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

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

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

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

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