Searched refs:num_unit_shards (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | rnn_cells.py | 187 num_unit_shards=None, argument
233 num_unit_shards=num_unit_shards,
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 197 num_unit_shards=None, argument
247 if num_unit_shards is not None or num_proj_shards is not None:
265 self._num_unit_shards = num_unit_shards
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 679 num_unit_shards = 2
693 num_unit_shards=num_unit_shards,
713 num_unit_shards = 2
725 num_unit_shards=num_unit_shards,
750 num_unit_shards = 2
763 num_unit_shards=num_unit_shards,
807 num_unit_shards = 2
821 num_unit_shards=num_unit_shards,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.rnn_cell.-l-s-t-m-cell.pbtxt | 108 …es\', \'cell_clip\', \'initializer\', \'num_proj\', \'proj_clip\', \'num_unit_shards\', \'num_proj…
|
| D | tensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt | 108 …es\', \'cell_clip\', \'initializer\', \'num_proj\', \'proj_clip\', \'num_unit_shards\', \'num_proj…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt | 108 …es\', \'cell_clip\', \'initializer\', \'num_proj\', \'proj_clip\', \'num_unit_shards\', \'num_proj…
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | rnn_cell.py | 141 num_unit_shards=1, argument
191 self._num_unit_shards = num_unit_shards
342 num_unit_shards=1, argument
373 self._num_unit_shards = num_unit_shards
521 num_unit_shards=1, argument
582 self._num_unit_shards = num_unit_shards
981 num_unit_shards=1, argument
1030 initializer, num_unit_shards, forget_bias, feature_size, frequency_skip,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_cell_impl.py | 834 num_unit_shards=None, num_proj_shards=None, argument
884 if num_unit_shards is not None or num_proj_shards is not None:
903 self._num_unit_shards = num_unit_shards
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/g3doc/ |
| D | README.md | 377 * `num_unit_shards` & `num_proj_shards` in LSTMCell are not supported as
|