Searched refs:retrieve_layer (Results 1 – 2 of 2) sorted by relevance
19 from mindspore.explainer._utils import deprecated_error, ForwardProbe, retrieve_layer, unify_inputs…106 self._saliency_cell = retrieve_layer(self._backward_model, target_layer=layer)
149 def retrieve_layer(model: _Module, target_layer: Union[str, _Module] = ''): function