Searched refs:ComputeMaxUnrollFactor (Results 1 – 1 of 1) sorted by relevance
263 int ComputeMaxUnrollFactor(const HloInstruction* hlo) { in ComputeMaxUnrollFactor() function585 int unroll_factor = ComputeMaxUnrollFactor(fusion); in HandleFusion()999 rng, /*implements_whole_instruction=*/false, ComputeMaxUnrollFactor(rng)); in HandleRng()2249 unroll_factor = ComputeMaxUnrollFactor(&hlo); in EmitTargetElementLoop()