This commit introduces type inference of FixedReshape
Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>
loco::DataType visit(const loco::FeatureDecode *node) { return loco::dtype_get(node->input()); }
loco::DataType visit(const loco::FeatureEncode *node) { return loco::dtype_get(node->input()); }
loco::DataType visit(const loco::FilterEncode *node) { return loco::dtype_get(node->input()); }
- // TODO Support FixedReshape
+ loco::DataType visit(const loco::FixedReshape *node) { return loco::dtype_get(node->input()); }
loco::DataType visit(const loco::MaxPool2D *node) { return loco::dtype_get(node->ifm()); }
loco::DataType visit(const loco::Push *node) { return loco::dtype_get(node->from()); }
loco::DataType visit(const loco::Pull *node) { return node->dtype(); }