| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis.cc | 89 HloDataflowAnalysis::HloDataflowAnalysis(const HloModule& module, bool ssa_form, in HloDataflowAnalysis() function in xla::HloDataflowAnalysis
98 bool HloDataflowAnalysis::AreTransitiveUsesElementwiseOrTuple( in AreTransitiveUsesElementwiseOrTuple()
123 bool HloDataflowAnalysis::ValueIsDefinedAt(const HloInstruction* instruction, in ValueIsDefinedAt()
132 const HloValue& HloDataflowAnalysis::GetValueDefinedAt( in GetValueDefinedAt()
138 HloValue& HloDataflowAnalysis::GetValueDefinedAt( in GetValueDefinedAt()
144 HloValue* HloDataflowAnalysis::NewHloValue(HloInstruction* instruction, in NewHloValue()
158 void HloDataflowAnalysis::MarkValueForDeletion(HloValue::Id value_id) { in MarkValueForDeletion()
165 void HloDataflowAnalysis::DeleteMarkedValues() { in DeleteMarkedValues()
193 string HloDataflowAnalysis::ToString() const { in ToString()
228 bool HloDataflowAnalysis::Phi( in Phi()
[all …]
|
| D | hlo_alias_analysis.h | 45 const HloDataflowAnalysis::CanShareBuffer& can_share_buffer = nullptr);
84 HloDataflowAnalysis& dataflow_analysis() const { return *dataflow_analysis_; } in dataflow_analysis()
135 std::unique_ptr<HloDataflowAnalysis> dataflow_analysis_;
|
| D | loop_schedule_linearizer.h | 40 const HloDataflowAnalysis::CanShareBuffer& can_share_buffer = nullptr)
48 HloDataflowAnalysis::CanShareBuffer can_share_buffer_;
|
| D | hlo_ordering.h | 76 const HloDataflowAnalysis& dataflow) const;
80 const HloDataflowAnalysis& dataflow) const;
85 const HloDataflowAnalysis& dataflow) const;
|
| D | copy_insertion.h | 56 const HloDataflowAnalysis::CanShareBuffer& can_share_buffer = nullptr)
92 HloDataflowAnalysis::CanShareBuffer can_share_buffer_;
|
| D | hlo_dataflow_analysis.h | 47 class HloDataflowAnalysis {
80 static StatusOr<std::unique_ptr<HloDataflowAnalysis>> Run(
176 HloDataflowAnalysis(const HloModule& module, bool ssa_form,
|
| D | hlo_ordering_test.cc | 187 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
239 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
379 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
435 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
491 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
526 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true)); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 52 const HloDataflowAnalysis& RunAnalysis(bool ssa_form, in RunAnalysis()
57 HloDataflowAnalysis::Run(*module_, ssa_form, bitcast_defines_value) in RunAnalysis()
96 std::unique_ptr<HloDataflowAnalysis> analysis_;
117 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
166 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
222 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
278 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
352 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
428 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
534 const HloDataflowAnalysis& analysis = RunAnalysis(ssa_form); in TEST_P()
[all …]
|
| D | hlo_alias_analysis.cc | 63 const HloDataflowAnalysis& dataflow) in BufferValueMap()
318 HloDataflowAnalysis::GetInPlaceInputOutputPairs(instruction)) { in ComputeInPlaceOperationAliasedBuffers()
333 HloDataflowAnalysis::GetInPlaceInputOutputPairs(use.instruction)) { in ComputeInPlaceOperationAliasedBuffers()
366 const HloDataflowAnalysis& dataflow_;
527 const HloDataflowAnalysis::CanShareBuffer& can_share_buffer) { in Run()
533 HloDataflowAnalysis::Run(*module, /*ssa_form=*/true, in Run()
|
| D | memory_space_propagation.h | 40 std::unique_ptr<HloDataflowAnalysis> dataflow_analysis_;
|
| D | hlo_ordering.cc | 214 const HloDataflowAnalysis& dataflow) const { in UsesBeforeValueDefinition()
379 const HloDataflowAnalysis& dataflow) const { in LiveRangeStrictlyBefore()
435 const HloDataflowAnalysis& dataflow) const { in MayInterfere()
|
| D | bfloat16_propagation.h | 212 std::unique_ptr<HloDataflowAnalysis> dataflow_;
|
| D | buffer_assignment.h | 457 const HloDataflowAnalysis& dataflow_analysis() const { in dataflow_analysis()
619 HloDataflowAnalysis::CanShareBuffer can_share_buffer = nullptr,
637 HloDataflowAnalysis::CanShareBuffer can_share_buffer);
|
| D | memory_space_propagation.cc | 23 HloDataflowAnalysis::Run(*module)); in Run()
|
| D | loop_schedule_linearizer.cc | 72 HloDataflowAnalysis& dataflow = alias_analysis.dataflow_analysis(); in AddControlEdgesForLoopWrites()
|
| D | multi_output_fusion.cc | 346 HloDataflowAnalysis::GetInPlaceInputOutputPairs(instr1)) { in LegalToFuseMainConstraints()
350 HloDataflowAnalysis::GetInPlaceInputOutputPairs(instr2)) { in LegalToFuseMainConstraints()
|
| D | copy_insertion.cc | 168 bool IndicesToCopyForWhile(const HloDataflowAnalysis& dataflow, in IndicesToCopyForWhile()
196 bool IndicesToCopyForConditional(const HloDataflowAnalysis& dataflow, in IndicesToCopyForConditional()
958 const HloDataflowAnalysis& dataflow_;
1027 HloDataflowAnalysis::GetInPlaceInputOutputPairs(instruction)) { in AddCopiesToResolveInterference()
|
| D | bfloat16_conversion_folding.cc | 176 HloDataflowAnalysis::IsInPlaceOperation(hlo->opcode()) || // in DefaultAction()
|
| D | bfloat16_propagation.cc | 612 HloDataflowAnalysis::GetInPlaceInputOutputPairs(hlo)) { in ResolveInconsistencyOfAliasingBuffersHelper()
874 TF_ASSIGN_OR_RETURN(dataflow_, HloDataflowAnalysis::Run(*module)); in Run()
|
| D | buffer_assignment.cc | 875 const HloDataflowAnalysis& dataflow = this->dataflow_analysis(); in ToProto()
915 HloDataflowAnalysis::CanShareBuffer can_share_buffer, in Run()
1629 HloDataflowAnalysis::CanShareBuffer can_share_buffer) { in CreateAssignment()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_copy_insertion.h | 32 const HloDataflowAnalysis::CanShareBuffer& can_share_buffer = nullptr)
39 HloDataflowAnalysis::CanShareBuffer can_share_buffer_;
|
| D | nvptx_compiler.h | 49 HloDataflowAnalysis::CanShareBuffer GetCanShareBuffer() override;
|
| D | gpu_compiler.h | 78 virtual HloDataflowAnalysis::CanShareBuffer GetCanShareBuffer() { in GetCanShareBuffer()
|
| D | nvptx_compiler.cc | 283 HloDataflowAnalysis::CanShareBuffer NVPTXCompiler::GetCanShareBuffer() { in GetCanShareBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 502 const HloDataflowAnalysis& dataflow, const HloInstruction& param) { in FindConstrainedUses()
653 StatusOr<Literal> MakeConstrainedArgument(const HloDataflowAnalysis& dataflow, in MakeConstrainedArgument()
684 TF_ASSIGN_OR_RETURN(auto dataflow, HloDataflowAnalysis::Run(*module)); in MakeFakeArguments()
|