Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.h61 RngOutput PhiloxBitGenerator(XlaOp key, XlaOp initial_state,
Dprng.cc553 RngOutput PhiloxBitGenerator(XlaOp key, XlaOp initial_state, in PhiloxBitGenerator() function
/external/tensorflow/tensorflow/compiler/xla/service/
Drng_bit_generator_expander.cc75 output = PhiloxBitGenerator( in GetGeneratorComputation()
Drng_expander.cc87 return PhiloxBitGenerator(key, state, shape); in GetComputationForRng()