Searched refs:num_reduces (Results 1 – 2 of 2) sorted by relevance
78 const size_t num_reduces = 32; in TEST_F() local103 for (size_t i = 0; i < num_reduces; ++i) { in TEST_F()118 for (size_t i = 0; i < num_reduces; ++i) { in TEST_F()
5478 HloComputation* fused_computation, int num_reduces) { in DivideOutputInstructionsIntoGroups() argument5479 CHECK_NE(0, num_reduces); in DivideOutputInstructionsIntoGroups()5480 if (num_reduces == 1) { in DivideOutputInstructionsIntoGroups()5485 num_reduces); in DivideOutputInstructionsIntoGroups()5486 for (size_t i = 0; i < num_reduces; ++i) { in DivideOutputInstructionsIntoGroups()5495 for (size_t oid = 0; oid < num_reduces; ++oid) { in DivideOutputInstructionsIntoGroups()5518 for (size_t oid = 0; oid < num_reduces; ++oid) { in DivideOutputInstructionsIntoGroups()5535 int num_reduces = 1; in EmitReductionFromOrToContiguousDimensions() local5537 num_reduces = fusion.getFusionResults().size(); in EmitReductionFromOrToContiguousDimensions()5540 bool returns_tuple = num_reduces > 1; in EmitReductionFromOrToContiguousDimensions()[all …]