Home
last modified time | relevance | path

Searched refs:user_side_metadata (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_domain_verifier.cc52 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()
Dhlo_domain_map.cc122 .insert({&domain_instruction->user_side_metadata(), in PopulateDomainMetadataMap()
210 if (instruction->user_side_metadata().Kind() != domain_kind_) { in IsDomainInstruction()
Dhlo_domain_test.cc110 std::unique_ptr<DomainMetadata> user_side_metadata = in operator ()() local
114 std::move(user_side_metadata))); in operator ()()
Dhlo_instruction.h742 std::unique_ptr<DomainMetadata> user_side_metadata);
1607 const DomainMetadata& user_side_metadata() const;
Dhlo_instructions.cc2583 std::unique_ptr<DomainMetadata> user_side_metadata) in HloDomainInstruction() argument
2586 user_side_metadata_(std::move(user_side_metadata)) { in HloDomainInstruction()
2607 user_side_metadata().Matches(casted_other.user_side_metadata()); in IdenticalSlowPath()
Dhlo_instructions.h1514 std::unique_ptr<DomainMetadata> user_side_metadata);
1524 const DomainMetadata& user_side_metadata() const { in user_side_metadata() function
Dhlo_instruction.cc1327 std::unique_ptr<DomainMetadata> user_side_metadata) { in CreateDomain() argument
1330 std::move(user_side_metadata)); in CreateDomain()
3537 const DomainMetadata& HloInstruction::user_side_metadata() const { in user_side_metadata() function in xla::HloInstruction
3538 return Cast<HloDomainInstruction>(this)->user_side_metadata(); in user_side_metadata()