Searched refs:Philox4x32State (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 267 using Philox4x32State = std::array<XlaOp, 4>; typedef
270 Philox4x32State Philox4x32(Philox4x32State state, Philox4x32Key key) { in Philox4x32()
293 auto philox_round = [&](Philox4x32State x, Philox4x32Key key) { in Philox4x32()
296 return Philox4x32State{product1.high ^ x[1] ^ key[0], product1.low, in Philox4x32()
316 std::pair<Philox4x32State, Philox4x32Key> ScramblePhiloxKey(Philox4x32Key key) { in ScramblePhiloxKey()
321 Philox4x32State state = { in ScramblePhiloxKey()
331 return {Philox4x32State{zero, zero, state[2], state[3]}, in ScramblePhiloxKey()
377 std::pair<Philox4x32State, XlaOp> GetPhiloxInputsAndUpdatedState( in GetPhiloxInputsAndUpdatedState()
378 const Philox4x32State& state, int64 n) { in GetPhiloxInputsAndUpdatedState()
390 std::pair<Philox4x32State, XlaOp> GeneratePhiloxBits(int64 num_elems, in GeneratePhiloxBits()
[all …]
|