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 | 960 std::unique_ptr<DomainMetadata> operand_side_metadata,
1894 const DomainMetadata& operand_side_metadata() const;
|
| D | hlo_instructions.cc | 3020 std::unique_ptr<DomainMetadata> operand_side_metadata, in HloDomainInstruction() argument
3023 operand_side_metadata_(std::move(operand_side_metadata)), in HloDomainInstruction()
3043 return operand_side_metadata().Matches( in IdenticalSlowPath()
3044 casted_other.operand_side_metadata()) && in IdenticalSlowPath()
|
| D | hlo_instructions.h | 1821 std::unique_ptr<DomainMetadata> operand_side_metadata,
1828 const DomainMetadata& operand_side_metadata() const { in operand_side_metadata() function
|
| D | hlo_instruction.cc | 1637 std::unique_ptr<DomainMetadata> operand_side_metadata, in CreateDomain() argument
1640 shape, operand, std::move(operand_side_metadata), in CreateDomain()
4253 const DomainMetadata& HloInstruction::operand_side_metadata() const { in operand_side_metadata() function in xla::HloInstruction
4254 return Cast<HloDomainInstruction>(this)->operand_side_metadata(); in operand_side_metadata()
|