Searched defs:num_reduces (Results 1 – 2 of 2) sorted by relevance
78 const size_t num_reduces = 32; in TEST_F() local
5001 HloComputation* fused_computation, int num_reduces) { in GroupDisjointReductions()5059 int num_reduces = fusion.getFusionResults().size(); in EmitUnnestedReduction() local