def GraphDef_Grad(meta_graph, targets):
"""Inject the gradient targets into GraphDef.
Parameters
----------
meta_graph : dragon_pb2.GraphDef
The definition of meta graph.
targets : list
The solving targets.
Returns
-------
None
See Also
--------
`T.grad(*args, **kwargs)`_ - How the generate gradient targets.
"""
all_pairs = set()
for target in targets:
for wrt in target.grad_wrts:
all_pairs.add((target.name, wrt))
for pair in all_pairs:
g_target = pb.GradientTarget()
g_target.cost = str(pair[0])
g_target.wrt = str(pair[1])
meta_graph.g_target.extend([g_target])
def GraphDef_Grad(graph_def, targets):
""" generate all graident targets for CC Graph """
all_pairs = set()
for target in targets:
for wrt in target.grad_wrts:
all_pairs.add((target.name, wrt))
for pair in all_pairs:
g_target = pb.GradientTarget()
g_target.cost = str(pair[0])
g_target.wrt = str(pair[1])
graph_def.g_target.extend([g_target])
