Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Doperator.h209 c10::AliasAnalysisKind alias_analysis = schemaRef.aliasAnalysis(); in aliasAnalysisKind() local
304 AliasAnalysisKind alias_analysis) { in OperatorGenerator()
313 AliasAnalysisKind alias_analysis) { in OperatorGenerator()
324 AliasAnalysisKind alias_analysis) { in OperatorGenerator()
335 AliasAnalysisKind alias_analysis) { in OperatorGenerator()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_memory_scheduler.cc416 const HloAliasAnalysis& alias_analysis, in ScheduleComputationHelper()
439 const HloAliasAnalysis& alias_analysis, in DFSMemoryScheduler()
535 int64_t* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()
561 const HloAliasAnalysis& alias_analysis, in ListMemoryScheduler()
584 const HloAliasAnalysis& alias_analysis, in PostOrderMemoryScheduler()
605 const HloAliasAnalysis& alias_analysis, in DefaultMemoryScheduler()
665 const HloAliasAnalysis& alias_analysis, in DefaultModuleScheduler()
H A Dheap_simulator.cc74 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
88 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
102 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
122 const HloAliasAnalysis& alias_analysis, in Run()
143 const HloAliasAnalysis& alias_analysis, in Run()
161 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation()
H A Dcopy_insertion.cc288 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile()
346 Status AddCopiesForInPlaceOperation(const HloAliasAnalysis& alias_analysis, in AddCopiesForInPlaceOperation()
1146 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover()
1784 const HloAliasAnalysis& alias_analysis, HloInstruction* conditional) { in AddCopiesForConditional()
H A Dbuffer_assignment.h461 HloAliasAnalysis& alias_analysis() const { return *alias_analysis_; } in alias_analysis() function
504 std::unique_ptr<HloAliasAnalysis> alias_analysis, in BufferAssignment()
H A Dhlo_live_range.h110 const HloAliasAnalysis& alias_analysis, in HloLiveRange()
H A Dloop_schedule_linearizer.cc71 HloInstruction* xla_while, HloAliasAnalysis& alias_analysis) { in AddControlEdgesForLoopWrites()
H A Dmemory_space_assignment.cc136 FindCrossProgramPrefetchCandidate(const HloAliasAnalysis& alias_analysis, in FindCrossProgramPrefetchCandidate()
974 const Options& options, const HloAliasAnalysis& alias_analysis, in AlternateMemoryBestFitHeap()
3450 const HloAliasAnalysis& alias_analysis, in Run()
3466 const HloAliasAnalysis& alias_analysis) { in RunMemorySpaceAssignment()
3499 const HloAliasAnalysis& alias_analysis) { in FindAllocationSequence()
H A Dhlo_module_group_metadata.h209 HloAliasAnalysis* alias_analysis(HloModule* module) const { in alias_analysis() function
H A Dhlo_live_range.cc34 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
H A Dhlo_memory_scheduler_test.cc46 std::unique_ptr<HloAliasAnalysis> alias_analysis = in PeakMemoryUseOfEntryComputation() local
H A Dmemory_space_assignment.h212 std::unique_ptr<HloAliasAnalysis> alias_analysis, in MemorySpaceAssignmentCostAnalysis()
1261 const HloAliasAnalysis& alias_analysis() { return alias_analysis_; } in alias_analysis() function
H A Dhlo_alias_analysis.cc390 auto alias_analysis = absl::WrapUnique(new HloAliasAnalysis(module)); in Run() local
H A Dbuffer_assignment.cc1310 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignBuffersForComputations() local
1436 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignPresetBuffers() local
H A Dmemory_space_assignment_test.cc67 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpaceUsingCostAnalysis() local
163 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpace() local
273 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); local
417 std::unique_ptr<HloAliasAnalysis> alias_analysis, in FakeMemorySpaceAssignmentCostAnalysis()
H A Dbuffer_assignment_test.cc603 auto colorer = [&](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
670 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
H A Dheap_simulator_test.cc210 std::unique_ptr<HloAliasAnalysis> alias_analysis = in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
H A Dside_effect_analysis.h62 const TF::ResourceAliasAnalysis::Info& alias_analysis) in SideEffectAnalysisInfo()
71 const TF::ResourceAliasAnalysis::Info& alias_analysis) in SideEffectAnalysisInfo()
H A Dside_effect_analysis.cc70 const ResourceAliasAnalysis::Info& alias_analysis) { in GetResourceUniqueIdsOrUnknown()
350 const TF::ResourceAliasAnalysis::Info& alias_analysis) { in CollectSideEffectsByResourceId()
/aosp_15_r20/external/pytorch/torch/
H A Dlibrary.py113 def define(self, schema, alias_analysis="", *, tags=()): argument
488 def _(lib: Library, schema, alias_analysis=""): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dresource_device_inference.cc73 const TF::ResourceAliasAnalysis::Info& alias_analysis) in PerFunctionResult()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp455 const std::vector<at::Tag>& tags) { in initDispatchBindings()
/aosp_15_r20/external/pytorch/test/
H A Dtest_python_dispatch.py521 def test_helper(alias_analysis=""): argument