torch 0.15 · 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__fft_c2c : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    unit F.return)
    F.result

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

val stubs__linalg_check_errors : (t -> string -> int -> unit F.return) F.result

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

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

val stubs__log_softmax_backward_data_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    int64 ->
    int ->
    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__lstm_mps : 
  (t Ctypes_static.ptr ->
    t ->
    t Ctypes_static.ptr ->
    int ->
    t Ctypes_static.ptr ->
    int ->
    int ->
    int64 ->
    float ->
    int ->
    int ->
    int ->
    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_dual_copy : 
  (t Ctypes_static.ptr -> t -> t -> int64 -> unit F.return) F.result

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

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

val stubs__mps_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__mps_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 ->
    unit F.return)
    F.result

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

package torch

Parameters

Signature