diff --git a/src/relay/backend/interpreter.cc b/src/relay/backend/interpreter.cc index 3128d2a711595..d464f099355f5 100644 --- a/src/relay/backend/interpreter.cc +++ b/src/relay/backend/interpreter.cc @@ -10,6 +10,7 @@ #include #include #include +#include #include "compile_engine.h" namespace tvm { diff --git a/src/relay/op/nn/convolution.cc b/src/relay/op/nn/convolution.cc index b53f57464e813..9533ad69ed6d5 100644 --- a/src/relay/op/nn/convolution.cc +++ b/src/relay/op/nn/convolution.cc @@ -64,6 +64,7 @@ bool Conv2DRel(const Array& types, channels = param->channels; dilated_ksize_y = 1 + (param->kernel_size[0] - 1) * param->dilation[0]; dilated_ksize_x = 1 + (param->kernel_size[1] - 1) * param->dilation[1]; + // assign result to reporter reporter->Assign(types[1], TensorTypeNode::make(wshape, data->dtype)); } else { diff --git a/tests/python/relay/test_ir_nodes.py b/tests/python/relay/test_ir_nodes.py index e041acca0e0c4..c797ee0b4340f 100644 --- a/tests/python/relay/test_ir_nodes.py +++ b/tests/python/relay/test_ir_nodes.py @@ -209,6 +209,9 @@ def test_tuple_get_item(): str(get) check_json_roundtrip(get) +def test_op(): + add = op.op.get("add") + check_json_roundtrip(add) def test_op(): add = op.op.get("add")