Searched refs:nodedef_fn (Results 1 – 2 of 2) sorted by relevance
121 def nodedef_fn(node_def): function128 ge.transform.copy_op_handler, nodedef_fn=nodedef_fn)
131 def copy_op_handler(info, op, new_inputs, copy_shape=False, nodedef_fn=None): argument163 if nodedef_fn is not None:164 node_def_ = nodedef_fn(node_def_)