diff --git a/src/finn/custom_op/maxpoolnhwc.py b/src/finn/custom_op/maxpoolnhwc.py index fb21ff9f40f4dc1cddbeb829172336db570fd976..824b37159e61dc577216bd3f14f895ee04fdeef8 100644 --- a/src/finn/custom_op/maxpoolnhwc.py +++ b/src/finn/custom_op/maxpoolnhwc.py @@ -40,10 +40,13 @@ class MaxPoolNHWC(CustomOp): return {} def make_shape_compatible_op(self): - pass + raise Exception("MaxPoolNHWC does not yet support shape inference") def infer_node_datatype(self, model): - pass + node = self.onnx_node + # data type stays the same + dtype = model.get_tensor_datatype(node.input[0]) + model.set_tensor_datatype(node.output[0], dtype) def execute_node(self, context, graph): node = self.onnx_node