Searched refs:InstructionValueSet (Results 1 – 7 of 7) sorted by relevance
102 const InstructionValueSet& GetInstructionValueSet(104 InstructionValueSet& GetInstructionValueSet(245 absl::Span<const InstructionValueSet* const> inputs);255 HloInstruction* instruction, const InstructionValueSet& new_value_set,256 const InstructionValueSet* prev_value_set = nullptr);273 std::unordered_map<const HloInstruction*, InstructionValueSet> value_sets_;
260 bool InstructionValueSet::IsAmbiguous() const { in IsAmbiguous()268 bool InstructionValueSet::AssignUnionOf( in AssignUnionOf()269 absl::Span<const InstructionValueSet* const> inputs) { in AssignUnionOf()280 for (const InstructionValueSet* input : inputs) { in AssignUnionOf()290 const InstructionValueSet& instruction_value_set) { in operator <<()295 string InstructionValueSet::ToString() const { in ToString()
173 const InstructionValueSet& instruction_value_set = pair.second; in DeleteMarkedValues()230 absl::Span<const InstructionValueSet* const> inputs) { in Phi()235 for (const InstructionValueSet* input : inputs) { in Phi()240 absl::c_for_each(inputs, [&](const InstructionValueSet* input) { in Phi()247 absl::c_for_each(inputs, [&](const InstructionValueSet* input) { in Phi()267 for (const InstructionValueSet* input : inputs) { in Phi()355 const InstructionValueSet& value_set_tree = in GetFlattenedValueSet()389 const InstructionValueSet& operand_set = in UpdateBitcastValueSet()391 InstructionValueSet& bitcast_set = GetInstructionValueSet(bitcast); in UpdateBitcastValueSet()402 const InstructionValueSet& operand_set = in UpdateSetDimensionSizeValueSet()[all …]
252 class InstructionValueSet : public ShapeTree<HloValueSet> {254 explicit InstructionValueSet(const Shape& shape) in InstructionValueSet() function259 bool AssignUnionOf(absl::Span<const InstructionValueSet* const> inputs);269 const InstructionValueSet& instruction_value_set);
126 const InstructionValueSet& GetRootValueSet() const;
408 const InstructionValueSet& CpuExecutable::GetRootValueSet() const { in GetRootValueSet()
588 InstructionValueSet root_value_set = in GetOutputInfo()