From 58945702e5747501999a0d0c288b7a35662f55db Mon Sep 17 00:00:00 2001 From: Menooker Date: Mon, 20 Apr 2020 09:20:39 +0800 Subject: [PATCH] style change --- src/relay/transforms/fold_scale_axis.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/relay/transforms/fold_scale_axis.cc b/src/relay/transforms/fold_scale_axis.cc index b8c5b39bd113a..d36fa960a5525 100644 --- a/src/relay/transforms/fold_scale_axis.cc +++ b/src/relay/transforms/fold_scale_axis.cc @@ -899,8 +899,8 @@ Message Conv2DBackwardPrep(const Call& call, const Array& in_messages) if (param->groups == 1 || is_depthwise_conv2d) { auto ko_small_axis = kernel_layout.IndexOf(LayoutAxis::Get('o')); auto ki_small_axis = kernel_layout.IndexOf(LayoutAxis::Get('i')); - if ( (ko_small_axis < 0 && ki_small_axis < 0 && c_small_axis < 0) || // simple layout - (ko_small_axis >= 0 && ki_small_axis >= 0 && c_small_axis >= 0)) { // blocked layout + if ( (ko_small_axis < 0 && ki_small_axis < 0 && c_small_axis < 0) || // simple layout + (ko_small_axis >= 0 && ki_small_axis >= 0 && c_small_axis >= 0)) { // blocked layout Array arr{c_big_axis}; if (c_small_axis >= 0) { arr.push_back(c_small_axis);