We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents c39a471 + 7fa090b commit 22481ceCopy full SHA for 22481ce
tf2onnx/rewriter/gru_rewriter.py
@@ -120,10 +120,10 @@ def _state_variable_finder(self, context):
120
def parse_attributes(self, context):
121
# in tf, only activation of hidden gate is optional, input and update gate always use sigmoid
122
match = context.cell_match
123
- activations = ["sigmoid", "Tanh"]
+ activations = ["Sigmoid", "Tanh"]
124
if self.gru_cell_type == RNNUnitType.GRUCell:
125
activation_op = match.get_op("optional_activation")
126
- activations = ["sigmoid", activation_op.type]
+ activations = ["Sigmoid", activation_op.type]
127
context.attributes["activations"] = activations
128
return True
129
0 commit comments