diff --git a/src/compiler/code.jl b/src/compiler/code.jl index f2bbaf45..712ad9a8 100644 --- a/src/compiler/code.jl +++ b/src/compiler/code.jl @@ -59,6 +59,8 @@ import Lazy: groupby reifyparams(v::IVertex) = mapconst(x -> isa(x, Param) ? x.x : x, v) +# TODO: type hints for parameters + function process_type(ex) @capture(ex, type T_ fs__ end) @destruct [params = false || [], diff --git a/src/layers/affine.jl b/src/layers/affine.jl index 0d6207da..087905a8 100644 --- a/src/layers/affine.jl +++ b/src/layers/affine.jl @@ -1,7 +1,5 @@ export Affine -# TODO: type hints for parameters - @net type Affine W b