MCPcopy Create free account
hub / github.com/alibaba/euler / _iter_body

Function _iter_body

tf_euler/python/euler_ops/sample_ops.py:30–35  ·  view source on GitHub ↗
(i, state)

Source from the content-addressed store, hash-verified

28sample_graph_label = base._LIB_OP.sample_graph_label
29
30def _iter_body(i, state):
31 y, count, n, ta = state
32 curr_result = sample_node(count[i] * n, y[i])
33 curr_result = tf.reshape(curr_result, [-1, n])
34 out_ta = ta.write(i, curr_result)
35 return i+1, (y, count, n, out_ta)
36
37
38def sample_node(count, node_type, condition=''):

Callers

nothing calls this directly

Calls 2

sample_nodeFunction · 0.85
writeMethod · 0.80

Tested by

no test coverage detected