Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dcudnn_rnn_grad.py39 output_h_backprop=grads[1],
64 output_h_backprop=grad[1],
93 output_h_backprop=grads[1],
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc796 const Tensor* output_backprop, const Tensor* output_h_backprop, in DoBackward() argument
830 auto output_h_backprop_data = AsDeviceMemory<T>(output_h_backprop); in DoBackward()
1486 Tensor output_h_backprop; in MaybeAutoTune() local
1496 data_type, model_shapes.hidden_state_shape, &output_h_backprop)); in MaybeAutoTune()
1550 &output_h_backprop, &output_c_backprop, &reserve_space, in MaybeAutoTune()
1688 const Tensor* output_h_backprop = nullptr; in ComputeImpl() local
1694 &output_backprop, &output_h_backprop, in ComputeImpl()
1722 output_h_backprop, output_c_backprop, reserve_space, input_backprop, in ComputeImpl()
1746 const Tensor** output_backprop, const Tensor** output_h_backprop, in ExtractBackwardInputs() argument
1751 TF_RETURN_IF_ERROR(context->input("output_h_backprop", output_h_backprop)); in ExtractBackwardInputs()
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_CudnnRNNBackprop.pbtxt31 output_h_backprop: A 3-D tensor with the same shape as output_h in the forward
Dapi_def_CudnnRNNBackpropV2.pbtxt34 output_h_backprop: A 3-D tensor with the same shape as output_h in the forward
Dapi_def_CudnnRNNBackpropV3.pbtxt38 output_h_backprop: A 3-D tensor with the same shape as output_h in the forward
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt737 …s\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h_backprop\', \'output_…
741 …s\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h_backprop\', \'output_…
745 …s\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h_backprop\', \'output_…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt737 …s\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h_backprop\', \'output_…
741 …s\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h_backprop\', \'output_…
745 …s\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h_backprop\', \'output_…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go4798 … output_h tf.Output, output_c tf.Output, output_backprop tf.Output, output_h_backprop tf.Output, o…
4809 …ams, sequence_lengths, output, output_h, output_c, output_backprop, output_h_backprop, output_c_ba…
4912 … output_h tf.Output, output_c tf.Output, output_backprop tf.Output, output_h_backprop tf.Output, o…
4923 …input, input_h, input_c, params, output, output_h, output_c, output_backprop, output_h_backprop, o…
21878 … output_h tf.Output, output_c tf.Output, output_backprop tf.Output, output_h_backprop tf.Output, o…
21889 …input, input_h, input_c, params, output, output_h, output_c, output_backprop, output_h_backprop, o…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt7374 name: "output_h_backprop"
7512 name: "output_h_backprop"
7658 name: "output_h_backprop"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt16836 name: "output_h_backprop"
16974 name: "output_h_backprop"
17120 name: "output_h_backprop"
17266 name: "output_h_backprop"
Dops_history.v2.pbtxt16778 name: "output_h_backprop"
16916 name: "output_h_backprop"
17062 name: "output_h_backprop"