Searched refs:uniform_state (Results 1 – 1 of 1) sorted by relevance
231 xla::RngOutput uniform_state = StatefulRngUniform( in Compile() local235 xla::XlaOp uniform = uniform_state.value; in Compile()236 state = uniform_state.state; in Compile()