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