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:Yong Tang <yong.tang.github@outlook.com> * Update functors for int64 multiples support in `tf.tile` Signed-off-by:
Yong Tang <yong.tang.github@outlook.com> * Update test cases for int64 of multiples in `tf.tile` Signed-off-by:
Yong Tang <yong.tang.github@outlook.com> * Add GPU and non GPU tests Signed-off-by:
Yong Tang <yong.tang.github@outlook.com> * format with clang-format -i Signed-off-by:
Yong Tang <yong.tang.github@outlook.com> * Move Tmultiples after T (as it is auxilliary) And use `use_gpu=True` Signed-off-by:
Yong Tang <yong.tang.github@outlook.com>
Loading
Please sign in to comment