Commit 6f3cc9d3 authored by Yong Tang's avatar Yong Tang Committed by drpngx
Browse files

Pass dtype to constructor in LSTMCell (#18178)



* Use float32 in case the dtype is not set in the constructor

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add test case for 16228.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add test case where dype is passed explicitly.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Fix pylint issue

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Replace strings to objects to address review feedback.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent 17cb3cdd
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment