Home
last modified time | relevance | path

Searched refs:ThreeFryBitGenerator (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.h50 RngOutput ThreeFryBitGenerator(XlaOp key, XlaOp initial_state,
Dprng.cc532 RngOutput ThreeFryBitGenerator(XlaOp key, XlaOp initial_state, in ThreeFryBitGenerator() function
/external/tensorflow/tensorflow/compiler/xla/service/
Drng_bit_generator_expander.cc71 output = ThreeFryBitGenerator(key_op, Slice(state_param, {1}, {2}, {1}), in GetGeneratorComputation()