Searched refs:params_backprop (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 800 Tensor* params_backprop, const Tensor* sequence_lengths, bool time_major, in DoBackward() argument
841 auto params_backprop_data = AsDeviceMemory<T>(params_backprop); in DoBackward()
1491 Tensor params_backprop; in MaybeAutoTune() local
1499 context->allocate_temp(data_type, params->shape(), ¶ms_backprop)); in MaybeAutoTune()
1552 ¶ms_backprop, nullptr, true, &workspace_allocator, in MaybeAutoTune()
1700 Tensor* params_backprop = nullptr; in ComputeImpl() local
1704 &input_c_backprop, ¶ms_backprop)); in ComputeImpl()
1723 input_h_backprop, input_c_backprop, params_backprop, sequence_lengths, in ComputeImpl()
1804 Tensor** input_c_backprop, Tensor** params_backprop) { in AllocateOutputs() argument
1821 context->allocate_output(3, params_shape, params_backprop)); in AllocateOutputs()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CudnnRNNBackprop.pbtxt | 42 params_backprop: The backprop to the params buffer in the forward pass. Has the
|
| D | api_def_CudnnRNNBackpropV2.pbtxt | 46 params_backprop: The backprop to the params buffer in the forward pass. Has the
|
| D | api_def_CudnnRNNBackpropV3.pbtxt | 51 params_backprop: The backprop to the params buffer in the forward pass. Has the
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 4798 …rop tf.Output, input_h_backprop tf.Output, input_c_backprop tf.Output, params_backprop tf.Output) {
4912 …rop tf.Output, input_h_backprop tf.Output, input_c_backprop tf.Output, params_backprop tf.Output) {
21878 …rop tf.Output, input_h_backprop tf.Output, input_c_backprop tf.Output, params_backprop tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7398 name: "params_backprop"
7540 name: "params_backprop"
7686 name: "params_backprop"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 16860 name: "params_backprop"
17002 name: "params_backprop"
17148 name: "params_backprop"
17294 name: "params_backprop"
|
| D | ops_history.v2.pbtxt | 16802 name: "params_backprop"
16944 name: "params_backprop"
17090 name: "params_backprop"
|