Searched refs:ThreeFryRngBit32 (Results 1 – 1 of 1) sorted by relevance
243 RngOutput ThreeFryRngBit32(XlaOp key, XlaOp initial_state, const Shape& shape) { in ThreeFryRngBit32() function539 return ThreeFryRngBit32(key, initial_state, shape); in ThreeFryBitGenerator()