Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis.h102 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_;
Dhlo_value.cc260 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()
Dhlo_dataflow_analysis.cc173 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 …]
Dhlo_value.h252 class InstructionValueSet : public ShapeTree<HloValueSet> {
254 explicit InstructionValueSet(const Shape& shape) in InstructionValueSet() function
259 bool AssignUnionOf(absl::Span<const InstructionValueSet* const> inputs);
269 const InstructionValueSet& instruction_value_set);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.h126 const InstructionValueSet& GetRootValueSet() const;
Dcpu_executable.cc408 const InstructionValueSet& CpuExecutable::GetRootValueSet() const { in GetRootValueSet()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc588 InstructionValueSet root_value_set = in GetOutputInfo()