From cc2908e7228debf1c150cd1be40a47379886e143 Mon Sep 17 00:00:00 2001 From: Liangliang He Date: Mon, 12 Feb 2018 15:49:37 +0800 Subject: [PATCH] Update conv2d filter from HWIO to HWOI --- python/tools/tf_converter_lib.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tools/tf_converter_lib.py b/python/tools/tf_converter_lib.py index f388a9f1..fa27feb5 100644 --- a/python/tools/tf_converter_lib.py +++ b/python/tools/tf_converter_lib.py @@ -315,6 +315,7 @@ class TFConverter(object): op_def.type = 'DepthwiseConv2d' else: op_def.type = op.type + self.transpose_filter_tensor[get_input_tensor(op, 1).name] = (0, 1, 3, 2) if self.device == 'gpu': op_def.input.extend([op.inputs[0].name]) buffer_type = "DW_CONV2D_FILTER" if op_def.type == 'DepthwiseConv2d' else "CONV2D_FILTER" -- GitLab