Searched refs:state_tuple (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | outfeed_in_nested_computation_test.cc | 44 XlaOp state_tuple = Parameter(&cond_builder, 0, state_tuple_shape, "state"); in XLA_TEST_F() local
45 XlaOp loop_counter = GetTupleElement(state_tuple, 0); in XLA_TEST_F()
53 XlaOp state_tuple = Parameter(&body_builder, 0, state_tuple_shape, "state"); in XLA_TEST_F() local
54 XlaOp loop_counter = GetTupleElement(state_tuple, 0); in XLA_TEST_F()
55 XlaOp buffer_inside = GetTupleElement(state_tuple, 1); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_ops_test.py | 189 outputs, state_tuple, inp_grad, state_grad, wgrad, bgrad = sess.run([
203 logging.vlog(1, "state_tuple: %s" % str(state_tuple))
213 return (outputs, cu_outputs, state_tuple, cu_state_tuple, inp_grad,
217 outputs, state_tuple = sess.run([outputs_op, state_tuple_op])
225 logging.vlog(1, "state_tuple: %s" % str(state_tuple))
227 return outputs, cu_outputs, state_tuple, cu_state_tuple
357 (outputs, cu_outputs, state_tuple, cu_state_tuple, inp_grad, cu_inp_grad,
370 for s, cu_s in zip(state_tuple, cu_state_tuple):
443 (outputs, cu_outputs, state_tuple, cu_state_tuple) = RunLSTM(
457 self.assertAllClose(state_tuple.h, cu_state_tuple.h)
[all …]
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | head.py | 475 def state_to_dictionary(state_tuple): argument
478 for state_number, state_value in enumerate(nest.flatten(state_tuple)):
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 642 outputs_tuple, state_tuple = rnn.static_rnn(
650 self.assertTrue(isinstance(state_tuple, tuple))
669 state_tuple_v = sess.run(state_tuple, feed_dict={inputs[0]: input_value})
|