Function reduce_func

tests/python/common/function/test_basics.py:21–26 · view source on GitHub ↗

(nodes)

Source from the content-addressed store, hash-verified

19
20
21	def reduce_func(nodes):
22	msgs = nodes.mailbox["m"]
23	reduce_msg_shapes.add(tuple(msgs.shape))
24	assert F.ndim(msgs) == 3
25	assert F.shape(msgs)[2] == D
26	return {"accum": F.sum(msgs, 1)}
27
28
29	def apply_node_func(nodes):

nothing calls this directly

shapeMethod · 0.45

no test coverage detected