| 87 | : computation_(computation) {} |
| 88 | |
| 89 | const CallSite* CallGraphNode::GetCallSite( |
| 90 | const HloInstruction* instruction) const { |
| 91 | auto it = callsite_instructions_.find(instruction); |
| 92 | if (it == callsite_instructions_.end()) { |
| 93 | return nullptr; |
| 94 | } |
| 95 | return &callsites_[it->second]; |
| 96 | } |
| 97 | |
| 98 | std::string CallGraphNode::ToString() const { return computation_->name(); } |
| 99 |