All the inputs of the BATCH_MATMUL should be quantized by the type of the activation

2020-09-23 17:05:53 +01:00 · 2020-09-23 17:05:53 +01:00 · dff42b15e2
commit dff42b15e2
parent 0f082f6b40
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/lite/tools/optimize/operator_property.cc
+++ b/tensorflow/lite/tools/optimize/operator_property.cc
@ -97,6 +97,7 @@ OperatorProperty GetOperatorProperty(const ModelT* model, int subgraph_index,
      property.inputs = {{0, {}}, {1, {}}};
      property.outputs = {{0, {}}};
      property.version = 2;
+      property.quantize_input_as_activations = true;
      break;
    }
    case BuiltinOperator_BATCH_TO_SPACE_ND: