From 2e6316ac9fae0e8d556debe408133f9e5f417ad8 Mon Sep 17 00:00:00 2001 From: Eddie Yan Date: Tue, 19 Feb 2019 12:35:26 -0800 Subject: [PATCH] check in --- python/tvm/relay/op/nn/_nn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/relay/op/nn/_nn.py b/python/tvm/relay/op/nn/_nn.py index 42ddabf34b0c..5d4cda162ee3 100644 --- a/python/tvm/relay/op/nn/_nn.py +++ b/python/tvm/relay/op/nn/_nn.py @@ -313,13 +313,13 @@ def compute_contrib_conv2d_NCHWc(attrs, inputs, out_dtype, target): padding = attrs.get_int_tuple("padding") strides = attrs.get_int_tuple("strides") dilation = attrs.get_int_tuple("dilation") + data_layout = attrs.get_str("data_layout") out_layout = attrs.get_str("out_layout") out_dtype = attrs.get_str("out_dtype") out_dtype = inputs[0].dtype if out_dtype == "" else out_dtype out = topi.nn.conv2d_NCHWc(inputs[0], inputs[1], strides, padding, dilation, - out_layout, out_dtype) - + data_layout, out_layout, out_dtype) return [out] @reg.register_schedule("nn.contrib_conv2d_NCHWc")