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

Fix compilation error, lhs should be used and not rhs

Browse Source
This commit is contained in:
Thibaut Goetghebuer-Planchon 2020-09-28 17:50:46 +01:00
parent e06c862dd1
commit 286195f422
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/kernels/batch_matmul.cc
View File

@ -627,7 +627,7 @@ TfLiteTensor* GetTempLhs(TfLiteContext* context, TfLiteNode* node,
return nullptr;
}
if (lhs->type == kTfLiteInt8 || rhs->type == kTfLiteInt16) {
if (lhs->type == kTfLiteInt8 || lhs->type == kTfLiteInt16) {
// Get the quantization params from the LHS tensor.
transposed_lhs->params.scale = lhs->params.scale;
transposed_lhs->params.zero_point = lhs->params.zero_point;