[XLA] Refactor the code to generate XlaOpKernel for stateless random ops.
This is to prepare for a change to add the Philox algorithm for stateless random ops. PiperOrigin-RevId: 236798759
Loading
Please sign in to comment
This is to prepare for a change to add the Philox algorithm for stateless random ops. PiperOrigin-RevId: 236798759