diff --git a/tensorflow/lite/delegates/xnnpack/xnnpack_delegate.cc b/tensorflow/lite/delegates/xnnpack/xnnpack_delegate.cc index 7f979ee9127..655d3dd2196 100644 --- a/tensorflow/lite/delegates/xnnpack/xnnpack_delegate.cc +++ b/tensorflow/lite/delegates/xnnpack/xnnpack_delegate.cc @@ -1335,7 +1335,7 @@ class Subgraph { } const TfLiteTensor& bias_tensor = tensors[bias_tensor_id]; TF_LITE_ENSURE_STATUS(CheckTensorFloatType( - logging_context, filter_tensor, node->inputs->data[2], node_index)); + logging_context, bias_tensor, node->inputs->data[2], node_index)); TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, bias_tensor, 1, node->inputs->data[2])); if (quasi_static_tensors.count(node->inputs->data[2]) == 0) { @@ -1526,7 +1526,7 @@ class Subgraph { } const TfLiteTensor& bias_tensor = tensors[bias_tensor_id]; TF_LITE_ENSURE_STATUS(CheckTensorFloatType( - logging_context, filter_tensor, node->inputs->data[2], node_index)); + logging_context, bias_tensor, node->inputs->data[2], node_index)); TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, bias_tensor, 1, node->inputs->data[2])); if (quasi_static_tensors.count(node->inputs->data[2]) == 0) { @@ -1983,7 +1983,7 @@ class Subgraph { const TfLiteTensor& bias_tensor = tensors[node->inputs->data[2]]; TF_LITE_ENSURE_STATUS(CheckTensorFloatType( - logging_context, filter_tensor, node->inputs->data[2], node_index)); + logging_context, bias_tensor, node->inputs->data[2], node_index)); TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, bias_tensor, 1, node->inputs->data[2])); if (quasi_static_tensors.count(node->inputs->data[2]) == 0) {