Commit f3122887 by Leyuan Wang Committed by Tianqi Chen

correct error (#4093)

parent 9bbc98cd
...@@ -237,7 +237,8 @@ def _alter_conv2d_layout(attrs, inputs, tinfo, F): ...@@ -237,7 +237,8 @@ def _alter_conv2d_layout(attrs, inputs, tinfo, F):
if cfg.is_fallback: if cfg.is_fallback:
_get_default_config(cfg, data, kernel, strides, padding, out_dtype, is_depthwise) _get_default_config(cfg, data, kernel, strides, padding, out_dtype, is_depthwise)
ic_bn, oc_bn = cfg["tile_ic"].size[-1], cfg["tile_oc"].size[-1] ic_bn = cfg["tile_ic"].val if hasattr(cfg["tile_ic"], "val") else cfg["tile_ic"].size[-1]
oc_bn = cfg["tile_oc"].val if hasattr(cfg["tile_oc"], "val") else cfg["tile_oc"].size[-1]
new_attrs[layout_name] = 'NCHW%dc' % ic_bn new_attrs[layout_name] = 'NCHW%dc' % ic_bn
new_attrs['out_layout'] = 'NCHW%dc' % oc_bn new_attrs['out_layout'] = 'NCHW%dc' % oc_bn
......
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