Safe Haskell | None |
---|---|
Language | Haskell2010 |
Torch.Internal.Unmanaged.Native.Native12
Documentation
_linalg_svd_out_ttttbbs :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> Ptr StdString -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #
_linalg_svd_out_ttttbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #
_linalg_svd_out_ttttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #
_linalg_svd_out_tttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #
linalg_svd_tbs :: Ptr Tensor -> CBool -> Ptr StdString -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #
linalg_svd_out_ttttbs :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> Ptr StdString -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #
linalg_svd_out_ttttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #
linalg_svd_out_tttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor))) Source #
linalg_pinv_out_ttttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #
linalg_pinv_out_tttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
linalg_pinv_out_ttddb :: Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> CBool -> IO (Ptr Tensor) Source #
_linalg_solve_ex_ttbb :: Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #
_linalg_solve_ex_ttb :: Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #
_linalg_solve_ex_tt :: Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #
_linalg_solve_ex_out_ttttttbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #
_linalg_solve_ex_out_ttttttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #
_linalg_solve_ex_out_tttttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor, Tensor, Tensor))) Source #
linalg_solve_ex_ttbb :: Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
linalg_solve_ex_ttb :: Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
linalg_solve_ex_out_ttttbb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
linalg_solve_ex_out_ttttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
linalg_solve_ex_out_tttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
linalg_solve_out_tttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #
linalg_tensorsolve_out_tttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr IntArray -> IO (Ptr Tensor) Source #
linalg_qr_out_ttts :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
linalg_qr_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
linalg_matrix_rank_tttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #
linalg_matrix_rank_out_ttttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #
linalg_matrix_rank_out_tttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
linalg_matrix_rank_out_ttddb :: Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> CBool -> IO (Ptr Tensor) Source #
linalg_matrix_rank_out_ttdd :: Ptr Tensor -> Ptr Tensor -> CDouble -> CDouble -> IO (Ptr Tensor) Source #
linalg_matrix_rank_out_ttdb :: Ptr Tensor -> Ptr Tensor -> CDouble -> CBool -> IO (Ptr Tensor) Source #
linalg_matrix_rank_out_tttb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CBool -> IO (Ptr Tensor) Source #
linalg_multi_dot_l :: Ptr TensorList -> IO (Ptr Tensor) Source #
linalg_multi_dot_out_tl :: Ptr Tensor -> Ptr TensorList -> IO (Ptr Tensor) Source #
_test_serialization_subcmul_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #
_test_string_default_tss :: Ptr Tensor -> Ptr StdString -> Ptr StdString -> IO (Ptr Tensor) Source #
segment_reduce_tstttlbs :: Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> Ptr Scalar -> IO (Ptr Tensor) Source #
segment_reduce_tstttlb :: Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> CBool -> IO (Ptr Tensor) Source #
segment_reduce_tstttl :: Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #
segment_reduce_tsttt :: Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
segment_reduce_tstt :: Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_segment_reduce_backward_tttsttls :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Int64 -> Ptr Scalar -> IO (Ptr Tensor) Source #
_segment_reduce_backward_tttsttl :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> Int64 -> IO (Ptr Tensor) Source #
_segment_reduce_backward_tttstt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_segment_reduce_backward_tttst :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> Ptr Tensor -> IO (Ptr Tensor) Source #
_segment_reduce_backward_ttts :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr StdString -> IO (Ptr Tensor) Source #
pad_sequence_lbd :: Ptr TensorList -> CBool -> CDouble -> IO (Ptr Tensor) Source #
pad_sequence_lb :: Ptr TensorList -> CBool -> IO (Ptr Tensor) Source #
pad_sequence_l :: Ptr TensorList -> IO (Ptr Tensor) Source #
flatten_dense_tensors_l :: Ptr TensorList -> IO (Ptr Tensor) Source #
unflatten_dense_tensors_tl :: Ptr Tensor -> Ptr TensorList -> IO (Ptr TensorList) Source #
_nested_tensor_from_tensor_list_lsLDb :: Ptr TensorList -> ScalarType -> Layout -> DeviceType -> CBool -> IO (Ptr Tensor) Source #
_nested_tensor_from_tensor_list_lsLD :: Ptr TensorList -> ScalarType -> Layout -> DeviceType -> IO (Ptr Tensor) Source #
_nested_tensor_from_tensor_list_lsL :: Ptr TensorList -> ScalarType -> Layout -> IO (Ptr Tensor) Source #
_nested_tensor_from_tensor_list_ls :: Ptr TensorList -> ScalarType -> IO (Ptr Tensor) Source #
as_strided_copy_tlll :: Ptr Tensor -> Ptr IntArray -> Ptr IntArray -> Int64 -> IO (Ptr Tensor) Source #
split_copy_tll :: Ptr Tensor -> Int64 -> Int64 -> IO (Ptr TensorList) Source #
split_copy_tl :: Ptr Tensor -> Int64 -> IO (Ptr TensorList) Source #
split_with_sizes_copy_tl :: Ptr Tensor -> Ptr IntArray -> IO (Ptr TensorList) Source #
unbind_copy_tl :: Ptr Tensor -> Int64 -> IO (Ptr TensorList) Source #
unbind_copy_t :: Ptr Tensor -> IO (Ptr TensorList) Source #
unbind_copy_out_ltl :: Ptr TensorList -> Ptr Tensor -> Int64 -> IO () Source #
unbind_copy_out_lt :: Ptr TensorList -> Ptr Tensor -> IO () Source #
split_copy_out_ltll :: Ptr TensorList -> Ptr Tensor -> Int64 -> Int64 -> IO () Source #
split_copy_out_ltl :: Ptr TensorList -> Ptr Tensor -> Int64 -> IO () Source #
split_with_sizes_copy_out_ltll :: Ptr TensorList -> Ptr Tensor -> Ptr IntArray -> Int64 -> IO () Source #
split_with_sizes_copy_out_ltl :: Ptr TensorList -> Ptr Tensor -> Ptr IntArray -> IO () Source #
view_copy_ts :: Ptr Tensor -> ScalarType -> IO (Ptr Tensor) Source #
_transformer_encoder_layer_fwd_tllttttbbdtttttttttl :: 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_tllttttbbdttttttttt :: 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_tllttttbbdtttttttt :: 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_tttlltttttbbl :: 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_tttlltttttbb :: 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_tttlltttttb :: 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_tttllttttt :: 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_tttlltttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr (StdTuple '(Tensor, Tensor))) Source #
scaled_dot_product_attention_ttttdb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> CBool -> IO (Ptr Tensor) Source #
scaled_dot_product_attention_ttttd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> IO (Ptr Tensor) Source #
scaled_dot_product_attention_tttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
scaled_dot_product_attention_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_fused_sdp_choice_ttttdb :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> CBool -> IO Int64 Source #
_fused_sdp_choice_ttttd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> IO Int64 Source #
_triton_scaled_dot_attention_tttd :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> CDouble -> IO (Ptr Tensor) Source #
_triton_scaled_dot_attention_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
_triton_multi_head_attention_tttllttttt :: 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_tttlltttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
special_chebyshev_polynomial_t_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
special_chebyshev_polynomial_t_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #
special_chebyshev_polynomial_t_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #
special_chebyshev_polynomial_u_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
special_chebyshev_polynomial_u_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #
special_chebyshev_polynomial_u_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #
special_chebyshev_polynomial_v_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #
special_chebyshev_polynomial_v_out_tst :: Ptr Tensor -> Ptr Scalar -> Ptr Tensor -> IO (Ptr Tensor) Source #
special_chebyshev_polynomial_v_out_tts :: Ptr Tensor -> Ptr Tensor -> Ptr Scalar -> IO (Ptr Tensor) Source #
special_chebyshev_polynomial_w_out_ttt :: Ptr Tensor -> Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Source #