Commit c27a2ecb authored by codrut3's avatar codrut3 Committed by Vijay Vasudevan
Browse files

Allow num parameter in tf.linspace to be int64. (#13755) 

* Allow num parameter in tf.linspace to be int64.

Currently tf.linspace is defined for num int32 or int64.
However the kernel only allows int32, even though the op in
core/ops/math_ops permits int64 too.
I slightly changed the kernel to allow int64 too.
I also added tests for RangeOp and LinSpaceOp.

* Change variable names.
parent 9aadd980
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment