Searched refs:GeneratePhiloxBits (Results 1 – 1 of 1) sorted by relevance
390 std::pair<Philox4x32State, XlaOp> GeneratePhiloxBits(int64 num_elems, in GeneratePhiloxBits() function415 std::tie(bits, new_state) = GeneratePhiloxBits(num_elems, initial_state, key); in PhiloxRngBit32()443 GeneratePhiloxBits(num_elems * 2, initial_state, key); in PhiloxRngBit64()