mathflow-0.1.0.0: Dependently typed tensorflow modeler

Safe HaskellNone
LanguageHaskell2010

MathFlow.TF.NN

Documentation

avgPool' :: SingI n => String -> String -> Sing n -> String -> String -> String -> Tensor n t a Source #

avgPool :: SingI n => String -> String -> Sing n -> String -> Tensor n t a Source #

avgPool3d' :: SingI n => String -> String -> Sing n -> String -> String -> String -> Tensor n t a Source #

avgPool3d :: SingI n => String -> String -> Sing n -> String -> Tensor n t a Source #

conv2d' :: SingI n => String -> Tensor n t a -> Sing n -> String -> String -> String -> String -> Tensor n t a Source #

conv2d :: SingI n => String -> Tensor n t a -> Sing n -> String -> Tensor n t a Source #

conv2dBackpropInput' :: SingI n => String -> Tensor n t a -> String -> Sing n -> String -> String -> String -> String -> Tensor n t a Source #

conv2dBackpropInput :: SingI n => String -> Tensor n t a -> String -> Sing n -> String -> Tensor n t a Source #

conv2dTranspose' :: SingI n => String -> Tensor n t a -> String -> Sing n -> String -> String -> String -> Tensor n t a Source #

conv2dTranspose :: SingI n => String -> Tensor n t a -> String -> Sing n -> Tensor n t a Source #

conv3d' :: SingI n => String -> Tensor n t a -> Sing n -> String -> String -> String -> Tensor n t a Source #

conv3d :: SingI n => String -> Tensor n t a -> Sing n -> String -> Tensor n t a Source #

conv3dTranspose' :: SingI n => String -> Tensor n t a -> String -> Sing n -> String -> String -> String -> Tensor n t a Source #

conv3dTranspose :: SingI n => String -> Tensor n t a -> String -> Sing n -> Tensor n t a Source #

convolution' :: SingI n => String -> Tensor n t a -> String -> Sing n -> String -> String -> String -> Tensor n t a Source #

convolution :: String -> Tensor n t a -> String -> Tensor n t a Source #

crelu :: String -> Tensor n t a Source #

depthwiseConv2d' :: SingI n => String -> Tensor n t a -> Sing n -> String -> String -> String -> String -> Tensor n t a Source #

depthwiseConv2d :: SingI n => String -> Tensor n t a -> Sing n -> String -> Tensor n t a Source #

depthwiseConv2dNative' :: SingI n => String -> Tensor n t a -> Sing n -> String -> String -> String -> Tensor n t a Source #

depthwiseConv2dNative :: SingI n => String -> Tensor n t a -> Sing n -> String -> Tensor n t a Source #

dilation2d' :: SingI n => String -> Tensor n t a -> Sing n -> String -> String -> String -> Tensor n t a Source #

dilation2d :: SingI n => String -> Tensor n t a -> Sing n -> String -> String -> Tensor n t a Source #

dropout' :: Tensor n t a -> String -> String -> String -> String -> Tensor n t a Source #

dropout :: Tensor n t a -> String -> Tensor n t a Source #

elu' :: String -> String -> Tensor n t a Source #

elu :: String -> Tensor n t a Source #

erosion2d' :: SingI n => String -> String -> Sing n -> String -> String -> String -> Tensor n t a Source #

erosion2d :: SingI n => String -> String -> Sing n -> String -> String -> Tensor n t a Source #

fusedBatchNorm :: Tensor n t a -> String -> String -> Tensor n t a Source #

l2Normalize' :: Tensor n t a -> String -> String -> String -> Tensor n t a Source #

l2Normalize :: Tensor n t a -> String -> Tensor n t a Source #

lrn' :: String -> String -> String -> String -> String -> String -> Tensor n t a Source #

lrn :: String -> Tensor n t a Source #

maxPool' :: SingI n => String -> String -> Sing n -> String -> String -> String -> Tensor n t a Source #

maxPool :: SingI n => String -> String -> Sing n -> String -> Tensor n t a Source #

maxPool3d' :: SingI n => String -> String -> Sing n -> String -> String -> String -> Tensor n t a Source #

maxPool3d :: SingI n => String -> String -> Sing n -> String -> Tensor n t a Source #

moments' :: Tensor n t a -> String -> String -> String -> String -> Tensor n t a Source #

moments :: Tensor n t a -> String -> Tensor n t a Source #

pool' :: SingI n => String -> String -> String -> String -> String -> Sing n -> String -> String -> Tensor n t a Source #

pool :: String -> String -> String -> String -> Tensor n t a Source #

quantizedConv2d' :: SingI n => String -> Tensor n t a -> String -> String -> String -> String -> Sing n -> String -> String -> String -> Tensor n t a Source #

quantizedConv2d :: SingI n => String -> Tensor n t a -> String -> String -> String -> String -> Sing n -> String -> Tensor n t a Source #

relu' :: String -> String -> Tensor n t a Source #

relu :: String -> Tensor n t a Source #

relu6 :: String -> Tensor n t a Source #

reluLayer' :: Tensor n t a -> String -> String -> String -> Tensor n t a Source #

reluLayer :: Tensor n t a -> String -> String -> Tensor n t a Source #

sigmoid' :: Tensor n t a -> String -> Tensor n t a Source #

sigmoid :: Tensor n t a -> Tensor n t a Source #

tanh' :: Tensor n t a -> String -> Tensor n t a Source #

tanh :: Tensor n t a -> Tensor n t a Source #

topK' :: String -> String -> String -> String -> Tensor n t a Source #

topK :: String -> Tensor n t a Source #

xwPlusB' :: Tensor n t a -> String -> String -> String -> Tensor n t a Source #

xwPlusB :: Tensor n t a -> String -> String -> Tensor n t a Source #