diff --git a/tools/python/transform/tensorflow_converter.py b/tools/python/transform/tensorflow_converter.py index 7d4057fc0e9dc4fdba9e5950c763ab0846db72f0..7f940c1d9104b02440341cebf3a319f65a22969b 100644 --- a/tools/python/transform/tensorflow_converter.py +++ b/tools/python/transform/tensorflow_converter.py @@ -1007,11 +1007,11 @@ class TensorflowConverter(base_converter.ConverterInterface): if dtype == tf.int32: op.output_type.extend([mace_pb2.DT_INT32]) elif dtype == tf.float32: - op.output_type.extend([self._option.data_type]) + op.output_type.extend([mace_pb2.DT_FLOAT]) else: mace_check(False, "data type %s not supported" % dtype) except ValueError: - op.output_type.extend([self._option.data_type]) + op.output_type.extend([mace_pb2.DT_FLOAT]) def convert_argmax(self, tf_op): op = self.convert_general_op(tf_op)