Commit 690003cc authored by Yong Tang's avatar Yong Tang Committed by Vijay Vasudevan
Browse files

Add `int64` type `multiples` support for `tf.tile` (#13884)



* Add `int64` type `multiples` support for `tf.tile`

In the doc of `tf.tile` (tf.tile.__doc__) both `int32`
and `int64` are supported for `multiples`. However, the kernel
for `int64` is not registered yet.

This fix adds the support of `int64` `multiples` so that the
behavior matches the description of the docs.

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

* Update functors for int64 multiples support in `tf.tile`

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

* Update test cases for int64 of multiples in `tf.tile`

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

* Add GPU and non GPU tests

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

* format with clang-format -i

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

* Move Tmultiples after T (as it is  auxilliary)

And use `use_gpu=True`

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