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

Function GetComputationHloDotGraph

tensorflow/compiler/xla/python/xla.cc:119–126  ·  view source on GitHub ↗

Converts a computation to HLO dot graph form.

Source from the content-addressed store, hash-verified

117
118// Converts a computation to HLO dot graph form.
119StatusOr<std::string> GetComputationHloDotGraph(
120 const XlaComputation& computation) {
121 TF_ASSIGN_OR_RETURN(std::shared_ptr<HloModule> hlo_module,
122 GetHloModule(computation));
123 return RenderGraph(*hlo_module->entry_computation(), /*label=*/"",
124 hlo_module->config().debug_options(),
125 RenderedGraphFormat::kDot);
126}
127
128// Hashes the HLO module.
129StatusOr<uint64> HashComputation(const XlaComputation& computation) {

Callers

nothing calls this directly

Calls 5

GetHloModuleFunction · 0.85
RenderGraphFunction · 0.85
entry_computationMethod · 0.80
TF_ASSIGN_OR_RETURNFunction · 0.50
configMethod · 0.45

Tested by

no test coverage detected