Searched refs:input_u64 (Results 1 – 1 of 1) sorted by relevance
145 auto input_u64 = Broadcast(Reshape(initial_state, {}), shape.dimensions()); in GetThreeFryInputsAndUpdatedState() local151 input_u64 = in GetThreeFryInputsAndUpdatedState()152 input_u64 + (Iota(builder, u64_shape, i) * in GetThreeFryInputsAndUpdatedState()158 return std::make_pair(Uint64ToUint32s(input_u64), new_state); in GetThreeFryInputsAndUpdatedState()