Home
last modified time | relevance | path

Searched refs:ScramblePhiloxKey (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.h64 std::pair<XlaOp, XlaOp> ScramblePhiloxKey(XlaOp key);
Dprng.cc316 std::pair<Philox4x32State, Philox4x32Key> ScramblePhiloxKey(Philox4x32Key key) { in ScramblePhiloxKey() function
574 std::pair<XlaOp, XlaOp> ScramblePhiloxKey(XlaOp key) { in ScramblePhiloxKey() function
576 auto state_key = ScramblePhiloxKey(pkey); in ScramblePhiloxKey()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateless_random_ops.cc46 std::tie(state, key) = xla::ScramblePhiloxKey(key); in GetBitGeneratorForDevice()
Dstateless_random_ops_v2.cc88 auto counter_key = xla::ScramblePhiloxKey(key); in GetKeyCounter()