[TF:XLA] Fix more int32/int64 literal bugs.
Make XlaOpKernelContext::ConstantInputReshaped private, and change its users to use other methods, since its use cases can be handled other ways. Move code to copy a tensor to a literal into common literal_util module and simplify it. PiperOrigin-RevId: 220502409
Loading
Please sign in to comment