Commits on Apr 2, 2024

1. Support RngBitGenerator HloInstruction with single output in the HLO … …

   …-> MHLO conversion.
   
   For the rng-bit-generator operator, xla HLO instruction can have two kinds of shapes, (1) tuple(output_state, output_data), and (2) output_data. On the contrary, `mhlo::RngBitGeneratorOp` has only one shape, (output_state, output_data). This cl supports RngBitGenerator HloInstruction with single output in the HLO -> MHLO conversion.
   
   PiperOrigin-RevId: 621059534

   

   ZixuanJiang authored and tensorflower-gardener committed Apr 2, 2024
   Configuration menu

   • View commit details
   • Copy full SHA for e3b58d9
   • Browse repository at this point

   Copy the full SHA

   e3b58d9 View commit details

   Browse the repository at this point in the history