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

Merge pull request #42314 from biagiom:fix_input_fully_connected_cmsis

Browse Source 
PiperOrigin-RevId: 333452724
Change-Id: I79fa4723efba44f6f487601914c207eca307800d
This commit is contained in:
TensorFlower Gardener 2020-09-23 23:36:08 -07:00
commit 0e7ce5d283
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/lite/micro/kernels/cmsis-nn/fully_connected.cc
View File

@ -142,7 +142,7 @@ TfLiteStatus EvalQuantizedInt8(TfLiteContext* context, TfLiteNode* node,
// The 'if' condition can be removed when null handling of bias is added to
// arm_fully_connected_s8
if (nullptr != tflite::micro::GetTensorData<int32_t>(bias)) {
RuntimeShape output_shape = tflite::micro::GetTensorShape(output);
const RuntimeShape output_shape = tflite::micro::GetTensorShape(output);
TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 2);
const int batches = output_shape.Dims(0);
const int output_depth = output_shape.Dims(1);
@ -165,9 +165,9 @@ TfLiteStatus EvalQuantizedInt8(TfLiteContext* context, TfLiteNode* node,
cmsis_nn_dims input_dims;
input_dims.n = batches;
input_dims.h = input_shape.Dims(1);
input_dims.w = input_shape.Dims(2);
input_dims.c = input_shape.Dims(3);
input_dims.h = 1;
input_dims.w = 1;
input_dims.c = accum_depth;
cmsis_nn_dims filter_dims;
filter_dims.n = accum_depth;