Searched refs:Philox4x32Key (Results 1 – 1 of 1) sorted by relevance
265 using Philox4x32Key = std::array<XlaOp, 2>; typedef270 Philox4x32State Philox4x32(Philox4x32State state, Philox4x32Key key) { in Philox4x32()293 auto philox_round = [&](Philox4x32State x, Philox4x32Key key) { in Philox4x32()301 auto raise_key = [](Philox4x32Key key) { in Philox4x32()303 return Philox4x32Key{key[0] + ConstantR0<uint32>(builder, kPhiloxW32A), in Philox4x32()316 std::pair<Philox4x32State, Philox4x32Key> ScramblePhiloxKey(Philox4x32Key key) { in ScramblePhiloxKey()332 Philox4x32Key{state[0], state[1]}}; in ScramblePhiloxKey()392 Philox4x32Key key) { in GeneratePhiloxBits()412 Philox4x32Key key = Uint64ToUint32s(op_key); in PhiloxRngBit32()439 Philox4x32Key key = Uint64ToUint32s(op_key); in PhiloxRngBit64()[all …]