(
self,
origin: Module = None,
prefix: str = "",
name: str = "",
belonged_graph: weakref.ProxyTypes = None,
)
| 81 | |
| 82 | class ProxyModule(Proxy): |
| 83 | def __init__( |
| 84 | self, |
| 85 | origin: Module = None, |
| 86 | prefix: str = "", |
| 87 | name: str = "", |
| 88 | belonged_graph: weakref.ProxyTypes = None, |
| 89 | ): |
| 90 | assert not isinstance(origin, Proxy) |
| 91 | super().__init__() |
| 92 | self._oneflow_internal_graphblock__ = GraphModule( |
| 93 | prefix, name, belonged_graph, weakref.proxy(self) |
| 94 | ) |
| 95 | self._modules = OrderedDict() |
| 96 | self._parameters = OrderedDict() |
| 97 | self._buffers = OrderedDict() |
| 98 | |
| 99 | self._oneflow_internal_graphblock__set_origin(origin) |
| 100 | |
| 101 | def _oneflow_internal_graphblock__set_origin(self, origin): |
| 102 | self._oneflow_internal_origin__ = origin |
nothing calls this directly
no test coverage detected