Commit e527b6b7 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[TF:XLA] Fix a few more potential int32/int64 issues, found by auditing the... 

[TF:XLA] Fix a few more potential int32/int64 issues, found by auditing the users of XlaOpKernelContext::ConstantInput.

Simplify code in a few places. For Split, the change to require only a scalar is safe because the CPU kernel requires a scalar here too.

PiperOrigin-RevId: 220797437
parent fc9c23a7
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