Searched refs:inputs_state (Results 1 – 2 of 2) sorted by relevance
229 enum { STARTING, WAS_LIST_INPUT, WAS_SOLO_INPUT } inputs_state = STARTING; in FlattenInputs() local238 if (inputs_state == WAS_SOLO_INPUT) { in FlattenInputs()240 } else if (inputs_state == WAS_LIST_INPUT) { in FlattenInputs()244 inputs_state = WAS_LIST_INPUT; in FlattenInputs()254 if (inputs_state == WAS_SOLO_INPUT) { in FlattenInputs()256 } else if (inputs_state == WAS_LIST_INPUT) { in FlattenInputs()262 inputs_state = WAS_SOLO_INPUT; in FlattenInputs()266 if (inputs_state == STARTING) return "[]"; in FlattenInputs()267 if (inputs_state == WAS_SOLO_INPUT) { in FlattenInputs()
245 std::pair<ThreeFry2x32State, XlaOp> inputs_state = in ThreeFryRngBit32() local247 ThreeFry2x32State inputs = inputs_state.first; in ThreeFryRngBit32()250 return {result, inputs_state.second}; in ThreeFryRngBit32()256 std::pair<ThreeFry2x32State, XlaOp> inputs_state = in ThreeFryRngBit64() local258 ThreeFry2x32State inputs = inputs_state.first; in ThreeFryRngBit64()261 return {result, inputs_state.second}; in ThreeFryRngBit64()