Searched refs:domain_kind_ (Results 1 – 2 of 2) sorted by relevance
82 HloDomainMap(string domain_kind) : domain_kind_(std::move(domain_kind)) {} in HloDomainMap()120 string domain_kind_; variable
209 if (!domain_kind_.empty()) { in IsDomainInstruction()210 if (instruction->user_side_metadata().Kind() != domain_kind_) { in IsDomainInstruction()214 CHECK(instruction->operand_side_metadata().Kind() == domain_kind_) in IsDomainInstruction()