/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()
|