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

Add uint32 and uint64 support with tf.train.batch (#18805)



* Add uint32 and uint64 support with tf.train.batch

This fix tries to address the issue raised in 18586
to have uint32 and uint64 support with tf.train.batch.

This fix add uint32 and uint64 to `CopyElementToSlice`
for the support.

This fix fixes 18586.

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

* Add test case for uint32 with tf.train.batch

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

* Add uint64 test case

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