Searched refs:GetThreeFryInputsAndUpdatedState (Results 1 – 1 of 1) sorted by relevance
140 std::pair<ThreeFry2x32State, XlaOp> GetThreeFryInputsAndUpdatedState( in GetThreeFryInputsAndUpdatedState() function246 GetThreeFryInputsAndUpdatedState(initial_state, shape_pair.half_shape); in ThreeFryRngBit32()257 GetThreeFryInputsAndUpdatedState(initial_state, shape); in ThreeFryRngBit64()