Skip over optional bias tensors.

PiperOrigin-RevId: 284767966 Change-Id: Ib198cf71b970504dd424c0a0db487bd3103012af
hughdev · Dec 10, 2019 · 84add98 · 84add98
1 parent 3dedec0
commit 84add98
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tensorflow/lite/tools/optimize/quantize_model.cc b/tensorflow/lite/tools/optimize/quantize_model.cc
@@ -993,6 +993,9 @@ TfLiteStatus EnsureBiasScaleCompatibility(
 
       // Loop over all bias tensors.
       for (const int bias_idx : property.biases) {
+        if (op->inputs[bias_idx] == kTfLiteOptionalTensor) {
+          continue;
+        }
         TensorT* bias_tensor = subgraph->tensors[op->inputs[bias_idx]].get();
         int32_t channel_dim_size = bias_tensor->shape[0];
         if (bias_tensor->shape.size() != 1) {