Searched refs:GetPhiloxStateOp (Results 1 – 1 of 1) sorted by relevance
35 XlaOp GetPhiloxStateOp(XlaOp input_state, const Shape& state_shape) { in GetPhiloxStateOp() function76 key_op, GetPhiloxStateOp(state_param, state_shape), data_shape); in GetGeneratorComputation()