Function Uint32sToUint64

tensorflow/compiler/xla/client/lib/prng.cc:130–135

Converts two uint32s to a uint64.

// Converts two uint32s to a uint64.
XlaOp Uint32sToUint64(std::array<XlaOp, 2> u32s) {
  XlaBuilder* builder = u32s[0].builder();
  return ConvertElementType(u32s[0], U64) |
         ShiftLeft(ConvertElementType(u32s[1], U64),
                   ConstantR0WithType(builder, U64, 32));
}

// Given the initial state and the request number of random numbers to be
// generated, returns the input for the random number generator and a new state.