Home
last modified time | relevance | path

Searched defs:alias_analysis (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dside_effect_analysis.h43 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in SideEffectAnalysisInfo()
49 Region* region, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in SideEffectAnalysisInfo()
Dside_effect_analysis.cc70 Operation* op, const ResourceAliasAnalysis::Info& alias_analysis) { in FindAccessedResources()
207 const ResourceAliasAnalysis::Info& alias_analysis) { in OpIsDeclaration()
219 Operation* op, const ResourceAliasAnalysis::Info& alias_analysis, in GetResourceIdsByValue()
323 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in AnalyzeFunction()
351 Region* region, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in AnalyzeRegion()
527 ResourceAliasAnalysis alias_analysis(module); in SideEffectAnalysis() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc416 const HloAliasAnalysis& alias_analysis, in ScheduleComputationHelper()
439 const HloAliasAnalysis& alias_analysis, in DFSMemoryScheduler()
526 int64* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()
551 const HloAliasAnalysis& alias_analysis, in ListMemoryScheduler()
574 const HloAliasAnalysis& alias_analysis, in PostOrderMemoryScheduler()
595 const HloAliasAnalysis& alias_analysis, in DefaultMemoryScheduler()
655 const HloAliasAnalysis& alias_analysis, in DefaultModuleScheduler()
Dheap_simulator.cc69 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
84 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
99 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
119 const HloAliasAnalysis& alias_analysis, in Run()
140 const HloAliasAnalysis& alias_analysis, in Run()
158 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation()
Dhlo_live_range.h107 const HloAliasAnalysis& alias_analysis, in HloLiveRange()
Dloop_schedule_linearizer.cc71 HloInstruction* xla_while, HloAliasAnalysis& alias_analysis) { in AddControlEdgesForLoopWrites()
Dhlo_module_group_metadata.h211 HloAliasAnalysis* alias_analysis(HloModule* module) const { in alias_analysis() function
Dbuffer_assignment.h463 HloAliasAnalysis& alias_analysis() const { return *alias_analysis_; } in alias_analysis() function
503 std::unique_ptr<HloAliasAnalysis> alias_analysis, in BufferAssignment()
Dhlo_live_range.cc25 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
Dhlo_memory_scheduler_test.cc46 std::unique_ptr<HloAliasAnalysis> alias_analysis = in PeakMemoryUseOfEntryComputation() local
Dmemory_space_assignment.h180 std::unique_ptr<HloAliasAnalysis> alias_analysis, in MemorySpaceAssignmentCostAnalysis()
1091 const Options& options, const HloAliasAnalysis& alias_analysis, in AlternateMemoryBestFitHeap()
1140 const HloAliasAnalysis& alias_analysis() { return alias_analysis_; } in alias_analysis() function
Dcopy_insertion.cc284 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile()
342 Status AddCopiesForInPlaceOperation(const HloAliasAnalysis& alias_analysis, in AddCopiesForInPlaceOperation()
1136 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover()
1747 const HloAliasAnalysis& alias_analysis, HloInstruction* conditional) { in AddCopiesForConditional()
Dhlo_alias_analysis.cc531 auto alias_analysis = absl::WrapUnique(new HloAliasAnalysis(module)); in Run() local
Dmemory_space_assignment.cc94 FindCrossProgramPrefetchCandidate(const HloAliasAnalysis& alias_analysis, in FindCrossProgramPrefetchCandidate()
2881 const HloAliasAnalysis& alias_analysis, in Run()
2897 const HloAliasAnalysis& alias_analysis) { in RunMemorySpaceAssignment()
2930 const HloAliasAnalysis& alias_analysis) { in FindAllocationSequence()
Dbuffer_assignment.cc1267 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignBuffersForComputations() local
1393 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignPresetBuffers() local
Dmemory_space_assignment_test.cc65 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpaceUsingCostAnalysis() local
158 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpace() local
259 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); local
403 std::unique_ptr<HloAliasAnalysis> alias_analysis, in FakeMemorySpaceAssignmentCostAnalysis()
Dbuffer_assignment_test.cc587 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
653 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
Dheap_simulator_test.cc211 std::unique_ptr<HloAliasAnalysis> alias_analysis = in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_device_inference.cc78 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) in PerFunctionResult()