| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 581 num_proj = 4
592 num_proj=num_proj,
606 num_proj = 4
618 num_proj=num_proj,
624 num_proj=num_proj,
677 num_proj = 4
692 num_proj=num_proj,
711 num_proj = 4
724 num_proj=num_proj,
748 num_proj = 4
[all …]
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | rnn_cells.py | 185 num_proj=None, argument
231 num_proj=num_proj,
298 num_proj = self._num_units if self._num_proj is None else self._num_proj
305 m_prev = array_ops.slice(state, [0, self._num_units], [-1, num_proj])
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 195 num_proj=None, argument
263 self._num_proj = num_proj
271 self._output_size = num_proj if num_proj else num_units
415 num_proj = self._num_units if self._num_proj is None else self._num_proj
422 m_prev = array_ops.slice(state, [0, self._num_units], [-1, num_proj])
|
| D | unidirectional_sequence_lstm_test.py | 61 self.num_units, num_proj=8, forget_bias=1.0, name="rnn2"),
65 num_proj=8,
|
| D | bidirectional_sequence_lstm_test.py | 62 self.num_units, num_proj=8, forget_bias=0, name="rnn2"),
66 num_proj=8,
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | rnn_cell.py | 139 num_proj=None, argument
189 self._num_proj = num_proj
201 if num_proj:
203 rnn_cell_impl.LSTMStateTuple(num_units, num_proj)
204 if state_is_tuple else num_units + num_proj)
205 self._output_size = num_proj
246 num_proj = self._num_units if self._num_proj is None else self._num_proj
252 m_prev = array_ops.slice(state, [0, self._num_units], [-1, num_proj])
260 [input_size.value + num_proj, 3 * self._num_units],
1482 def __init__(self, num_units, num_proj=None, use_bias=False, reuse=None, argument
[all …]
|
| D | core_rnn_cell.py | 281 num_proj, argument
304 self._num_proj = num_proj
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_cell_impl.py | 833 initializer=None, num_proj=None, proj_clip=None, argument
901 self._num_proj = num_proj
912 if num_proj:
914 LSTMStateTuple(num_units, num_proj)
915 if state_is_tuple else num_units + num_proj)
916 self._output_size = num_proj
1004 num_proj = self._num_units if self._num_proj is None else self._num_proj
1011 m_prev = array_ops.slice(state, [0, self._num_units], [-1, num_proj])
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | rnn_cell_test.py | 197 num_proj = 3
204 h = array_ops.zeros([batch_size, num_proj])
208 num_proj=num_proj,
219 h.name: 0.1 * np.ones((batch_size, num_proj))
224 self.assertEqual(res[0].shape, (batch_size, num_proj))
226 self.assertEqual(res[1][1].shape, (batch_size, num_proj))
259 rnn_cell_impl.GRUCell(3), num_proj=3)
1030 num_proj = 5
1048 cell = contrib_rnn_cell.NASCell(num_units=num_units, num_proj=num_proj)
1058 0.1 * np.ones((batch_size, num_proj), dtype=np.float32),
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.rnn_cell.-l-s-t-m-cell.pbtxt | 108 …\', \'num_units\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_proj\', \'proj_clip\',…
|
| D | tensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt | 108 …\', \'num_units\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_proj\', \'proj_clip\',…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt | 108 …\', \'num_units\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_proj\', \'proj_clip\',…
|