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

TFLM: Fix typo in cmsis conv.cc

Browse Source 
There is no kernel_utils.cc only kernel_util.cc

PiperOrigin-RevId: 350275273
Change-Id: I847a5f35ba69c30291ecc1537ec9226c7e205a9b
This commit is contained in:
Tiezhen WANG 2021-01-05 20:46:03 -08:00 committed by TensorFlower Gardener
parent 51463f4b40
commit ca7c72bbbd
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/micro/kernels/cmsis-nn/conv.cc
View File

@ -157,7 +157,7 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
// Dynamically allocate per-channel quantization parameters. // Dynamically allocate per-channel quantization parameters.
// TODO(#42883): This allocation is done even for non-int8 cases to get around // TODO(#42883): This allocation is done even for non-int8 cases to get around
// a bug in kernel_utils.cc which incorrectly uses per_channel_output_shift in // a bug in kernel_util.cc which incorrectly uses per_channel_output_shift in
// non-int8 cases. Protect this section with a if (input->type == kTfLiteInt8) // non-int8 cases. Protect this section with a if (input->type == kTfLiteInt8)
// when the issue is fixed. // when the issue is fixed.
const int num_channels = filter->dims->data[kConvQuantizedDimension]; const int num_channels = filter->dims->data[kConvQuantizedDimension];