package torch

  1. Overview
  2. Docs
package torch
Legend:
Library
Module
Module type
Parameter
Class
Class type

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__cudnn_rnn : (t Ctypes_static.ptr -> t -> t Ctypes_static.ptr -> int -> int64 -> t -> t -> t -> int64 -> int64 -> int64 -> int64 -> int -> float -> int -> int -> int64 Ctypes_static.ptr -> int -> t -> unit F.return) F.result
val stubs__cudnn_rnn_flatten_weight : (t Ctypes_static.ptr -> t Ctypes_static.ptr -> int -> int64 -> int64 -> int64 -> int64 -> int64 -> int64 -> int -> int -> unit F.return) F.result
val stubs__cudnn_rnn_flatten_weight_out : (t Ctypes_static.ptr -> t -> t Ctypes_static.ptr -> int -> int64 -> int64 -> int64 -> int64 -> int64 -> int64 -> int -> int -> unit F.return) F.result
val stubs__cudnn_rnn_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t Ctypes_static.ptr -> int -> int64 -> t -> t -> t -> int64 -> int64 -> int64 -> int64 -> int -> float -> int -> int -> int64 Ctypes_static.ptr -> int -> t -> unit F.return) F.result
val stubs__dim_arange : (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result
val stubs__dirichlet_grad : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs__dirichlet_grad_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> unit F.return) F.result
val stubs__efficientzerotensor : (t Ctypes_static.ptr -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs__efficientzerotensor_out : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs__embedding_bag : (t Ctypes_static.ptr -> t -> t -> t -> int -> int64 -> int -> t -> int -> int64 -> unit F.return) F.result
val stubs__embedding_bag_backward : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> int64 -> int -> int64 -> int -> t -> int64 -> unit F.return) F.result
val stubs__embedding_bag_dense_backward : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> int64 -> int -> int64 -> t -> int64 -> unit F.return) F.result
val stubs__embedding_bag_dense_backward_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> int64 -> int -> int64 -> t -> int64 -> unit F.return) F.result
val stubs__embedding_bag_forward_only : (t Ctypes_static.ptr -> t -> t -> t -> int -> int64 -> int -> t -> int -> int64 -> unit F.return) F.result
val stubs__embedding_bag_forward_only_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t -> int -> int64 -> int -> t -> int -> int64 -> unit F.return) F.result
val stubs__embedding_bag_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t -> int -> int64 -> int -> t -> int -> int64 -> unit F.return) F.result
val stubs__embedding_bag_per_sample_weights_backward : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> int64 -> int64 -> unit F.return) F.result
val stubs__embedding_bag_per_sample_weights_backward_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> int64 -> int64 -> unit F.return) F.result
val stubs__embedding_bag_sparse_backward : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> int64 -> int -> int64 -> t -> int64 -> unit F.return) F.result
val stubs__empty_affine_quantized : (t Ctypes_static.ptr -> int64 Ctypes_static.ptr -> int -> int -> int -> float -> int64 -> unit F.return) F.result
val stubs__empty_affine_quantized_out : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> float -> int64 -> unit F.return) F.result
val stubs__empty_per_channel_affine_quantized : (t Ctypes_static.ptr -> int64 Ctypes_static.ptr -> int -> t -> t -> int64 -> int -> int -> unit F.return) F.result
val stubs__empty_per_channel_affine_quantized_out : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> t -> t -> int64 -> unit F.return) F.result
val stubs__euclidean_dist : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs__euclidean_dist_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs__fake_quantize_learnable_per_channel_affine : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> int64 -> int64 -> float -> unit F.return) F.result
val stubs__fake_quantize_learnable_per_channel_affine_backward : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 -> int64 -> int64 -> float -> unit F.return) F.result
val stubs__fake_quantize_learnable_per_channel_affine_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 -> int64 -> int64 -> float -> unit F.return) F.result
val stubs__fake_quantize_learnable_per_tensor_affine : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> int64 -> float -> unit F.return) F.result
val stubs__fake_quantize_learnable_per_tensor_affine_backward : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 -> int64 -> float -> unit F.return) F.result
val stubs__fake_quantize_learnable_per_tensor_affine_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 -> int64 -> float -> unit F.return) F.result
val stubs__fake_quantize_per_tensor_affine_cachemask_tensor_qparams : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 -> int64 -> unit F.return) F.result
val stubs__fake_quantize_per_tensor_affine_cachemask_tensor_qparams_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> int64 -> int64 -> unit F.return) F.result
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__flash_scaled_dot_product_attention : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> int64 -> int64 -> float -> int -> unit F.return) F.result
val stubs__foobar : (t Ctypes_static.ptr -> t -> int -> int -> int -> unit F.return) F.result
val stubs__foobar_out : (t Ctypes_static.ptr -> t -> t -> int -> int -> int -> unit F.return) F.result
val stubs__fused_dropout : (t Ctypes_static.ptr -> t -> float -> unit F.return) F.result
val stubs__fused_dropout_out : (t Ctypes_static.ptr -> t -> t -> 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__grid_sampler_2d_cpu_fallback_out : (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_bin_edges_out : (t Ctypes_static.ptr -> int -> t -> int64 Ctypes_static.ptr -> int -> float Ctypes_static.ptr -> int -> t -> int -> unit 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_cts_out : (t Ctypes_static.ptr -> t -> 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__histogramdd_from_bin_tensors_out : (t Ctypes_static.ptr -> t -> t -> t Ctypes_static.ptr -> int -> t -> 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_ : (t Ctypes_static.ptr -> t -> t Ctypes_static.ptr -> int -> t -> int -> int -> unit F.return) F.result
val stubs__index_put_impl_out : (t Ctypes_static.ptr -> t -> 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_det : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs__linalg_det_result : (t Ctypes_static.ptr -> t -> t -> t -> t -> unit F.return) F.result
val stubs__linalg_eigh : (t Ctypes_static.ptr -> t -> string -> int -> unit F.return) F.result
val stubs__linalg_eigh_eigenvalues : (t Ctypes_static.ptr -> t -> t -> t -> string -> int -> unit F.return) F.result
val stubs__linalg_slogdet : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs__linalg_slogdet_sign : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> unit F.return) F.result
val stubs__linalg_solve_ex : (t Ctypes_static.ptr -> t -> t -> int -> int -> unit F.return) F.result
val stubs__linalg_solve_ex_result : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> int -> int -> unit F.return) F.result
val stubs__linalg_svd : (t Ctypes_static.ptr -> t -> int -> int -> string -> unit F.return) F.result
val stubs__linalg_svd_u : (t Ctypes_static.ptr -> t -> t -> t -> t -> int -> int -> 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 -> 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__lstm_mps_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> 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_channel_quantized_tensor_out : (t Ctypes_static.ptr -> t -> 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__make_per_tensor_quantized_tensor_out : (t Ctypes_static.ptr -> t -> 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_scale_out : (t Ctypes_static.ptr -> t -> t -> t -> float -> unit F.return) F.result
val stubs__masked_softmax : (t Ctypes_static.ptr -> t -> t -> int64 -> int -> 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__masked_softmax_backward_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 -> int -> unit F.return) F.result
val stubs__masked_softmax_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> int -> 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