diff --git a/python/tools/tf_converter_lib.py b/python/tools/tf_converter_lib.py index f388a9f15f3d4b29ec7c5b3cb6a88a98c49bb9a1..fa27feb5657862e33cc794e58b192e5fcce12593 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"