From e862dd43612322f8e7c510e4a4b466cb6beb7564 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Devansh=20Bansal/SNAP=20/SRI-Bangalore/Engineer/=EC=82=BC?= =?utf8?q?=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Tue, 4 Dec 2018 14:28:05 +0530 Subject: [PATCH] [tfltool] Correction in select operator for RNN op (#3852) This patch rectifies RNN in tflitetool in select_operator.py Signed-off-by: b.devansh --- tools/tflitefile_tool/select_operator.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py index c037472..ca6b1f5 100755 --- a/tools/tflitefile_tool/select_operator.py +++ b/tools/tflitefile_tool/select_operator.py @@ -298,12 +298,12 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t import tflite.RNNOptions if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().RNNOptions: - add_option = tflite.RNNOptions.RNNOptions() - add_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos) + rnn_option = tflite.RNNOptions.RNNOptions() + rnn_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos) tflite.RNNOptions.RNNOptionsStart(new_builder) tflite.RNNOptions.RNNOptionsAddFusedActivationFunction( - new_builder, add_option.FusedActivationFunction()) + new_builder, rnn_option.FusedActivationFunction()) return tflite.RNNOptions.RNNOptionsEnd(new_builder) # FullyConnectedOptions -- 2.7.4