Searched refs:ReserveRandomOutputs (Results 1 – 2 of 2) sorted by relevance
64 random::PhiloxRandom ReserveRandomOutputs(int64 output_count, int multiplier) { in ReserveRandomOutputs() function
118 random_generator.ReserveRandomOutputs(batch_size * sample_count_aligned, 256); in EvalFloat32()