Fix issue for float16 data type with reuse in CudnnLSTM
This fix tries to address the issue raised in 18699 where
for float16 data type, the reuse in CudnnLSTM throws a ValueError.
This fix fixes the issue by passing the data type. This fix
fixes 18699.
Signed-off-by:
Yong Tang <yong.tang.github@outlook.com>
Loading
Please sign in to comment