Home
last modified time | relevance | path

Searched refs:state_saver (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/training/python/training/
Dsequence_queueing_state_saver_test.py77 state_saver = sqss.SequenceQueueingStateSaver(
87 initial_key_value_0, _ = sess.run((key, state_saver.prefetch_op))
88 initial_key_value_1, _ = sess.run((key, state_saver.prefetch_op))
94 next_batch = state_saver.next_batch
159 self.assertEqual(0, state_saver.barrier.ready_size().eval())
177 state_saver = sqss.SequenceQueueingStateSaver(
188 sess.run([state_saver.prefetch_op],
212 state_saver = sqss.SequenceQueueingStateSaver(
222 sess.run([state_saver.prefetch_op],
262 state_saver = sqss.SequenceQueueingStateSaver(
[all …]
Dsequence_queueing_state_saver.py365 def __init__(self, state_saver): argument
366 self._state_saver = state_saver
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py445 state_saver = TestStateSaver(batch_size, 2 * num_units)
456 cell, inputs, state_saver=state_saver, state_name="save_lstm")
464 [state, state_saver.saved_state["save_lstm"]],
479 state_saver = TestStateSaver(batch_size, num_units)
490 cell, inputs, state_saver=state_saver, state_name=("c", "m"))
498 state + (state_saver.saved_state["c"], state_saver.saved_state["m"]),
514 state_saver = TestStateSaver(
548 cell, inputs, state_saver=state_saver, state_name=state_names)
562 list(state_saver.saved_state.values()),
570 zip(state_saver.saved_state.keys(), saved_states))
[all …]
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dstate_saving_rnn_estimator.py48 state_saver, argument
75 state_saver=state_saver,
498 state_saver=batch,
/external/tensorflow/tensorflow/python/ops/
Drnn.py1407 state_saver, argument
1456 flat_sequence=[state_saver.state(s) for s in state_name_flat])
1458 initial_state = state_saver.state(state_name)
1471 state_saver.save_state(name, substate)
1475 save_state = [state_saver.save_state(state_name, state)]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt305 …argspec: "args=[\'cell\', \'inputs\', \'state_saver\', \'state_name\', \'sequence_length\', \'scop…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt393 …argspec: "args=[\'cell\', \'inputs\', \'state_saver\', \'state_name\', \'sequence_length\', \'scop…