Searched refs:user_side_metadata (Results 1 – 7 of 7) sorted by relevance
52 TF_RET_CHECK(instruction->user_side_metadata().Kind() == in PopulateDomainKinds()55 kinds.insert(string(instruction->user_side_metadata().Kind())); in PopulateDomainKinds()98 const DomainMetadata& meta = instruction->user_side_metadata(); in VerifyDomain()
122 .insert({&domain_instruction->user_side_metadata(), in PopulateDomainMetadataMap()210 if (instruction->user_side_metadata().Kind() != domain_kind_) { in IsDomainInstruction()
110 std::unique_ptr<DomainMetadata> user_side_metadata = in operator ()() local114 std::move(user_side_metadata))); in operator ()()
742 std::unique_ptr<DomainMetadata> user_side_metadata);1607 const DomainMetadata& user_side_metadata() const;
2583 std::unique_ptr<DomainMetadata> user_side_metadata) in HloDomainInstruction() argument2586 user_side_metadata_(std::move(user_side_metadata)) { in HloDomainInstruction()2607 user_side_metadata().Matches(casted_other.user_side_metadata()); in IdenticalSlowPath()
1514 std::unique_ptr<DomainMetadata> user_side_metadata);1524 const DomainMetadata& user_side_metadata() const { in user_side_metadata() function
1327 std::unique_ptr<DomainMetadata> user_side_metadata) { in CreateDomain() argument1330 std::move(user_side_metadata)); in CreateDomain()3537 const DomainMetadata& HloInstruction::user_side_metadata() const { in user_side_metadata() function in xla::HloInstruction3538 return Cast<HloDomainInstruction>(this)->user_side_metadata(); in user_side_metadata()