{-# LANGUAGE DataKinds #-} {-# LANGUAGE PolyKinds #-} {-# LANGUAGE TemplateHaskell #-} {-# LANGUAGE QuasiQuotes #-} {-# LANGUAGE ScopedTypeVariables #-} {-# LANGUAGE OverloadedStrings #-} {-# LANGUAGE TypeFamilies #-} {-# LANGUAGE FlexibleInstances #-} module Torch.Internal.Managed.Type.Extra where import Foreign.C.String import Foreign.C.Types import Foreign import Torch.Internal.Type import Torch.Internal.Class import Torch.Internal.Cast import Torch.Internal.Objects import qualified Torch.Internal.Unmanaged.Type.Extra as Unmanaged tensor_assign1_l :: ForeignPtr Tensor -> Int64 -> Int64 -> IO () tensor_assign1_l :: ForeignPtr Tensor -> Int64 -> Int64 -> IO () tensor_assign1_l = forall a ca x1 cx1 x2 cx2 y cy. (Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) => (ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y _cast3 Ptr Tensor -> Int64 -> Int64 -> IO () Unmanaged.tensor_assign1_l tensor_assign2_l :: ForeignPtr Tensor -> Int64 -> Int64 -> Int64 -> IO () tensor_assign2_l :: ForeignPtr Tensor -> Int64 -> Int64 -> Int64 -> IO () tensor_assign2_l = forall a ca x1 cx1 x2 cx2 x3 cx3 y cy. (Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3, Castable y cy) => (ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y _cast4 Ptr Tensor -> Int64 -> Int64 -> Int64 -> IO () Unmanaged.tensor_assign2_l tensor_assign1_d :: ForeignPtr Tensor -> Int64 -> CDouble -> IO () tensor_assign1_d :: ForeignPtr Tensor -> Int64 -> CDouble -> IO () tensor_assign1_d = forall a ca x1 cx1 x2 cx2 y cy. (Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) => (ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y _cast3 Ptr Tensor -> Int64 -> CDouble -> IO () Unmanaged.tensor_assign1_d tensor_assign2_d :: ForeignPtr Tensor -> Int64 -> Int64 -> CDouble -> IO () tensor_assign2_d :: ForeignPtr Tensor -> Int64 -> Int64 -> CDouble -> IO () tensor_assign2_d = forall a ca x1 cx1 x2 cx2 x3 cx3 y cy. (Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3, Castable y cy) => (ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y _cast4 Ptr Tensor -> Int64 -> Int64 -> CDouble -> IO () Unmanaged.tensor_assign2_d tensor_assign1_t :: ForeignPtr Tensor -> Int64 -> ForeignPtr Tensor -> IO () tensor_assign1_t :: ForeignPtr Tensor -> Int64 -> ForeignPtr Tensor -> IO () tensor_assign1_t = forall a ca x1 cx1 x2 cx2 y cy. (Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) => (ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y _cast3 Ptr Tensor -> Int64 -> Ptr Tensor -> IO () Unmanaged.tensor_assign1_t tensor_assign2_t :: ForeignPtr Tensor -> Int64 -> Int64 -> ForeignPtr Tensor -> IO () tensor_assign2_t :: ForeignPtr Tensor -> Int64 -> Int64 -> ForeignPtr Tensor -> IO () tensor_assign2_t = forall a ca x1 cx1 x2 cx2 x3 cx3 y cy. (Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3, Castable y cy) => (ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y _cast4 Ptr Tensor -> Int64 -> Int64 -> Ptr Tensor -> IO () Unmanaged.tensor_assign2_t tensor_names :: ForeignPtr Tensor -> IO (ForeignPtr DimnameList) tensor_names :: ForeignPtr Tensor -> IO (ForeignPtr DimnameList) tensor_names = forall a ca y cy. (Castable a ca, Castable y cy) => (ca -> IO cy) -> a -> IO y _cast1 Ptr Tensor -> IO (Ptr DimnameList) Unmanaged.tensor_names tensor_to_device :: ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor) tensor_to_device :: ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor) tensor_to_device = forall a ca x1 cx1 y cy. (Castable a ca, Castable x1 cx1, Castable y cy) => (ca -> cx1 -> IO cy) -> a -> x1 -> IO y _cast2 Ptr Tensor -> Ptr Tensor -> IO (Ptr Tensor) Unmanaged.tensor_to_device new_empty_tensor :: [Int] -> ForeignPtr TensorOptions -> IO (ForeignPtr Tensor) new_empty_tensor :: [Int] -> ForeignPtr TensorOptions -> IO (ForeignPtr Tensor) new_empty_tensor = forall a ca x1 cx1 y cy. (Castable a ca, Castable x1 cx1, Castable y cy) => (ca -> cx1 -> IO cy) -> a -> x1 -> IO y _cast2 [Int] -> Ptr TensorOptions -> IO (Ptr Tensor) Unmanaged.new_empty_tensor