experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request from dev0x13:patch-2

Browse Source 
PiperOrigin-RevId: 356038426
Change-Id: I0eddfa821bff7c535bf5df286ac13760cdb6e3a2
This commit is contained in:
TensorFlower Gardener 2021-02-06 10:13:51 -08:00
commit 57bbc5e0d4
1 changed files with 3 additions and 3 deletions

6
tensorflow/lite/delegates/xnnpack/xnnpack_delegate.cc
View File

@ -1353,7 +1353,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) {
@ -1586,7 +1586,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) {
@ -2043,7 +2043,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) {