Searched refs:domain_kind (Results 1 – 2 of 2) sorted by relevance
45 HloComputation* computation, string domain_kind);52 string domain_kind);82 HloDomainMap(string domain_kind) : domain_kind_(std::move(domain_kind)) {} in HloDomainMap() argument
30 HloComputation* computation, string domain_kind) { in Create() argument31 auto domain_map = absl::WrapUnique(new HloDomainMap(std::move(domain_kind))); in Create()37 HloModule* module, string domain_kind) { in Create() argument38 auto domain_map = absl::WrapUnique(new HloDomainMap(std::move(domain_kind))); in Create()