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

Method start_callback

lite/pylite/test/test_network_device.py:190–202  ·  view source on GitHub ↗
(ios)

Source from the content-addressed store, hash-verified

188 start_checked = False
189
190 def start_callback(ios):
191 nonlocal start_checked
192 start_checked = True
193 assert len(ios) == 1
194 for key in ios:
195 io = key
196 data = ios[key].to_numpy().flatten()
197 input_data = self.input_data.flatten()
198 assert data.size == input_data.size
199 assert io.name.decode("utf-8") == "data"
200 for i in range(data.size):
201 assert data[i] == input_data[i]
202 return 0
203
204 network.set_start_callback(start_callback)
205 self.do_forward(network, 1)

Callers

nothing calls this directly

Calls 2

flattenMethod · 0.45
to_numpyMethod · 0.45

Tested by

no test coverage detected