Searched refs:input_h_backprop (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 799 Tensor* input_backprop, Tensor* input_h_backprop, Tensor* input_c_backprop, in DoBackward() argument
836 auto input_h_backprop_data = AsDeviceMemory<T>(input_h_backprop); in DoBackward()
1489 Tensor input_h_backprop; in MaybeAutoTune() local
1503 data_type, model_shapes.hidden_state_shape, &input_h_backprop)); in MaybeAutoTune()
1551 &input_backprop, &input_h_backprop, &input_c_backprop, in MaybeAutoTune()
1698 Tensor* input_h_backprop = nullptr; in ComputeImpl() local
1703 &input_backprop, &input_h_backprop, in ComputeImpl()
1723 input_h_backprop, input_c_backprop, params_backprop, sequence_lengths, in ComputeImpl()
1803 Tensor** input_backprop, Tensor** input_h_backprop, in AllocateOutputs() argument
1811 context->allocate_output(1, hidden_state_shape, input_h_backprop)); in AllocateOutputs()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CudnnRNNBackprop.pbtxt | 38 input_h_backprop: The backprop to input_h in the forward pass. Has the same
|
| D | api_def_CudnnRNNBackpropV2.pbtxt | 42 input_h_backprop: The backprop to input_h in the forward pass. Has the same
|
| D | api_def_CudnnRNNBackpropV3.pbtxt | 47 input_h_backprop: The backprop to input_h in the forward pass. Has the same
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 4798 …put, optional ...CudnnRNNBackpropV3Attr) (input_backprop tf.Output, input_h_backprop tf.Output, in…
4912 …put, optional ...CudnnRNNBackpropV2Attr) (input_backprop tf.Output, input_h_backprop tf.Output, in…
21878 …utput, optional ...CudnnRNNBackpropAttr) (input_backprop tf.Output, input_h_backprop tf.Output, in…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7390 name: "input_h_backprop"
7532 name: "input_h_backprop"
7678 name: "input_h_backprop"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 16852 name: "input_h_backprop"
16994 name: "input_h_backprop"
17140 name: "input_h_backprop"
17286 name: "input_h_backprop"
|
| D | ops_history.v2.pbtxt | 16794 name: "input_h_backprop"
16936 name: "input_h_backprop"
17082 name: "input_h_backprop"
|