Home
last modified time | relevance | path

Searched refs:get_activation (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/nn/layer/
Dcombined.py21 from .activation import get_activation, LeakyReLU
138 … self.activation = get_activation(activation) if isinstance(activation, str) else activation
229 … self.activation = get_activation(activation) if isinstance(activation, str) else activation
Dactivation.py879 def get_activation(name, prim_name=None): function
Dbasic.py33 from .activation import get_activation
313 self.activation = get_activation(activation) if isinstance(activation, str) else activation
Dthor_layer.py27 from mindspore.nn.layer.activation import get_activation
117 self.activation = get_activation(activation)
Dquant.py30 from .activation import get_activation
1448 self.activation = get_activation(activation) if isinstance(activation, str) else activation
/third_party/mindspore/tests/ut/python/pynative_mode/nn/
Dtest_activation.py32 relu = nn.get_activation('relu')
41 relu = nn.get_activation('relu')
/third_party/mindspore/mindspore/nn/probability/bnn_layers/
Ddense_variational.py19 from ...layer.activation import get_activation
62 self.activation = get_activation(activation)
/third_party/mindspore/tests/st/gnn/
Daggregator.py21 from mindspore.nn.layer.activation import get_activation
165 self.activation = get_activation(activation)
/third_party/mindspore/mindspore/parallel/nn/
Dlayers.py30 from mindspore.nn.layer.activation import get_activation
368 self.activation = get_activation(activation) if isinstance(activation, str) else activation
/third_party/mindspore/mindspore/nn/loss/
Dloss.py26 from mindspore.nn.layer.activation import get_activation
731 self.activation = get_activation(activation) if isinstance(activation, str) else activation