Home
last modified time | relevance | path

Searched refs:final_state (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_test.py370 outputs, final_state = rnn.static_rnn(
378 gradients = gradients_impl.gradients(outputs + [final_state],
381 return control_flow_ops.group(final_state, *(gradients + outputs))
392 outputs, final_state = rnn.dynamic_rnn(
397 gradients = gradients_impl.gradients([outputs, final_state],
400 return control_flow_ops.group(final_state, outputs, *gradients)
503 outputs, final_state = rnn.static_rnn(
511 gradients = gradients_impl.gradients(outputs + [final_state],
514 return control_flow_ops.group(final_state, *(gradients + outputs))
569 outputs, final_state = rnn.static_rnn(
[all …]
Drnn_cell_test.py2093 outputs_ta, final_state, _ = rnn.raw_rnn(cell, loop_fn, scope=reuse_scope)
2106 gradients = gradients_impl.gradients([outputs, final_state],
2124 [outputs, outputs_dynamic_rnn, final_state, final_state_dynamic_rnn],
3118 outputs, final_state = rnn.dynamic_rnn(
3129 res = self.evaluate([outputs, final_state])
Dtensor_array_ops_test.py183 _, final_state = control_flow_ops.while_loop(
186 c0 = final_state.concat()
/external/tensorflow/tensorflow/python/ops/
Drnn.py284 final_state = final_output_and_state[len(flat_zero_output):]
288 for substate, flat_substate in zip(final_state, flat_state):
294 final_state = nest.pack_sequence_as(
295 structure=state, flat_sequence=final_state)
297 return final_output, final_state
684 (outputs, final_state) = _dynamic_rnn_loop(
700 return (outputs, final_state)
888 _, output_final_ta, final_state = control_flow_ops.while_loop(
914 return (final_outputs, final_state)
1233 (emit_ta, final_state, final_loop_state) = returned[-3:]
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Drng_bit_generator_expander.cc84 XlaOp final_state = in GetGeneratorComputation() local
86 Tuple(&builder, {final_state, output.value}); in GetGeneratorComputation()
/external/tensorflow/tensorflow/python/autograph/operators/
Dcontrol_flow_deprecated_py2.py720 final_vars, final_state = final_aug_vars
721 set_state(final_state)
786 final_vars, final_state = ds.reduce(aug_vars, reduce_body)
787 set_state(final_state)
920 final_state = final_aug_vars[state_slice]
921 set_state(final_state)
1080 final_vars, final_state = control_flow_ops.cond(cond, error_checking_body,
1083 set_state(final_state)
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn.py262 outputs, final_state = _dynamic_rnn_loop(
279 return outputs, final_state
/external/autotest/client/cros/networking/
Dwifi_proxy.py271 (successful, final_state, duration) = result
275 'timed out in state: %s.' % (ssid, final_state))
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dgradients_test.py120 inp, (_, final_state) = dynamic_lstm_model_fn(batch_size, state_size,
122 pfor_jacobian = gradients.batch_jacobian(final_state.c, inp, use_pfor=True)
127 gradient_ops.gradients(array_ops.gather(final_state.c, i, axis=1), inp)[0]
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py954 has_data, data, final_state = control_flow_ops.while_loop(
956 return final_state
/external/mesa3d/src/intel/vulkan/
DgenX_cmd_buffer.c593 const enum isl_aux_state final_state = in transition_depth_buffer() local
603 isl_aux_state_has_valid_primary(final_state); in transition_depth_buffer()
605 isl_aux_state_has_valid_aux(final_state); in transition_depth_buffer()
611 assert(final_state != ISL_AUX_STATE_PASS_THROUGH); in transition_depth_buffer()