Safe Haskell | None |
---|---|
Language | Haskell2010 |
Documentation
convolution :: (MatchKVList kvs '["stride" := String, "dilate" := String, "pad" := String, "num_group" := Int, "workspace" := Int, "no_bias" := Bool, "cudnn_tune" := String, "cudnn_off" := Bool, "layout" := String], ShowKV kvs) => String -> SymbolHandle -> [Int] -> Int -> HMap kvs -> IO SymbolHandle Source #
fullyConnected :: (MatchKVList kvs '["no_bias" := Bool, "flatten" := Bool], ShowKV kvs) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle Source #