Searched refs:residual_fn (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | rnn_cell_wrapper_v2_test.py | 182 wrapper = wrapper_cls(cell, residual_fn=lambda i, o: i + i + o)
189 def residual_fn(inputs, outputs): function
192 wrapper = wrapper_cls(cell, residual_fn=residual_fn)
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_wrapper_impl.py | 326 def __init__(self, cell, residual_fn=None, **kwargs): argument
338 self._residual_fn = residual_fn
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 3010 wrapper = wrapper_cls(cell, residual_fn=lambda i, o: i + i + o)
3020 def residual_fn(inputs, outputs): function
3023 wrapper = wrapper_cls(cell, residual_fn=residual_fn)
|