Searched refs:HloDomainRemover (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_domain_remover.cc | 28 class HloDomainRemover::RunContext {
30 RunContext(HloModule* module, HloDomainRemover* remover) in RunContext()
41 HloDomainRemover* remover_;
44 Status HloDomainRemover::RunContext::VerifyAndNormalizeDomain( in VerifyAndNormalizeDomain()
60 StatusOr<bool> HloDomainRemover::RunContext::Run() { in Run()
101 StatusOr<bool> HloDomainRemover::Run(HloModule* module) { in Run()
|
| D | hlo_domain_test.cc | 154 HloDomainRemover remover(ShardingMetadata::KindName(), in TEST_F()
224 HloDomainRemover remover(ShardingMetadata::KindName(), in TEST_F()
276 HloDomainRemover remover(ShardingMetadata::KindName(), in TEST_F()
324 HloDomainRemover sharding_remover(ShardingMetadata::KindName(), in TEST_F()
330 HloDomainRemover opname_remover(OpNameMetadata::KindName(), in TEST_F()
410 HloDomainRemover remover(ShardingMetadata::KindName(), in TEST_F()
465 HloDomainRemover remover(ShardingMetadata::KindName(), in TEST_F()
522 HloDomainRemover remover(ShardingMetadata::KindName(), in TEST_F()
574 HloDomainRemover sharding_remover(ShardingMetadata::KindName(), in TEST_F()
580 HloDomainRemover opname_remover(OpNameMetadata::KindName(), in TEST_F()
[all …]
|
| D | hlo_domain_remover.h | 29 class HloDomainRemover : public HloModulePass {
38 HloDomainRemover(absl::string_view kind, in HloDomainRemover() function
|