Numerical.Array.Layout.Builder

Documentation

data BatchInit v

Constructors

BatchInit

Fields

• batchInitSize :: !Int
• batchInitKV :: !(Either [v] (IntFun v))

Instances

Functor BatchInit

Defined in Numerical.Array.Layout.Builder

Methods

fmap :: (a -> b) -> BatchInit a -> BatchInit b

(<$) :: a -> BatchInit b -> BatchInit a

Show a => Show (BatchInit a)

Defined in Numerical.Array.Layout.Builder

Methods

showsPrec :: Int -> BatchInit a -> ShowS

show :: BatchInit a -> String

showList :: [BatchInit a] -> ShowS

materializeBatchMV :: (PrimMonad m, MVector mv a) => BatchInit a -> m (mv (PrimState m) a)

newtype AnyMV mv e

Constructors

AMV (forall s. mv s e)

newtype IntFun a

Constructors

IntFun (forall m. PrimMonad m => Int -> m a)

Instances

Functor IntFun

Defined in Numerical.Array.Layout.Builder

Methods

fmap :: (a -> b) -> IntFun a -> IntFun b

(<$) :: a -> IntFun b -> IntFun a

fromListBI :: [a] -> BatchInit a

fromVectorBI :: Vector v e => v e -> BatchInit e

fromMVectorBI :: MVector mv e => AnyMV mv e -> BatchInit e

class Layout form (rank :: Nat) => LayoutBuilder form (rank :: Nat) | form -> rank where

Methods

buildFormatM :: (store ~ FormatStorageRep form, Buffer store Int, Buffer store a, PrimMonad m) => Index rank -> proxy form -> a -> Maybe (BatchInit (Index rank, a)) -> m (form, BufferMut store (PrimState m) a)

Instances

(Foldable (Shape r), Traversable (Shape r), Applicative (Shape r)) => LayoutBuilder (Format Column Contiguous r rep) r

Defined in Numerical.Array.Layout.Builder

Methods

buildFormatM :: (store ~ FormatStorageRep (Format Column Contiguous r rep), Buffer store Int, Buffer store a, PrimMonad m) => Index r -> proxy (Format Column Contiguous r rep) -> a -> Maybe (BatchInit (Index r, a)) -> m (Format Column Contiguous r rep, BufferMut store (PrimState m) a)

(Foldable (Shape r), Traversable (Shape r), Applicative (Shape r)) => LayoutBuilder (Format Row Contiguous r rep) r

Defined in Numerical.Array.Layout.Builder

Methods

buildFormatM :: (store ~ FormatStorageRep (Format Row Contiguous r rep), Buffer store Int, Buffer store a, PrimMonad m) => Index r -> proxy (Format Row Contiguous r rep) -> a -> Maybe (BatchInit (Index r, a)) -> m (Format Row Contiguous r rep, BufferMut store (PrimState m) a)

Buffer rep Int => LayoutBuilder (Format DirectSparse Contiguous (S Z) rep) (S Z)

Defined in Numerical.Array.Layout.Builder

Methods

buildFormatM :: (store ~ FormatStorageRep (Format DirectSparse Contiguous (S Z) rep), Buffer store Int, Buffer store a, PrimMonad m) => Index (S Z) -> proxy (Format DirectSparse Contiguous (S Z) rep) -> a -> Maybe (BatchInit (Index (S Z), a)) -> m (Format DirectSparse Contiguous (S Z) rep, BufferMut store (PrimState m) a)

Buffer rep Int => LayoutBuilder (Format CompressedSparseRow Contiguous (S (S Z)) rep) (S (S Z))

Defined in Numerical.Array.Layout.Builder

Methods

buildFormatM :: (store ~ FormatStorageRep (Format CompressedSparseRow Contiguous (S (S Z)) rep), Buffer store Int, Buffer store a, PrimMonad m) => Index (S (S Z)) -> proxy (Format CompressedSparseRow Contiguous (S (S Z)) rep) -> a -> Maybe (BatchInit (Index (S (S Z)), a)) -> m (Format CompressedSparseRow Contiguous (S (S Z)) rep, BufferMut store (PrimState m) a)

LayoutBuilder (Format Direct Contiguous (S Z) rep) (S Z)

Defined in Numerical.Array.Layout.Builder

Methods

buildFormatM :: (store ~ FormatStorageRep (Format Direct Contiguous (S Z) rep), Buffer store Int, Buffer store a, PrimMonad m) => Index (S Z) -> proxy (Format Direct Contiguous (S Z) rep) -> a -> Maybe (BatchInit (Index (S Z), a)) -> m (Format Direct Contiguous (S Z) rep, BufferMut store (PrimState m) a)

buildFormatPure :: forall store form rank proxy m a. (LayoutBuilder form (rank :: Nat), store ~ FormatStorageRep form, Buffer store Int, Buffer store a, Monad m) => Index rank -> proxy form -> a -> Maybe (BatchInit (Index rank, a)) -> m (form, BufferPure store a)

isStrictlyMonotonicV :: Vector v e => (e -> e -> Ordering) -> v e -> Maybe Int

computeRunLengths :: (Vector v e, Eq e) => v e -> [(e, Int)]

computeStarts :: (Enum a, Ord a, Num b) => [(a, b)] -> a -> a -> [(a, b)]