Searched refs:InstructionValueSet (Results 1 – 4 of 4) sorted by relevance
98 const InstructionValueSet& GetInstructionValueSet(100 InstructionValueSet& GetInstructionValueSet(205 absl::Span<const InstructionValueSet* const> inputs);215 HloInstruction* instruction, const InstructionValueSet& new_value_set,216 const InstructionValueSet* prev_value_set = nullptr);233 std::unordered_map<const HloInstruction*, InstructionValueSet> value_sets_;
127 const InstructionValueSet& instruction_value_set = pair.second; in DeleteMarkedValues()184 absl::Span<const InstructionValueSet* const> inputs) { in Phi()189 for (const InstructionValueSet* input : inputs) { in Phi()192 for (const InstructionValueSet* input : inputs) { in Phi()210 for (const InstructionValueSet* input : inputs) { in Phi()308 const InstructionValueSet& operand_set = in UpdateBitcastValueSet()310 InstructionValueSet& bitcast_set = GetInstructionValueSet(bitcast); in UpdateBitcastValueSet()364 InstructionValueSet& value_set = GetInstructionValueSet(call); in UpdateCallValueSet()365 InstructionValueSet& root_value_set = in UpdateCallValueSet()377 std::vector<const InstructionValueSet*> inputs(conditional->branch_count()); in UpdateConditionalValueSet()[all …]
240 class InstructionValueSet : public ShapeTree<HloValueSet> {242 InstructionValueSet(const Shape& shape) : ShapeTree<HloValueSet>(shape) {} in InstructionValueSet() function246 bool AssignUnionOf(absl::Span<const InstructionValueSet* const> inputs);252 const InstructionValueSet& instruction_value_set);
255 bool InstructionValueSet::AssignUnionOf( in AssignUnionOf()256 absl::Span<const InstructionValueSet* const> inputs) { in AssignUnionOf()267 for (const InstructionValueSet* input : inputs) { in AssignUnionOf()277 const InstructionValueSet& instruction_value_set) { in operator <<()282 string InstructionValueSet::ToString() const { in ToString()