Searched refs:layer_norm (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | rnn_cell.py | 94 normalized = layers.layer_norm(inp, reuse=True, scope=scope)
147 layer_norm=False, argument
197 self._layer_norm = layer_norm
1367 layer_norm=True, argument
1403 self._layer_norm = layer_norm
1424 normalized = layers.layer_norm(inp, reuse=True, scope=scope)
2507 layer_norm=False, argument
2552 self._layer_norm = layer_norm
2577 layer_norm=False): argument
2636 if not layer_norm:
[all …]
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | rnn_cell_test.py | 210 layer_norm=True,
1718 …single_cell = lambda: contrib_rnn_cell.LayerNormBasicLSTMCell(2, layer_norm=False) # pylint: disa…
1756 cell = contrib_rnn_cell.LayerNormBasicLSTMCell(2, layer_norm=False)
1824 2, layer_norm=True, norm_gain=1.0, norm_shift=0.0)
1879 num_units, layer_norm=False, dropout_keep_prob=keep_prob)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | rev_block_lib_test.py | 238 x = layers.layer_norm(x)
|
| D | layers_test.py | 2818 _layers.layer_norm(inputs)
2825 _layers.layer_norm(inputs)
2831 output = _layers.layer_norm(images)
2839 _layers.layer_norm(images)
2849 _layers.layer_norm(images, scope='ln')
2850 _layers.layer_norm(images, scope='ln', reuse=True)
2863 output_train = _layers.layer_norm(images, scope='LN')
2864 output_eval = _layers.layer_norm(images, scope='LN', reuse=True)
2893 output_t = _layers.layer_norm(
|
| D | layers.py | 2199 def layer_norm(inputs, function
|