| /external/tensorflow/tensorflow/compiler/xla/service/ |
| multi_output_fusion.h | 37 // (3) reachability_: reachability map in this computation.
44 // After fusion, it updates the associated structure such as reachability_,
89 HloReachabilityMap* reachability() const { return reachability_.get(); }
134 std::unique_ptr<HloReachabilityMap> reachability_; member in class:xla::MultiOutputFusion
163 return reachability_->IsConnected(instr1, instr2);
|
| multi_output_fusion.cc | 261 reachability_ = HloReachabilityMap::Build(computation_);
272 if (reachability_->IsReachable(instr2, instr) &&
273 reachability_->IsReachable(instr1, instr)) {
277 if (reachability_->IsReachable(instr2, instr)) {
278 reachability_->FastSetReachabilityToUnion({instr, instr1}, instr);
280 if (reachability_->IsReachable(instr1, instr)) {
281 reachability_->FastSetReachabilityToUnion({instr, instr2}, instr);
|
| instruction_fusion.h | 122 std::unique_ptr<HloReachabilityMap> reachability_; member in class:xla::InstructionFusion
|
| instruction_fusion.cc | 215 if (!reachability_->IsReachable(producer, consumer_operand)) {
457 reachability_ = HloReachabilityMap::Build(computation_);
583 if (reachability_->IsPresent(producer) &&
584 reachability_->IsPresent(operand) &&
585 reachability_->IsReachable(producer, operand)) {
|
