package torch

  1. Overview
  2. Docs

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_softmax_out : (t Ctypes_static.ptr -> t -> 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_sparse_matmul_out : (t Ctypes_static.ptr -> t -> 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_backward_out : (t Ctypes_static.ptr -> t -> 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_dim_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__sparse_sum_dtype : (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result
val stubs__spdiags : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__spdiags_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> 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__standard_gamma_grad_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs__standard_gamma_out : (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__symeig_helper_out : (t Ctypes_static.ptr -> t -> t -> 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_autograd_multiple_dispatch : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs__test_autograd_multiple_dispatch_fullcoverage_out : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs__test_autograd_multiple_dispatch_ntonly : (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result
val stubs__test_autograd_multiple_dispatch_view : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs__test_autograd_multiple_dispatch_view_copy : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs__test_autograd_multiple_dispatch_view_copy_out : (t Ctypes_static.ptr -> t -> t -> 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_filled_intlist_out : (t Ctypes_static.ptr -> t -> 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_floatlist_out : (t Ctypes_static.ptr -> t -> 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_optional_intlist_out : (t Ctypes_static.ptr -> t -> 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__test_warn_in_autograd_out : (t Ctypes_static.ptr -> t -> 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_gru_cell_backward_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t -> int -> unit F.return) F.result
val stubs__thnn_fused_gru_cell_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t -> 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__thnn_fused_lstm_cell_backward_impl_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t -> t -> int -> unit F.return) F.result
val stubs__thnn_fused_lstm_cell_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t -> t -> 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_copy_out : (t Ctypes_static.ptr -> t -> t -> 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__to_dense_out : (t Ctypes_static.ptr -> t -> 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__torch_cuda_cu_linker_symbol_op_out : (t Ctypes_static.ptr -> t -> 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__transform_bias_rescale_qkv_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> int64 -> unit F.return) F.result
val stubs__transformer_decoder_only_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 -> t -> t -> unit F.return) F.result
val stubs__transformer_decoder_only_layer_fwd_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 -> int64 -> t -> t -> t -> t -> int -> int -> float -> t -> t -> t -> t -> t -> t -> t -> t -> t -> t -> t -> 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 -> int64 -> int -> unit F.return) F.result
val stubs__transformer_encoder_layer_fwd_out : (t Ctypes_static.ptr -> t -> t -> int64 -> int64 -> t -> t -> t -> t -> int -> int -> float -> t -> t -> t -> t -> t -> t -> t -> t -> t -> int64 -> int -> 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__trilinear_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 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs__triton_multi_head_attention : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> int64 -> t -> t -> t -> t -> t -> unit F.return) F.result
val stubs__triton_multi_head_attention_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 -> int64 -> t -> t -> t -> t -> t -> unit F.return) F.result
val stubs__triton_scaled_dot_attention : (t Ctypes_static.ptr -> t -> t -> t -> float -> unit F.return) F.result
val stubs__triton_scaled_dot_attention_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> float -> 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__unique2_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int -> int -> int -> unit F.return) F.result
val stubs__unique_out : (t Ctypes_static.ptr -> t -> t -> t -> 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__unsafe_view_out : (t Ctypes_static.ptr -> t -> 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_backward_vec_out : (t Ctypes_static.ptr -> t -> 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_bicubic2d_aa_vec_out : (t Ctypes_static.ptr -> t -> 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_backward_vec_out : (t Ctypes_static.ptr -> t -> 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_bilinear2d_aa_vec_out : (t Ctypes_static.ptr -> t -> 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_backward_vec_out : (t Ctypes_static.ptr -> t -> 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_exact1d_vec_out : (t Ctypes_static.ptr -> t -> 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