Skip to content

Commit 7fa090b

Browse files
Capitalize sigmoid, consistent to ONNX spec
1 parent c39a471 commit 7fa090b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tf2onnx/rewriter/gru_rewriter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,10 +120,10 @@ def _state_variable_finder(self, context):
120120
def parse_attributes(self, context):
121121
# in tf, only activation of hidden gate is optional, input and update gate always use sigmoid
122122
match = context.cell_match
123-
activations = ["sigmoid", "Tanh"]
123+
activations = ["Sigmoid", "Tanh"]
124124
if self.gru_cell_type == RNNUnitType.GRUCell:
125125
activation_op = match.get_op("optional_activation")
126-
activations = ["sigmoid", activation_op.type]
126+
activations = ["Sigmoid", activation_op.type]
127127
context.attributes["activations"] = activations
128128
return True
129129

0 commit comments

Comments
 (0)