Safe Haskell | None |
---|---|
Language | Haskell2010 |
Torch.Internal.Unmanaged.Native.Native15
Documentation
uniform_out_ttddG :: Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> Ptr Generator -> IO (Ptr Tensor) Source #
cauchy_out_ttddG :: Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> Ptr Generator -> IO (Ptr Tensor) Source #
log_normal_out_ttddG :: Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> Ptr Generator -> IO (Ptr Tensor) Source #
exponential_out_ttdG :: Ptr Tensor -> Ptr Tensor -> CDouble -> Ptr Generator -> IO (Ptr Tensor) Source #
geometric_out_ttdG :: Ptr Tensor -> Ptr Tensor -> CDouble -> Ptr Generator -> IO (Ptr Tensor) Source #
_cholesky_solve_helper_out_tttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #
_histogramdd_bin_edges_out_ltlatb :: Ptr TensorList -> Ptr Tensor -> Ptr IntArray -> Ptr (StdVector CDouble) -> Ptr Tensor -> CBool -> IO () Source #
_histogramdd_bin_edges_out_ltlat :: Ptr TensorList -> Ptr Tensor -> Ptr IntArray -> Ptr (StdVector CDouble) -> Ptr Tensor -> IO () Source #
_histogramdd_bin_edges_out_ltla :: Ptr TensorList -> Ptr Tensor -> Ptr IntArray -> Ptr (StdVector CDouble) -> IO () Source #
_histogramdd_bin_edges_out_ltl :: Ptr TensorList -> Ptr Tensor -> Ptr IntArray -> IO () Source #
_histogramdd_from_bin_cts_out_ttlatb :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr (StdVector CDouble) -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #
_histogramdd_from_bin_cts_out_ttlat :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr (StdVector CDouble) -> Ptr Tensor -> IO (Ptr Tensor) Source #
_histogramdd_from_bin_cts_out_ttla :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr (StdVector CDouble) -> IO (Ptr Tensor) Source #
_histogramdd_from_bin_cts_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
_histogramdd_from_bin_tensors_out_ttltb :: Ptr Tensor -> Ptr Tensor -> Ptr TensorList -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #
_histogramdd_from_bin_tensors_out_ttlt :: Ptr Tensor -> Ptr Tensor -> Ptr TensorList -> Ptr Tensor -> IO (Ptr Tensor) Source #
_histogramdd_from_bin_tensors_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr TensorList -> IO (Ptr Tensor) Source #
argsort_out_ttblb :: Ptr Tensor -> Ptr Tensor -> CBool -> Int64 -> CBool -> IO (Ptr Tensor) Source #
unfold_backward_out_ttllll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Int64 -> Int64 -> Int64 -> IO (Ptr Tensor) Source #
normal_out_ttddG :: Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> Ptr Generator -> IO (Ptr Tensor) Source #
_amp_foreach_non_finite_check_and_unscale_out_lltt :: Ptr TensorList -> Ptr TensorList -> Ptr Tensor -> Ptr Tensor -> IO () Source #
_amp_foreach_non_finite_check_and_unscale_ltt :: Ptr TensorList -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(TensorList, Tensor))) Source #
_amp_update_scale_out_ttttddl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> Int64 -> IO (Ptr Tensor) Source #
_amp_update_scale_tttddl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> Int64 -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
_foreach_add_out_lls :: Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_sub_out_lls :: Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_mul_out_lls :: Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_div_out_lls :: Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_clamp_min_out_lls :: Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_clamp_max_out_lls :: Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_maximum_out_lls :: Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_minimum_out_lls :: Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_add_out_llls :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_add_out_lll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_sub_out_llls :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_sub_out_lll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_mul_out_lll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_div_out_lll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_clamp_min_out_lll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_clamp_max_out_lll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_maximum_out_lll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_minimum_out_lll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_add_out_llA :: Ptr TensorList -> Ptr TensorList -> Ptr (StdVector Scalar) -> IO () Source #
_foreach_sub_out_llA :: Ptr TensorList -> Ptr TensorList -> Ptr (StdVector Scalar) -> IO () Source #
_foreach_div_out_llA :: Ptr TensorList -> Ptr TensorList -> Ptr (StdVector Scalar) -> IO () Source #
_foreach_mul_out_llA :: Ptr TensorList -> Ptr TensorList -> Ptr (StdVector Scalar) -> IO () Source #
_foreach_clamp_min_out_llA :: Ptr TensorList -> Ptr TensorList -> Ptr (StdVector Scalar) -> IO () Source #
_foreach_clamp_max_out_llA :: Ptr TensorList -> Ptr TensorList -> Ptr (StdVector Scalar) -> IO () Source #
_foreach_maximum_out_llA :: Ptr TensorList -> Ptr TensorList -> Ptr (StdVector Scalar) -> IO () Source #
_foreach_minimum_out_llA :: Ptr TensorList -> Ptr TensorList -> Ptr (StdVector Scalar) -> IO () Source #
_foreach_exp_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_zero_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_zero_l :: Ptr TensorList -> IO (Ptr TensorList) Source #
_foreach_sqrt_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_abs_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_acos_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_asin_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_atan_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_ceil_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_cos_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_cosh_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_erf_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_erfc_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_expm1_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_floor_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_log_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_log10_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_log1p_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_log2_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_neg_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_tan_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_tanh_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_sin_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_sinh_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_round_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_lgamma_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_frac_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_reciprocal_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_sigmoid_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_trunc_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_addcdiv_out_lllls :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_addcdiv_out_llll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_addcmul_out_lllls :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_addcmul_out_llll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_addcdiv_out_llllA :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr (StdVector Scalar) -> IO () Source #
_foreach_addcdiv_out_llllt :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr Tensor -> IO () Source #
_foreach_addcmul_out_llllA :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr (StdVector Scalar) -> IO () Source #
_foreach_addcmul_out_llllt :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr Tensor -> IO () Source #
_foreach_norm_out_lls :: Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
_foreach_norm_out_ll :: Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_lerp_out_llll :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> IO () Source #
_foreach_lerp_out_llls :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr Scalar -> IO () Source #
bucketize_out_tstbb :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> CBool -> CBool -> IO (Ptr Tensor) Source #
searchsorted_out_ttsbbst :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> CBool -> CBool -> Ptr StdString -> Ptr Tensor -> IO (Ptr Tensor) Source #
searchsorted_out_ttsbbs :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> CBool -> CBool -> Ptr StdString -> IO (Ptr Tensor) Source #
searchsorted_out_ttsbb :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> CBool -> CBool -> IO (Ptr Tensor) Source #
searchsorted_out_ttsb :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> CBool -> IO (Ptr Tensor) Source #
glu_jvp_out_ttttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #
glu_backward_jvp_out_ttttttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #
rrelu_with_noise_backward_out_ttttssbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> Ptr Scalar -> CBool -> CBool -> IO (Ptr Tensor) Source #
mkldnn_adaptive_avg_pool2d_backward_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_adaptive_avg_pool2d_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
_adaptive_avg_pool2d_backward_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_adaptive_avg_pool3d_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
_adaptive_avg_pool3d_backward_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_slow_conv2d_backward_out_ttttttllla :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> Ptr (StdArray '(CBool, 3)) -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #
conv_depthwise3d_out_tttltlll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #
slow_conv_dilated2d_out_tttltlll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #
slow_conv_dilated2d_out_tttltll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #
slow_conv_dilated2d_out_tttltl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
slow_conv_dilated2d_out_tttlt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> IO (Ptr Tensor) Source #
slow_conv_dilated2d_out_tttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
slow_conv_dilated3d_out_tttltlll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #
slow_conv_dilated3d_out_tttltll :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #
slow_conv_dilated3d_out_tttltl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
slow_conv_dilated3d_out_tttlt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr Tensor -> IO (Ptr Tensor) Source #
slow_conv_dilated3d_out_tttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
_test_optional_intlist_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
_test_optional_filled_intlist_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
_test_optional_floatlist_out_tta :: Ptr Tensor -> Ptr Tensor -> Ptr (StdVector CDouble) -> IO (Ptr Tensor) Source #
_test_autograd_multiple_dispatch_view_copy_out_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
segment_reduce_out_ttstttlbs :: Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> Ptr Scalar -> IO (Ptr Tensor) Source #
segment_reduce_out_ttstttlb :: Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> IO (Ptr Tensor) Source #
segment_reduce_out_ttstttl :: Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #
segment_reduce_out_ttsttt :: Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
segment_reduce_out_ttstt :: Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
segment_reduce_out_ttst :: Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> IO (Ptr Tensor) Source #
_segment_reduce_backward_out_ttttsttls :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Int64 -> Ptr Scalar -> IO (Ptr Tensor) Source #
_segment_reduce_backward_out_ttttsttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #
_segment_reduce_backward_out_ttttstt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_segment_reduce_backward_out_ttttst :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> IO (Ptr Tensor) Source #
_segment_reduce_backward_out_tttts :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> IO (Ptr Tensor) Source #
_nested_tensor_from_tensor_list_out_tlsLDb :: Ptr Tensor -> Ptr TensorList -> ScalarType -> Layout -> DeviceType -> CBool -> IO (Ptr Tensor) Source #
_nested_tensor_from_tensor_list_out_tlsLD :: Ptr Tensor -> Ptr TensorList -> ScalarType -> Layout -> DeviceType -> IO (Ptr Tensor) Source #
_nested_tensor_from_tensor_list_out_tlsL :: Ptr Tensor -> Ptr TensorList -> ScalarType -> Layout -> IO (Ptr Tensor) Source #
_nested_tensor_from_tensor_list_out_tls :: Ptr Tensor -> Ptr TensorList -> ScalarType -> IO (Ptr Tensor) Source #
_nested_tensor_from_tensor_list_out_tl :: Ptr Tensor -> Ptr TensorList -> IO (Ptr Tensor) Source #
_make_dual_copy_out_tttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #
as_strided_copy_out_ttlll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Int64 -> IO (Ptr Tensor) Source #
as_strided_copy_out_ttll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #
_sparse_broadcast_to_copy_out_ttl :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
diagonal_copy_out_ttlll :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> IO (Ptr Tensor) Source #
expand_copy_out_ttlb :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> CBool -> IO (Ptr Tensor) Source #
_reshape_alias_copy_out_ttll :: Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> IO (Ptr Tensor) Source #
slice_copy_out_ttllll :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> Int64 -> IO (Ptr Tensor) Source #
slice_copy_out_ttlll :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> IO (Ptr Tensor) Source #
unfold_copy_out_ttlll :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Int64 -> IO (Ptr Tensor) Source #
to_padded_tensor_out_ttdl :: Ptr Tensor -> Ptr Tensor -> CDouble -> Ptr IntArray -> IO (Ptr Tensor) Source #
_transformer_encoder_layer_fwd_out_ttllttttbbdtttttttttl :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> CDouble -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #
_transformer_encoder_layer_fwd_out_ttllttttbbdttttttttt :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> CDouble -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_transformer_encoder_layer_fwd_out_ttllttttbbdtttttttt :: Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> CDouble -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_native_multi_head_attention_out_tttttlltttttbbl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> Int64 -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
_native_multi_head_attention_out_tttttlltttttbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
_native_multi_head_attention_out_tttttlltttttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
_native_multi_head_attention_out_tttttllttttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
_native_multi_head_attention_out_tttttlltttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
_triton_scaled_dot_attention_out_ttttd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> IO (Ptr Tensor) Source #
_triton_scaled_dot_attention_out_tttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_triton_multi_head_attention_out_ttttllttttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_triton_multi_head_attention_out_ttttlltttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_foobar_out_ttbbb :: Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> CBool -> IO (Ptr Tensor) Source #
_fused_adam_out_llllllldddddbbtt :: Ptr TensorList -> 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_out_llllllldddddbbt :: Ptr TensorList -> 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_out_llllllldddddbb :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> CBool -> IO () 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 (Ptr (StdTuple '(TensorList, TensorList, TensorList, TensorList, TensorList))) 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 (Ptr (StdTuple '(TensorList, TensorList, TensorList, TensorList, TensorList))) Source #
_fused_adam_lllllldddddbb :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> CBool -> IO (Ptr (StdTuple '(TensorList, TensorList, TensorList, TensorList, TensorList))) Source #
_fused_adamw_out_llllllldddddbbtt :: Ptr TensorList -> 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_out_llllllldddddbbt :: Ptr TensorList -> 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_out_llllllldddddbb :: Ptr TensorList -> 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 (Ptr (StdTuple '(TensorList, TensorList, TensorList, TensorList, TensorList))) 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 (Ptr (StdTuple '(TensorList, TensorList, TensorList, TensorList, TensorList))) Source #
_fused_adamw_lllllldddddbb :: Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> Ptr TensorList -> CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> CBool -> IO (Ptr (StdTuple '(TensorList, TensorList, TensorList, TensorList, TensorList))) Source #