Home
last modified time | relevance | path

Searched refs:reachability_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmulti_output_fusion.cc130 reachability_.reset(); in Run()
177 reachability_->Replace(base, input_fusion); in CreateFusion()
362 reachability_.reset(); in RecomputeReachability()
363 reachability_ = HloReachabilityMap::Build(computation_); in RecomputeReachability()
370 auto instr1_i = reachability_->GetIndex(instr1); in UpdateReachability()
371 auto instr2_i = reachability_->GetIndex(instr2); in UpdateReachability()
376 auto instr_i = reachability_->GetIndex(instr); in UpdateReachability()
377 bool instr2_instr = reachability_->IsReachable(instr2_i, instr_i); in UpdateReachability()
378 bool instr1_instr = reachability_->IsReachable(instr1_i, instr_i); in UpdateReachability()
384 reachability_->FastSetReachabilityToUnion({instr, instr1}, instr); in UpdateReachability()
[all …]
Dmulti_output_fusion.h95 HloReachabilityMap* reachability() const { return reachability_.get(); } in reachability()
193 return reachability_->IsConnected(instr1, instr2); in is_connected()
203 std::unique_ptr<HloReachabilityMap> reachability_; variable
Dinstruction_fusion.cc244 if (!reachability_->IsReachable(producer, consumer_operand)) { in CanFuseOnAllPaths()
492 reachability_ = HloReachabilityMap::Build(computation_); in Run()
613 reachability_.reset(); in Run()
676 if (reachability_->IsPresent(producer) && in MultiOutputFusionCreatesCycle()
677 reachability_->IsPresent(operand) && in MultiOutputFusionCreatesCycle()
678 reachability_->IsReachable(producer, operand)) { in MultiOutputFusionCreatesCycle()
Dinstruction_fusion.h136 std::unique_ptr<HloReachabilityMap> reachability_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmulti_output_fusion.cc191 reachability_ = HloReachabilityMap::Build(computation_); in RecomputeReachability()
214 if (!IsSiblingFusionCandidate(*j) || reachability_->IsConnected(*i, *j) || in FuseSiblings()
283 producer, *reachability_); in DoMultiOutputFusion()
Dmulti_output_fusion.h110 std::unique_ptr<HloReachabilityMap> reachability_; variable