Home
last modified time | relevance | path

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

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