neuromorphs / NIR

Neuromorphic Intermediate Representation reference implementation
https://neuroir.org/docs
BSD 3-Clause "New" or "Revised" License
76 stars 10 forks source link

Suggestion: add input_type argument to nir.from_list #66

Open matjobst opened 1 year ago

matjobst commented 1 year ago

As we have some nodes that can not define their own input_type (Conv, Flatten, SumPool), the nir.from_list should have the option of giving an input_type (or input_shape) to allow the creation of fully defined graphs. The from_list would then create the ir.Input with this given shape and could infer the full shapes if wanted.