Searched refs:HloDataflowAnalysis (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis.cc | 44 HloDataflowAnalysis::HloDataflowAnalysis( in HloDataflowAnalysis() function in xla::HloDataflowAnalysis
53 bool HloDataflowAnalysis::AreTransitiveUsesElementwiseOrTuple( in AreTransitiveUsesElementwiseOrTuple()
78 bool HloDataflowAnalysis::ValueIsDefinedAt(const HloInstruction* instruction, in ValueIsDefinedAt()
87 const HloValue& HloDataflowAnalysis::GetValueDefinedAt( in GetValueDefinedAt()
93 HloValue& HloDataflowAnalysis::GetValueDefinedAt( in GetValueDefinedAt()
99 HloValue* HloDataflowAnalysis::NewHloValue(HloInstruction* instruction, in NewHloValue()
113 void HloDataflowAnalysis::MarkValueForDeletion(HloValue::Id value_id) { in MarkValueForDeletion()
120 void HloDataflowAnalysis::DeleteMarkedValues() { in DeleteMarkedValues()
147 string HloDataflowAnalysis::ToString() const { in ToString()
182 bool HloDataflowAnalysis::Phi( in Phi()
[all …]
|
| D | hlo_alias_analysis.h | 45 const HloDataflowAnalysis::FusionCanShareBufferFunction&
85 const HloDataflowAnalysis& dataflow_analysis() const { in dataflow_analysis()
111 std::unique_ptr<HloDataflowAnalysis> dataflow_analysis_;
|
| D | hlo_ordering.h | 55 const HloDataflowAnalysis& dataflow) const;
59 const HloDataflowAnalysis& dataflow) const;
64 const HloDataflowAnalysis& dataflow) const;
|
| D | copy_insertion.h | 55 CopyInsertion(const HloDataflowAnalysis::FusionCanShareBufferFunction&
107 HloDataflowAnalysis::FusionCanShareBufferFunction fusion_can_share_buffer_;
|
| D | hlo_dataflow_analysis.h | 43 class HloDataflowAnalysis {
76 static StatusOr<std::unique_ptr<HloDataflowAnalysis>> Run(
157 HloDataflowAnalysis(
|
| D | hlo_dataflow_analysis_test.cc | 51 const HloDataflowAnalysis& RunAnalysis(bool ssa_form, in RunAnalysis()
54 HloDataflowAnalysis::Run(*module_, ssa_form, bitcast_defines_value) in RunAnalysis()
93 std::unique_ptr<HloDataflowAnalysis> analysis_;
114 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
163 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
219 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
275 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
329 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
382 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
445 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
[all …]
|
| D | hlo_ordering_test.cc | 188 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
240 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
373 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
429 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
485 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
|
| D | hlo_ordering.cc | 172 const HloDataflowAnalysis& dataflow) const { in UseIsBeforeValueDefinition()
257 const HloDataflowAnalysis& dataflow) const { in LiveRangeStrictlyBefore()
298 const HloDataflowAnalysis& dataflow) const { in MayInterfere()
|
| D | hlo_alias_analysis.cc | 63 const HloDataflowAnalysis& dataflow) in BufferValueMap()
331 const HloDataflowAnalysis& dataflow_;
491 HloModule* module, const HloDataflowAnalysis::FusionCanShareBufferFunction& in Run()
498 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true, in Run()
|
| D | bfloat16_propagation.h | 207 std::unique_ptr<HloDataflowAnalysis> dataflow_;
|
| D | copy_insertion.cc | 168 bool IndicesToCopyForWhile(const HloDataflowAnalysis& dataflow, in IndicesToCopyForWhile()
940 const HloDataflowAnalysis& dataflow_;
1241 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloDataflowAnalysis> dataflow, in AddCopiesForBufferAssignment()
1242 HloDataflowAnalysis::Run(*module)); in AddCopiesForBufferAssignment()
|
| D | bfloat16_propagation.cc | 754 TF_ASSIGN_OR_RETURN(dataflow_, HloDataflowAnalysis::Run(*module)); in Run()
|
| D | tuple_points_to_analysis.cc | 733 return HloDataflowAnalysis::AreTransitiveUsesElementwiseOrTuple( in CanShareOperandBufferWithUser()
|
| D | buffer_liveness_test.cc | 690 auto dataflow = HloDataflowAnalysis::Run(*module).ConsumeValueOrDie(); in RunWithHloDataflowAnalysis()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 393 const HloDataflowAnalysis& dataflow, const HloInstruction& param) { in FindConstrainedUses()
533 StatusOr<Literal> MakeConstrainedArgument(const HloDataflowAnalysis& dataflow, in MakeConstrainedArgument()
557 TF_ASSIGN_OR_RETURN(auto dataflow, HloDataflowAnalysis::Run(*module)); in MakeFakeArguments()
|