Searched refs:sample_with_threefry (Results 1 – 1 of 1) sorted by relevance
353 auto sample_with_threefry = in Compile() local362 /*shape_input_idx=*/2, sample_with_threefry)); in Compile()386 auto sample_with_threefry = [this](Algorithm alg, xla::XlaOp state, in Compile() local395 /*shape_input_idx=*/2, sample_with_threefry)); in Compile()