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

Fix 'uninitialized field' mSAN issue introduced in cl/347894186

Browse Source 
PiperOrigin-RevId: 348647984
Change-Id: I356d52161a8408ef410d36a2ea5d98e8d8ee9ee0
This commit is contained in:
Sachin Joglekar 2020-12-22 09:38:25 -08:00 committed by TensorFlower Gardener
parent 7d8e31edb0
commit 7f92a26eb0
1 changed files with 2 additions and 0 deletions

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

@ -284,12 +284,14 @@ void SubgraphBuilder::BuildAccumulateLoopBodySubgraph(Subgraph* subgraph) {
TfLiteAddParams* params =
reinterpret_cast<TfLiteAddParams*>(malloc(sizeof(TfLiteAddParams)));
params->activation = kTfLiteActNone;
params->pot_scale_int16 = false;
auto* add_reg = ops::builtin::Register_ADD();
add_reg->builtin_code = kTfLiteBuiltinAdd;
subgraph->AddNodeWithParameters({0, 4}, {2}, {}, nullptr, 0, params, add_reg,
&node_index);
params = reinterpret_cast<TfLiteAddParams*>(malloc(sizeof(TfLiteAddParams)));
params->activation = kTfLiteActNone;
params->pot_scale_int16 = false;
subgraph->AddNodeWithParameters({2, 1}, {3}, {}, nullptr, 0, params, add_reg,
&node_index);
}