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

Method compile

imperative/python/megengine/utils/network_node.py:171–184  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

169 return obj
170
171 def compile(self):
172 if (
173 self._opr is None
174 or len(self._opr.inputs) != len(self.inputs)
175 or any([i != j.var for i, j in zip(self._opr.inputs, self.inputs)])
176 ):
177 op = self.opdef(**self.params)
178 args = [i.var for i in self.inputs]
179 outputs = rt.invoke_op(op, args)
180 assert len(outputs) == len(self.outputs)
181 self._opr = outputs[0].owner
182 for i in range(len(self.outputs)):
183 self.outputs[i]._reset_var(outputs[i])
184 assert self.outputs[i].owner is self
185
186 def add_inp_var(self, x):
187 self.inputs.append(x)

Callers 5

_compileMethod · 0.45
make_constMethod · 0.45
make_reMethod · 0.45
__init__Method · 0.45

Calls 1

_reset_varMethod · 0.80

Tested by

no test coverage detected