Commit 6c90786b by Zhao Wu Committed by Lianmin Zheng

[Relay] Fix Relay ARM CPU depthwise spatial pack schedule alter op layout issue. (#2861)

* Fix Relay ARM CPU spatial pack depthwise alter op layout issue.

* Update tune_relay_arm.py
parent a02916b5
......@@ -85,7 +85,6 @@ def compute_conv2d(attrs, inputs, out_type, target):
inputs[0], inputs[1], strides, padding,
dilation, layout, out_dtype=out_dtype)
elif layout == "NCHW" and \
kernel_layout == "OIHW" and \
get_const_int(inputs[1].shape[0]) == groups and \
get_const_int(inputs[1].shape[1]) == 1:
out = topi.nn.depthwise_conv2d_nchw(
......
......@@ -186,7 +186,7 @@ tuning_option = {
'log_filename': log_file,
'tuner': 'xgb',
'n_trial': 1000,
'n_trial': 1500,
'early_stopping': 800,
'measure_option': autotvm.measure_option(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment