Searched refs:operand_side_metadata (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_domain_verifier.cc | 53 instruction->operand_side_metadata().Kind()) in PopulateDomainKinds()
110 const DomainMetadata& meta = instruction->operand_side_metadata(); in VerifyDomain()
|
| D | hlo_domain_map.cc | 129 .insert({&domain_instruction->operand_side_metadata(), in PopulateDomainMetadataMap()
214 CHECK(instruction->operand_side_metadata().Kind() == domain_kind_) in IsDomainInstruction()
|
| D | hlo_domain_test.cc | 108 std::unique_ptr<DomainMetadata> operand_side_metadata = in operator ()() local
113 operand->shape(), operand, std::move(operand_side_metadata), in operator ()()
|
| D | hlo_instruction.h | 741 std::unique_ptr<DomainMetadata> operand_side_metadata,
1604 const DomainMetadata& operand_side_metadata() const;
|
| D | hlo_instructions.cc | 2582 std::unique_ptr<DomainMetadata> operand_side_metadata, in HloDomainInstruction() argument
2585 operand_side_metadata_(std::move(operand_side_metadata)), in HloDomainInstruction()
2605 return operand_side_metadata().Matches( in IdenticalSlowPath()
2606 casted_other.operand_side_metadata()) && in IdenticalSlowPath()
|
| D | hlo_instructions.h | 1513 std::unique_ptr<DomainMetadata> operand_side_metadata,
1520 const DomainMetadata& operand_side_metadata() const { in operand_side_metadata() function
|
| D | hlo_instruction.cc | 1326 std::unique_ptr<DomainMetadata> operand_side_metadata, in CreateDomain() argument
1329 shape, operand, std::move(operand_side_metadata), in CreateDomain()
3533 const DomainMetadata& HloInstruction::operand_side_metadata() const { in operand_side_metadata() function in xla::HloInstruction
3534 return Cast<HloDomainInstruction>(this)->operand_side_metadata(); in operand_side_metadata()
|