(self, dag_node, op)
| 40 | """ |
| 41 | |
| 42 | def __init__(self, dag_node, op): |
| 43 | self._base_dag_node = dag_node |
| 44 | self._session = self._base_dag_node.session |
| 45 | self._op = op |
| 46 | # add op to dag |
| 47 | self._session.dag.add_op(self._op) |
| 48 | |
| 49 | |
| 50 | class UnloadedContext(DAGNode): |