Documentation

special_hermite_polynomial_h_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_hermite_polynomial_h_st :: Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_hermite_polynomial_h_ts :: Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_hermite_polynomial_h_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_hermite_polynomial_h_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_hermite_polynomial_h_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_hermite_polynomial_he_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_hermite_polynomial_he_st :: Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_hermite_polynomial_he_ts :: Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_hermite_polynomial_he_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_hermite_polynomial_he_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_hermite_polynomial_he_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_laguerre_polynomial_l_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_laguerre_polynomial_l_st :: Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_laguerre_polynomial_l_ts :: Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_laguerre_polynomial_l_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_laguerre_polynomial_l_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_laguerre_polynomial_l_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_legendre_polynomial_p_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_legendre_polynomial_p_st :: Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_legendre_polynomial_p_ts :: Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_legendre_polynomial_p_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_legendre_polynomial_p_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_legendre_polynomial_p_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_modified_bessel_i0_t :: Ptr Tensor -> IO (Ptr Tensor) Source #

special_modified_bessel_i0_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_modified_bessel_i1_t :: Ptr Tensor -> IO (Ptr Tensor) Source #

special_modified_bessel_i1_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_modified_bessel_k0_t :: Ptr Tensor -> IO (Ptr Tensor) Source #

special_modified_bessel_k0_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_modified_bessel_k1_t :: Ptr Tensor -> IO (Ptr Tensor) Source #

special_modified_bessel_k1_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_scaled_modified_bessel_k0_t :: Ptr Tensor -> IO (Ptr Tensor) Source #

special_scaled_modified_bessel_k0_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_scaled_modified_bessel_k1_t :: Ptr Tensor -> IO (Ptr Tensor) Source #

special_scaled_modified_bessel_k1_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_t_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_t_st :: Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_t_ts :: Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_t_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_t_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_t_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_u_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_u_st :: Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_u_ts :: Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_u_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_u_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_u_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_v_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_v_st :: Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_v_ts :: Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_v_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_v_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_v_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_w_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_w_st :: Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_w_ts :: Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_w_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_w_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #

special_shifted_chebyshev_polynomial_w_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

special_spherical_bessel_j0_t :: Ptr Tensor -> IO (Ptr Tensor) Source #

special_spherical_bessel_j0_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

_foobar_tbbb :: Ptr Tensor -> CBool -> CBool -> CBool -> IO (Ptr Tensor) Source #

_foobar_tbb :: Ptr Tensor -> CBool -> CBool -> IO (Ptr Tensor) Source #

_foobar_tb :: Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #

_foobar_t :: Ptr Tensor -> IO (Ptr Tensor) Source #

_fused_adam__lllllldddddbbtt :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> CBool -> Ptr Tensor -> Ptr Tensor -> IO () Source #

_fused_adam__lllllldddddbbt :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> CBool -> Ptr Tensor -> IO () Source #

_fused_adam__lllllldddddbb :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> CBool -> IO () Source #

_fused_adamw__lllllldddddbbtt :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> CBool -> Ptr Tensor -> Ptr Tensor -> IO () Source #

_fused_adamw__lllllldddddbbt :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> CBool -> Ptr Tensor -> IO () Source #

_fused_adamw__lllllldddddbb :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> CBool -> IO () Source #

_new_zeros_with_same_feature_meta_out_tttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

_new_zeros_with_same_feature_meta_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

_cudnn_ctc_loss_out_ttttlllbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Int64 -> CBool -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_cudnn_rnn_flatten_weight_out_tlllllllbb :: Ptr Tensor -> Ptr TensorList -> Int64 -> Int64 -> Int64 -> Int64 -> Int64 -> Int64 -> CBool -> CBool -> IO (Ptr Tensor) Source #

_cudnn_rnn_out_ttttttlltttllllbdbblt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr TensorList -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> Int64 -> CBool -> CDouble -> CBool -> CBool -> Ptr IntArray -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor, Tensor))) Source #

_cudnn_rnn_backward_out_tttltlltttttttllllbdbbltta :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr TensorList -> Ptr Tensor -> Ptr TensorList -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> Int64 -> CBool -> CDouble -> CBool -> CBool -> Ptr IntArray -> Ptr Tensor -> Ptr Tensor -> Ptr (StdArray '(CBool, 4)) -> IO () Source #

_cudnn_init_dropout_state_out_tdbl :: Ptr Tensor -> CDouble -> CBool -> Int64 -> IO (Ptr Tensor) Source #

_fused_dropout_out_tttdG :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> Ptr Generator -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_fused_dropout_out_tttd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_masked_scale_out_tttd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> IO (Ptr Tensor) Source #

native_dropout_out_tttdb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

native_dropout_backward_out_tttd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> IO (Ptr Tensor) Source #

_conj_physical_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

_add_relu_out_ttss :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> Ptr Scalar -> IO (Ptr Tensor) Source #

_add_relu_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

add_out_ttss :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> Ptr Scalar -> IO (Ptr Tensor) Source #

add_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

affine_grid_generator_out_ttlb :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> CBool -> IO (Ptr Tensor) Source #

bartlett_window_out_tl :: Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

bartlett_window_out_tlb :: Ptr Tensor -> Int64 -> CBool -> IO (Ptr Tensor) Source #

quantized_batch_norm_out_ttttttddl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> Int64 -> IO (Ptr Tensor) Source #

bernoulli_out_tttG :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Generator -> IO (Ptr Tensor) Source #

bernoulli_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

bernoulli_ttG :: Ptr Tensor -> Ptr Tensor -> Ptr Generator -> IO (Ptr Tensor) Source #

bernoulli_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

bernoulli_out_ttdG :: Ptr Tensor -> Ptr Tensor -> CDouble -> Ptr Generator -> IO (Ptr Tensor) Source #

bernoulli_out_ttd :: Ptr Tensor -> Ptr Tensor -> CDouble -> IO (Ptr Tensor) Source #

binary_cross_entropy_with_logits_out_tttttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

binary_cross_entropy_with_logits_out_ttttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

binary_cross_entropy_with_logits_out_tttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

binary_cross_entropy_with_logits_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

bincount_out_tttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

bincount_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

bincount_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

blackman_window_out_tl :: Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

blackman_window_out_tlb :: Ptr Tensor -> Int64 -> CBool -> IO (Ptr Tensor) Source #

block_diag_out_tl :: Ptr Tensor -> Ptr TensorList -> IO (Ptr Tensor) Source #

constant_pad_nd_out_ttls :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Scalar -> IO (Ptr Tensor) Source #

constant_pad_nd_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

convolution_out_ttttlllbll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> Ptr IntArray -> Int64 -> IO (Ptr Tensor) Source #

convolution_backward_out_ttttttllllblla :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> Ptr IntArray -> Int64 -> Ptr (StdArray '(CBool, 3)) -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

convolution_overrideable_out_ttttlllbll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> Ptr IntArray -> Int64 -> IO (Ptr Tensor) Source #

convolution_backward_overrideable_out_ttttttlllblla :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> Ptr IntArray -> Int64 -> Ptr (StdArray '(CBool, 3)) -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

_convolution_out_ttttlllbllbbbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> Ptr IntArray -> Int64 -> CBool -> CBool -> CBool -> CBool -> IO (Ptr Tensor) Source #

conv_tbc_out_ttttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

conv_tbc_out_tttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

copy_out_tttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #

copy_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

_copy_from_out_tttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #

_copy_from_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

_copy_from_and_resize_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

count_nonzero_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

count_nonzero_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

cudnn_affine_grid_generator_out_ttllll :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> Int64 -> IO (Ptr Tensor) Source #

cudnn_affine_grid_generator_backward_out_ttllll :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> Int64 -> IO (Ptr Tensor) Source #

cudnn_batch_norm_out_tttttttttbdd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CDouble -> CDouble -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

cudnn_batch_norm_backward_out_ttttttttttdt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

cudnn_convolution_out_tttllllbbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Int64 -> CBool -> CBool -> CBool -> IO (Ptr Tensor) Source #

cudnn_convolution_transpose_out_tttlllllbbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Int64 -> CBool -> CBool -> CBool -> IO (Ptr Tensor) Source #

_mps_convolution_transpose_out_tttlllll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Int64 -> IO (Ptr Tensor) Source #

mps_convolution_transpose_backward_out_tttttllllla :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Int64 -> Ptr (StdArray '(CBool, 2)) -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

cudnn_convolution_relu_out_ttttllll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Int64 -> IO (Ptr Tensor) Source #

cudnn_convolution_add_relu_out_ttttstllll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Int64 -> IO (Ptr Tensor) Source #

cudnn_grid_sampler_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

cudnn_grid_sampler_backward_out_ttttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_ctc_loss_out_ttttlllb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Int64 -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_ctc_loss_out_ttttlll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Int64 -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_ctc_loss_out_ttttll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_ctc_loss_out_ttttttlb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_ctc_loss_out_ttttttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_ctc_loss_out_tttttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_ctc_loss_backward_out_ttttllttlb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> IO (Ptr Tensor) Source #

_ctc_loss_backward_out_ttttllttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

diag_embed_out_ttlll :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> IO (Ptr Tensor) Source #

diag_embed_out_ttll :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> IO (Ptr Tensor) Source #

diag_embed_out_ttl :: Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

diag_embed_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

diagonal_backward_out_ttllll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Int64 -> Int64 -> Int64 -> IO (Ptr Tensor) Source #

div_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

div_out_ttss :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> Ptr StdString -> IO (Ptr Tensor) Source #

embedding_out_tttlbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> CBool -> IO (Ptr Tensor) Source #

embedding_out_tttlb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> IO (Ptr Tensor) Source #

embedding_out_tttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

embedding_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

embedding_dense_backward_out_tttllb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> CBool -> IO (Ptr Tensor) Source #

embedding_renorm_out_tttdd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> IO (Ptr Tensor) Source #

embedding_renorm_ttdd :: Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> IO (Ptr Tensor) Source #

_embedding_bag_forward_only_out_tttttttblbtbl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> CBool -> Ptr Tensor -> CBool -> Int64 -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_forward_only_out_tttttttblbtb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> CBool -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_forward_only_out_tttttttblbt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> CBool -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_forward_only_out_tttttttblb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_forward_only_out_tttttttbl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_forward_only_out_tttttttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_forward_only_out_ttttttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_out_tttttttblbtbl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> CBool -> Ptr Tensor -> CBool -> Int64 -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_out_tttttttblbtb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> CBool -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_out_tttttttblbt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> CBool -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_out_tttttttblb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_out_tttttttbl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_out_tttttttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_out_ttttttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

_embedding_bag_dense_backward_out_ttttttlbltl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> Int64 -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

_embedding_bag_dense_backward_out_ttttttlblt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> Int64 -> Ptr Tensor -> IO (Ptr Tensor) Source #

_embedding_bag_per_sample_weights_backward_out_ttttttll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> IO (Ptr Tensor) Source #

_embedding_bag_per_sample_weights_backward_out_ttttttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

empty_out_tlNM :: Ptr Tensor -> Ptr IntArray -> Ptr DimnameList -> MemoryFormat -> IO (Ptr Tensor) Source #

empty_out_tlN :: Ptr Tensor -> Ptr IntArray -> Ptr DimnameList -> IO (Ptr Tensor) Source #

new_empty_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

new_empty_strided_out_ttll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

new_full_out_ttls :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Scalar -> IO (Ptr Tensor) Source #

new_zeros_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

new_ones_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

_empty_affine_quantized_out_tldlM :: Ptr Tensor -> Ptr IntArray -> CDouble -> Int64 -> MemoryFormat -> IO (Ptr Tensor) Source #

_empty_affine_quantized_out_tldl :: Ptr Tensor -> Ptr IntArray -> CDouble -> Int64 -> IO (Ptr Tensor) Source #

_empty_affine_quantized_out_tld :: Ptr Tensor -> Ptr IntArray -> CDouble -> IO (Ptr Tensor) Source #

_empty_affine_quantized_out_tl :: Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

_empty_per_channel_affine_quantized_out_tlttlM :: Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr Tensor -> Int64 -> MemoryFormat -> IO (Ptr Tensor) Source #

_empty_per_channel_affine_quantized_out_tlttl :: Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

resize_out_ttlM :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> MemoryFormat -> IO (Ptr Tensor) Source #

resize_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

resize_tlM :: Ptr Tensor -> Ptr IntArray -> MemoryFormat -> IO (Ptr Tensor) Source #

resize_tl :: Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

_resize_output_out_ttlD :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> DeviceType -> IO (Ptr Tensor) Source #

_resize_output_tlD :: Ptr Tensor -> Ptr IntArray -> DeviceType -> IO (Ptr Tensor) Source #

empty_quantized_out_tltM :: Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> MemoryFormat -> IO (Ptr Tensor) Source #

empty_quantized_out_tlt :: Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> IO (Ptr Tensor) Source #

empty_like_out_ttM :: Ptr Tensor -> Ptr Tensor -> MemoryFormat -> IO (Ptr Tensor) Source #

empty_like_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

empty_strided_out_tll :: Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

fill_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

fill_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

full_out_tlsN :: Ptr Tensor -> Ptr IntArray -> Ptr Scalar -> Ptr DimnameList -> IO (Ptr Tensor) Source #

full_like_out_ttsM :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> MemoryFormat -> IO (Ptr Tensor) Source #

full_like_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #

from_file_out_tsbl :: Ptr Tensor -> Ptr StdString -> CBool -> Int64 -> IO (Ptr Tensor) Source #

from_file_out_tsb :: Ptr Tensor -> Ptr StdString -> CBool -> IO (Ptr Tensor) Source #

from_file_out_ts :: Ptr Tensor -> Ptr StdString -> IO (Ptr Tensor) Source #

grid_sampler_2d_out_tttllb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> CBool -> IO (Ptr Tensor) Source #

grid_sampler_2d_backward_out_tttttllba :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> CBool -> Ptr (StdArray '(CBool, 2)) -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_grid_sampler_2d_cpu_fallback_out_tttllb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> CBool -> IO (Ptr Tensor) Source #

grid_sampler_3d_out_tttllb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> CBool -> IO (Ptr Tensor) Source #

grid_sampler_3d_backward_out_tttttllba :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> CBool -> Ptr (StdArray '(CBool, 2)) -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

hann_window_out_tl :: Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

hann_window_out_tlb :: Ptr Tensor -> Int64 -> CBool -> IO (Ptr Tensor) Source #

hamming_window_out_tl :: Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

hamming_window_out_tlb :: Ptr Tensor -> Int64 -> CBool -> IO (Ptr Tensor) Source #

hamming_window_out_tlbd :: Ptr Tensor -> Int64 -> CBool -> CDouble -> IO (Ptr Tensor) Source #

hamming_window_out_tlbdd :: Ptr Tensor -> Int64 -> CBool -> CDouble -> CDouble -> IO (Ptr Tensor) Source #

kaiser_window_out_tl :: Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #

kaiser_window_out_tlb :: Ptr Tensor -> Int64 -> CBool -> IO (Ptr Tensor) Source #

kaiser_window_out_tlbd :: Ptr Tensor -> Int64 -> CBool -> CDouble -> IO (Ptr Tensor) Source #

native_group_norm_out_ttttttlllld :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> Int64 -> CDouble -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

native_group_norm_backward_out_ttttttttlllla :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> Int64 -> Ptr (StdArray '(CBool, 3)) -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

index_put_out_ttltb :: Ptr Tensor -> Ptr Tensor -> Ptr (C10List (C10Optional Tensor)) -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #

index_put_out_ttlt :: Ptr Tensor -> Ptr Tensor -> Ptr (C10List (C10Optional Tensor)) -> Ptr Tensor -> IO (Ptr Tensor) Source #

_index_put_impl_out_ttltbb :: Ptr Tensor -> Ptr Tensor -> Ptr (C10List (C10Optional Tensor)) -> Ptr Tensor -> CBool -> CBool -> IO (Ptr Tensor) Source #

_index_put_impl_out_ttltb :: Ptr Tensor -> Ptr Tensor -> Ptr (C10List (C10Optional Tensor)) -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #

_index_put_impl_out_ttlt :: Ptr Tensor -> Ptr Tensor -> Ptr (C10List (C10Optional Tensor)) -> Ptr Tensor -> IO (Ptr Tensor) Source #

_index_put_impl_tltbb :: Ptr Tensor -> Ptr (C10List (C10Optional Tensor)) -> Ptr Tensor -> CBool -> CBool -> IO (Ptr Tensor) Source #

_index_put_impl_tltb :: Ptr Tensor -> Ptr (C10List (C10Optional Tensor)) -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #

_index_put_impl_tlt :: Ptr Tensor -> Ptr (C10List (C10Optional Tensor)) -> Ptr Tensor -> IO (Ptr Tensor) Source #

isnan_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

native_layer_norm_out_ttttlttd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr Tensor -> CDouble -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

native_layer_norm_backward_out_tttttltttta :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr (StdArray '(CBool, 3)) -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

linear_backward_out_tttttta :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr (StdArray '(CBool, 3)) -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

mkldnn_linear_out_tttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

mkldnn_linear_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

mkldnn_linear_backward_input_out_tltt :: Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

mkldnn_linear_backward_weights_out_tttttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

mkldnn_linear_backward_out_tttttta :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr (StdArray '(CBool, 3)) -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

matmul_backward_out_ttttta :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr (StdArray '(CBool, 2)) -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_aminmax_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_aminmax_out_tttlb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

_aminmax_out_tttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #

max_pool2d_backward_out_tttllllb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> IO (Ptr Tensor) Source #

max_pool2d_backward_out_tttllll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

max_pool2d_backward_out_tttlll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

max_pool2d_backward_out_tttll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

max_pool2d_backward_out_tttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool2d_out_ttllllb :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> IO (Ptr Tensor) Source #

mkldnn_max_pool2d_out_ttllll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool2d_out_ttlll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool2d_out_ttll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool2d_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool2d_backward_out_ttttllllb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> IO (Ptr Tensor) Source #

mkldnn_max_pool2d_backward_out_ttttllll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool2d_backward_out_ttttlll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool2d_backward_out_ttttll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool2d_backward_out_ttttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool3d_out_ttllllb :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> IO (Ptr Tensor) Source #

mkldnn_max_pool3d_out_ttllll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool3d_out_ttlll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool3d_out_ttll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool3d_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool3d_backward_out_ttttllllb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> IO (Ptr Tensor) Source #

mkldnn_max_pool3d_backward_out_ttttllll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool3d_backward_out_ttttlll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool3d_backward_out_ttttll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

mkldnn_max_pool3d_backward_out_ttttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

quantized_max_pool1d_out_ttllllb :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> IO (Ptr Tensor) Source #

quantized_max_pool1d_out_ttllll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

quantized_max_pool1d_out_ttlll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

quantized_max_pool1d_out_ttll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

quantized_max_pool1d_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

quantized_max_pool2d_out_ttllllb :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> CBool -> IO (Ptr Tensor) Source #

quantized_max_pool2d_out_ttllll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

quantized_max_pool2d_out_ttlll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

quantized_max_pool2d_out_ttll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #

quantized_max_pool2d_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #

median_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

nanmedian_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #

_mps_convolution_out_ttttllll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Int64 -> IO (Ptr Tensor) Source #

mps_convolution_backward_out_ttttttlllla :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Int64 -> Ptr (StdArray '(CBool, 3)) -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

mkldnn_convolution_out_ttttllll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Int64 -> IO (Ptr Tensor) Source #

mkldnn_rnn_layer_out_tttttttttttbllllbbbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Ptr IntArray -> Int64 -> Int64 -> Int64 -> CBool -> CBool -> CBool -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #

mkldnn_rnn_layer_backward_out_ttttttttttttttttttttblllbbblbt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> Int64 -> Int64 -> CBool -> CBool -> CBool -> Ptr IntArray -> CBool -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor))) Source #

miopen_batch_norm_out_ttttttttbdd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CDouble -> CDouble -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #

miopen_batch_norm_backward_out_ttttttttttd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #