Searched refs:ThreeFry2x32State (Results 1 – 1 of 1) sorted by relevance
46 using ThreeFry2x32State = std::array<XlaOp, 2>; typedef51 ThreeFry2x32State ThreeFry2x32(ThreeFry2x32State input, ThreeFry2x32State key) { in ThreeFry2x32()58 ThreeFry2x32State x; in ThreeFry2x32()74 auto round = [](ThreeFry2x32State v, int rotation) { in ThreeFry2x32()140 std::pair<ThreeFry2x32State, XlaOp> GetThreeFryInputsAndUpdatedState( in GetThreeFryInputsAndUpdatedState()245 std::pair<ThreeFry2x32State, XlaOp> inputs_state = in ThreeFryRngBit32()247 ThreeFry2x32State inputs = inputs_state.first; in ThreeFryRngBit32()248 ThreeFry2x32State outputs = ThreeFry2x32(inputs, Uint64ToUint32s(key)); in ThreeFryRngBit32()256 std::pair<ThreeFry2x32State, XlaOp> inputs_state = in ThreeFryRngBit64()258 ThreeFry2x32State inputs = inputs_state.first; in ThreeFryRngBit64()[all …]