MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / _get_var

Method _get_var

imperative/python/megengine/utils/network.py:527–532  ·  view source on GitHub ↗

r"""Convert :class:`~._imperative_rt.graph.VarNode` to :class:`~.VarNode`.

(self, x)

Source from the content-addressed store, hash-verified

525 return None
526
527 def _get_var(self, x):
528 r"""Convert :class:`~._imperative_rt.graph.VarNode` to :class:`~.VarNode`."""
529 assert isinstance(x, _imperative_rt.graph.VarNode)
530 if x.id not in self.all_vars_map or self.all_vars_map[x.id].var != x:
531 self.all_vars_map[x.id] = VarNode.load(x, self._get_opr(x.owner))
532 return self.all_vars_map[x.id]
533
534
535def set_symbolic_shape(option: bool):

Callers 3

loadMethod · 0.95
_add_oprMethod · 0.95

Calls 2

_get_oprMethod · 0.95
loadMethod · 0.45

Tested by

no test coverage detected