diff options
Diffstat (limited to 'compiler/tflchef/core/src/Op/FullyConnected.cpp')
| -rw-r--r-- | compiler/tflchef/core/src/Op/FullyConnected.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/compiler/tflchef/core/src/Op/FullyConnected.cpp b/compiler/tflchef/core/src/Op/FullyConnected.cpp index 45269916c..7173a67ba 100644 --- a/compiler/tflchef/core/src/Op/FullyConnected.cpp +++ b/compiler/tflchef/core/src/Op/FullyConnected.cpp @@ -29,6 +29,7 @@ flatbuffers::Offset<void> FullyConnectedChef::value(flatbuffers::FlatBufferBuild tflite::FullyConnectedOptionsBuilder fc_options_builder{fbb}; fc_options_builder.add_fused_activation_function(tflite_activation); + fc_options_builder.add_keep_num_dims(operation.fullyconnected_options().keep_num_dims()); return fc_options_builder.Finish().Union(); } |