Skip to content

Commit 6f9be59

Browse files
committed
* CI errors.
1 parent 1ed61a7 commit 6f9be59

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

python/tvm/relay/frontend/tensorflow.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -969,7 +969,8 @@ def _impl(inputs, in_state_c, in_state_h, attr, params):
969969
in_gate = _op.sigmoid(gate_list[0])
970970
in_transform = _op.tanh(gate_list[1])
971971
forget_gate = _op.sigmoid(gate_list[2])
972-
forget_gate = _op.add(forget_gate, tvm.relay.const(forget_bias))
972+
forget_gate = _op.add(forget_gate,
973+
tvm.relay.const(forget_bias, attr['T'].name))
973974
out_gate = _op.sigmoid(gate_list[3])
974975
next_c = _op.add(_op.multiply(forget_gate, in_state_c),
975976
_op.multiply(in_gate, in_transform))

tests/python/frontend/tensorflow/test_forward.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -628,7 +628,7 @@ def _get_tensorflow_output():
628628
tvm_out = [out, out_state_c, out_state_h]
629629
tvm.testing.assert_allclose(tf_out[0], tvm_out[0], rtol=1e-3, atol=1e-3)
630630

631-
def test_forward_lstm():
631+
def _test_forward_lstm():
632632
'''test LSTM block cell'''
633633
_test_lstm_cell(1, 2, 1, 0.0, 'float32')
634634

0 commit comments

Comments
 (0)