Searched refs:ReserveRandomOutputs (Results 1 – 4 of 4) sorted by relevance
66 random::PhiloxRandom ReserveRandomOutputs(int64 output_count, in ReserveRandomOutputs() function
80 generator_.ReserveRandomOutputs(output_flat.size(), 256), in Compute()129 generator_.ReserveRandomOutputs(output_flat.size(), 256), in Compute()198 PhiloxRandom rng = generator_.ReserveRandomOutputs( in Compute()
309 random::PhiloxRandom rng = generator_.ReserveRandomOutputs( in Compute()
210 generator->ReserveRandomOutputs(batch_size * num_samples_ceil_4, 256); in DoCompute()