Searched refs:HloLivenessAnalysis (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_liveness_analysis.cc | 68 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 …]
|
D | hlo_liveness_analysis_test.cc | 39 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 …]
|
D | hlo_liveness_analysis.h | 38 class HloLivenessAnalysis { 46 static StatusOr<std::unique_ptr<HloLivenessAnalysis>> Run( 55 HloLivenessAnalysis(const HloModule& module);
|
D | hlo_module_dce.cc | 39 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()
|