未验证 提交 9eaf5129 编写于 作者: A Artem Dementyev 提交者: GitHub

Added STRIDED_SLICE and CONCAT to simple ops in requantize_flatbuffer.py (#1994)

adding simple ops to requantize_flatbuffer.
BUG=http://b/283242131
上级 ac52be1e
......@@ -75,18 +75,20 @@ _COMPLEX_OP_REQUANTIZE_REGISTRATION = {
# List of tested simple operators (no weight and bias, e.g., reshape) see tensorflow/lite/schema/schema.fbs for op code names
_TESTED_SIMPLE_OPS = [
schema_py_generated.BuiltinOperator.RESHAPE,
schema_py_generated.BuiltinOperator.QUANTIZE,
schema_py_generated.BuiltinOperator.ADD,
schema_py_generated.BuiltinOperator.CONCATENATION,
schema_py_generated.BuiltinOperator.DEQUANTIZE,
schema_py_generated.BuiltinOperator.LEAKY_RELU,
schema_py_generated.BuiltinOperator.LOGISTIC,
schema_py_generated.BuiltinOperator.MEAN,
schema_py_generated.BuiltinOperator.SQUARED_DIFFERENCE,
schema_py_generated.BuiltinOperator.ADD,
schema_py_generated.BuiltinOperator.RSQRT,
schema_py_generated.BuiltinOperator.MUL,
schema_py_generated.BuiltinOperator.PAD,
schema_py_generated.BuiltinOperator.QUANTIZE,
schema_py_generated.BuiltinOperator.RESHAPE,
schema_py_generated.BuiltinOperator.RSQRT,
schema_py_generated.BuiltinOperator.SQUARED_DIFFERENCE,
schema_py_generated.BuiltinOperator.STRIDED_SLICE,
schema_py_generated.BuiltinOperator.SUB,
schema_py_generated.BuiltinOperator.LEAKY_RELU,
schema_py_generated.BuiltinOperator.LOGISTIC,
schema_py_generated.BuiltinOperator.PAD
]
_SUPPORTED_OPS = set(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册