Home
last modified time | relevance | path

Searched refs:HloLivenessAnalysis (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_liveness_analysis.cc68 HloLivenessAnalysis::HloIndexMap* live_index_map, in MarkLiveAtIndex()
87 HloLivenessAnalysis::HloIndexMap* live_index_map, in MarkLiveAtAllIndices()
120 HloLivenessAnalysis::HloIndexMap* live_index_map, Worklist* worklist, in PropagateLivenessThroughTuple()
149 HloLivenessAnalysis::HloIndexMap* live_index_map, Worklist* worklist, in PropagateLivenessThroughGTE()
171 HloLivenessAnalysis::HloIndexMap* live_index_map, Worklist* worklist, in PropagateLivenessThroughWhile()
196 HloLivenessAnalysis::HloIndexMap* live_index_map, Worklist* worklist, in PropagateLivenessToParameterCallers()
227 HloLivenessAnalysis::HloIndexMap* live_index_map, Worklist* worklist, in PropagateLivenessThroughControlFlow()
252 HloLivenessAnalysis::HloLivenessAnalysis(const HloModule& module) in HloLivenessAnalysis() function in xla::HloLivenessAnalysis
260 void HloLivenessAnalysis::RunAnalysis() { in RunAnalysis()
311 bool HloLivenessAnalysis::IsLive(const HloInstruction* instruction, in IsLive()
[all …]
Dhlo_liveness_analysis_test.cc39 const HloLivenessAnalysis& RunLiveness(HloModule* module) { in RunLiveness()
40 liveness_ = HloLivenessAnalysis::Run(*module).ConsumeValueOrDie(); in RunLiveness()
57 std::unique_ptr<HloLivenessAnalysis> liveness_;
70 const HloLivenessAnalysis& liveness = RunLiveness(module.get()); in TEST_F()
87 const HloLivenessAnalysis& liveness = RunLiveness(module.get()); in TEST_F()
105 const HloLivenessAnalysis& liveness = RunLiveness(module.get()); in TEST_F()
126 const HloLivenessAnalysis& liveness = RunLiveness(module.get()); in TEST_F()
152 const HloLivenessAnalysis& liveness = RunLiveness(module.get()); in TEST_F()
176 const HloLivenessAnalysis& liveness = RunLiveness(module.get()); in TEST_F()
214 const HloLivenessAnalysis& liveness = RunLiveness(module.get()); in TEST_F()
[all …]
Dhlo_liveness_analysis.h38 class HloLivenessAnalysis {
46 static StatusOr<std::unique_ptr<HloLivenessAnalysis>> Run(
55 HloLivenessAnalysis(const HloModule& module);
Dhlo_module_dce.cc39 StatusOr<bool> RunWhileDCE(HloModule* module, HloLivenessAnalysis* liveness) { in RunWhileDCE()
94 std::unique_ptr<HloLivenessAnalysis> liveness; in Run()
95 TF_ASSIGN_OR_RETURN(liveness, HloLivenessAnalysis::Run(*module)); in Run()