diff --git a/mace/python/tools/converter_tool/base_converter.py b/mace/python/tools/converter_tool/base_converter.py index e6ba45f451359b508037189a20a49db64a76721d..277727d4041c97141d552740275488650ca4a3a2 100644 --- a/mace/python/tools/converter_tool/base_converter.py +++ b/mace/python/tools/converter_tool/base_converter.py @@ -210,9 +210,9 @@ class ConverterOption(object): TransformerRule.FOLD_BIASADD, TransformerRule.FOLD_ACTIVATION, TransformerRule.TRANSPOSE_FILTERS, - TransformerRule.RESHAPE_FC_WEIGHT, TransformerRule.TRANSPOSE_DATA_FORMAT, TransformerRule.TRANSFORM_GLOBAL_CONV_TO_FC, + TransformerRule.RESHAPE_FC_WEIGHT, TransformerRule.TRANSFORM_BUFFER_IMAGE, TransformerRule.ADD_DEVICE_AND_DATA_TYPE, TransformerRule.SORT_BY_EXECUTION, diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py index 14b768935444392f9b6e80bfd39a78f99d16003a..a1776b2232583a660b304d5d8245d8e1967bc82c 100644 --- a/mace/python/tools/converter_tool/transformer.py +++ b/mace/python/tools/converter_tool/transformer.py @@ -64,9 +64,9 @@ class Transformer(base_converter.ConverterInterface): TransformerRule.FOLD_BIASADD, TransformerRule.FOLD_ACTIVATION, TransformerRule.TRANSPOSE_FILTERS, - TransformerRule.RESHAPE_FC_WEIGHT, TransformerRule.TRANSPOSE_DATA_FORMAT, TransformerRule.TRANSFORM_GLOBAL_CONV_TO_FC, + TransformerRule.RESHAPE_FC_WEIGHT, TransformerRule.TRANSFORM_BUFFER_IMAGE, TransformerRule.ADD_DEVICE_AND_DATA_TYPE, TransformerRule.SORT_BY_EXECUTION, @@ -88,10 +88,10 @@ class Transformer(base_converter.ConverterInterface): TransformerRule.FOLD_BIASADD: self.fold_biasadd, TransformerRule.FOLD_ACTIVATION: self.fold_activation, TransformerRule.TRANSPOSE_FILTERS: self.transpose_filters, - TransformerRule.RESHAPE_FC_WEIGHT: self.reshape_fc_weight, TransformerRule.TRANSPOSE_DATA_FORMAT: self.transpose_data_format, TransformerRule.TRANSFORM_GLOBAL_CONV_TO_FC: self.transform_global_conv_to_fc, + TransformerRule.RESHAPE_FC_WEIGHT: self.reshape_fc_weight, TransformerRule.TRANSFORM_BUFFER_IMAGE: self.transform_buffer_image, TransformerRule.ADD_DEVICE_AND_DATA_TYPE: