Skip to content
Snippets Groups Projects
Commit bae8394b authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[CustomOp] add dt inference to MaxPoolNHWC, exception for SI

parent c2c6f108
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment