MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / FindComputation

Function FindComputation

tensorflow/compiler/xla/tools/interactive_graphviz.cc:136–144  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

134}
135
136HloComputation* FindComputation(const HloModule& module,
137 const string& comp_name) {
138 for (auto* computation : module.computations()) {
139 if (EqualsIgnoreCase(computation->name(), comp_name)) {
140 return computation;
141 }
142 }
143 return nullptr;
144}
145
146// Print a help message describing the various available commands.
147void DoHelpCommand() {

Callers 2

DoInfoCommandFunction · 0.85
DoPlotCommandFunction · 0.85

Calls 2

computationsMethod · 0.80
nameMethod · 0.65

Tested by

no test coverage detected