Searched refs:HloDomainIsolator (Results 1 – 3 of 3) sorted by relevance
29 HloDomainIsolator::DomainCreator* creator) { in RunInternal()62 HloDomainIsolator::HloDomainIsolator(DomainCreatorFactory creator_factory) in HloDomainIsolator() function in xla::HloDomainIsolator65 StatusOr<bool> HloDomainIsolator::Run(HloModule* module) { in Run()
143 HloDomainIsolator isolator([]() { return ShardingDomainCreator{}; }); in TEST_F()186 HloDomainIsolator isolator([]() { return ShardingDomainCreator{}; }); in TEST_F()214 HloDomainIsolator isolator([]() { return ShardingDomainCreator{}; }); in TEST_F()252 HloDomainIsolator isolator([]() { return ShardingDomainCreator{}; }); in TEST_F()308 HloDomainIsolator sharding_isolator([]() { return ShardingDomainCreator{}; }); in TEST_F()313 HloDomainIsolator opname_isolator([]() { return OpNameDomainCreator{}; }); in TEST_F()364 HloDomainIsolator isolator([]() { return ShardingDomainCreator{}; }); in TEST_F()451 HloDomainIsolator isolator([]() { return ShardingDomainCreator{}; }); in TEST_F()513 HloDomainIsolator isolator([]() { return ShardingDomainCreator{}; }); in TEST_F()563 HloDomainIsolator opname_isolator([]() { return OpNameDomainCreator{}; }); in TEST_F()[all …]
33 class HloDomainIsolator : public HloModulePass {44 explicit HloDomainIsolator(DomainCreatorFactory creator_factory_);