Searched refs:drop_outputs (Results 1 – 5 of 5) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/contrib/rnn/ |
| H A D | rnn_cell.py | 51 def __init__(self, base_cell, drop_inputs=0., drop_states=0., drop_outputs=0.): argument
62 self.drop_outputs = drop_outputs
86 if self.drop_outputs and self.drop_outputs_mask is None:
88 p=self.drop_outputs)
106 if self.drop_outputs:
186 if self.drop_outputs:
187 outputs = F.Dropout(outputs, p=self.drop_outputs, axes=(axis,))
325 def dynamic_unroll(cell, inputs, begin_state, drop_inputs=0, drop_outputs=0, argument
424 if drop_outputs:
425 outputs = F.Dropout(outputs, p=drop_outputs, axes=(axis,))
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/contrib/rnn/ |
| H A D | rnn_cell.py | 51 def __init__(self, base_cell, drop_inputs=0., drop_states=0., drop_outputs=0.): argument
62 self.drop_outputs = drop_outputs
86 if self.drop_outputs and self.drop_outputs_mask is None:
88 p=self.drop_outputs)
106 if self.drop_outputs:
186 if self.drop_outputs:
187 outputs = F.Dropout(outputs, p=self.drop_outputs, axes=(axis,))
325 def dynamic_unroll(cell, inputs, begin_state, drop_inputs=0, drop_outputs=0, argument
424 if drop_outputs:
425 outputs = F.Dropout(outputs, p=drop_outputs, axes=(axis,))
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
| H A D | test_gluon_contrib.py | 134 def check_vardrop(drop_inputs, drop_states, drop_outputs): argument
136 drop_outputs=drop_outputs,
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
| H A D | test_gluon_contrib.py | 134 def check_vardrop(drop_inputs, drop_states, drop_outputs): argument
136 drop_outputs=drop_outputs,
|
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | partial_eval.py | 206 jaxpr = _dce_open_jaxpr(jaxpr, out_unknowns, drop_outputs=True)
807 jaxpr_known_nores = _dce_jaxpr(jaxpr_known, out_knowns + [False] * num_res, drop_outputs=True)
822 new_jaxpr = _dce_jaxpr(closed_jaxpr, out_unknowns, drop_outputs=True).jaxpr
842 def _dce_jaxpr(closed_jaxpr: ClosedJaxpr, outputs: Sequence[bool], drop_outputs=False) -> ClosedJax… argument
843 new_jaxpr = _dce_open_jaxpr(closed_jaxpr.jaxpr, tuple(outputs), drop_outputs)
847 def _dce_open_jaxpr(jaxpr: Jaxpr, outputs: Tuple[bool, ...], drop_outputs=False) -> Jaxpr: argument
851 if drop_outputs:
|