Searched defs:_Module (Results 1 – 2 of 2) sorted by relevance
43 _Module = nn.Cell variable115 def retrieve_layer_by_name(model: _Module, layer_name: str):149 def retrieve_layer(model: _Module, target_layer: Union[str, _Module] = ''): argument192 def __init__(self, target_layer: _Module):
33 _Module = nn.Cell variable136 model: _Module,228 model: _Module,309 model: _Module,