Searched refs:SetModuleConfig (Results 1 – 1 of 1) sorted by relevance
37 void SetModuleConfig(HloModule& module, size_t replica_count) { in SetModuleConfig() function60 SetModuleConfig(*module, /*replica_count=*/8); in TEST_F()99 SetModuleConfig(*module, /*replica_count=*/16); in TEST_F()145 SetModuleConfig(*module, /*replica_count=*/8); in TEST_F()195 SetModuleConfig(*module, /*replica_count=*/8); in TEST_F()218 SetModuleConfig(*module, /*replica_count=*/16); in TEST_F()241 SetModuleConfig(*module, /*replica_count=*/8); in TEST_F()