Searched refs:restore_buffer (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/checkpoint/python/ |
| D | split_dependency.py | 49 def __init__(self, save_buffer, restore_buffer, name, dtype, num_components, argument
52 self._restore_buffer = restore_buffer
125 restore_buffer = {}
130 restore_buffer=restore_buffer,
|
| D | split_dependency_test.py | 39 def _consume_restore_buffer_fn(restore_buffer): argument
41 array_ops.concat([restore_buffer["first_half"],
42 restore_buffer["second_half"]],
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 746 def _trackable_restore(self, restore_buffer): argument
748 array_ops.identity(restore_buffer[name]) for name in self._param_names
|