| /external/tensorflow/tensorflow/python/ops/ |
| D | cudnn_rnn_grad.py | 37 output_c=op.outputs[2],
62 output_c=op.outputs[2],
91 output_c=op.outputs[2],
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 736 Tensor* output, Tensor* output_h, Tensor* output_c, in DoForward() argument
766 output_c_data = AsDeviceMemory<T>(output_c); in DoForward()
794 const Tensor* output, const Tensor* output_h, const Tensor* output_c, in DoBackward() argument
827 output_c_data = AsDeviceMemory<T>(output_c); in DoBackward()
1300 Tensor* output_c = nullptr; in ComputeAndReturnAlgorithm() local
1302 &output_h, &output_c)); in ComputeAndReturnAlgorithm()
1319 output_c, output_algo_config)); in ComputeAndReturnAlgorithm()
1332 input_c, params, is_training_, output, output_h, output_c, in ComputeAndReturnAlgorithm()
1346 Tensor* output_c, in MaybeAutoTune() argument
1362 Tensor** output_c) { in AllocateOutputs() argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/ |
| D | cudnn_rnn.py | 433 outputs, (output_h, output_c) = self._forward(
436 return outputs, (output_h, output_c)
500 output, output_h, output_c = cudnn_rnn_ops._cudnn_rnn( # pylint:disable=protected-access
513 return output, (output_h, output_c)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CudnnRNNBackprop.pbtxt | 29 output_c: The same shape as input_c for LSTM. An empty tensor for other models.
33 output_c_backprop: A 3-D tensor with the same shape as output_c in the forward
|
| D | api_def_CudnnRNNBackpropV2.pbtxt | 32 output_c: The same shape as input_c for LSTM. An empty tensor for other models.
36 output_c_backprop: A 3-D tensor with the same shape as output_c in the forward
|
| D | api_def_CudnnRNNBackpropV3.pbtxt | 36 output_c: The same shape as input_c for LSTM. An empty tensor for other models.
40 output_c_backprop: A 3-D tensor with the same shape as output_c in the forward
|
| D | api_def_CudnnRNN.pbtxt | 30 output_c: The same shape as input_c for LSTM. An empty tensor for other models.
|
| D | api_def_CudnnRNNV2.pbtxt | 31 output_c: The same shape as input_c for LSTM. An empty tensor for other models.
|
| D | api_def_CudnnRNNV3.pbtxt | 36 output_c: The same shape as input_c for LSTM. An empty tensor for other models.
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_ops_benchmark.py | 107 output, output_h, output_c = model(
114 [output, output_h, output_c],
|
| D | cudnn_rnn_test.py | 986 output_c = array_ops.stack([s.c for s in states])
997 output_h, output_c = [], []
1000 output_c.append(array_ops.stack([s_fw.c, s_bw.c]))
1002 output_c = array_ops.concat(output_c, axis=0)
1016 [outputs, output_h, output_c],
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 1029 outputs, output_h, output_c, _, _ = gen_cudnn_rnn_ops.cudnn_rnnv3(**args)
1036 outputs, output_h, output_c, _, _ = gen_cudnn_rnn_ops.cudnn_rnnv3(**args)
1038 outputs, output_h, output_c, _ = gen_cudnn_rnn_ops.cudnn_rnn(**args)
1040 outputs, output_h, output_c, _, _ = gen_cudnn_rnn_ops.cudnn_rnnv2(**args)
1041 return (outputs, output_h, output_c)
1766 output, output_h, output_c = super(CudnnLSTM, self).__call__(
1774 return (output, output_h, output_c)
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | keras_test.py | 92 output_c = keras.layers.Dense(3, activation='softmax', name='dense_2')(merged)
95 inputs=[input_a, input_b], outputs=[output_c, output_d])
111 output_c = keras.layers.Dense(3, activation='softmax', name='dense_2')(merged)
114 inputs=[input_a, input_b, input_m], outputs=[output_c, output_d])
|
| D | keras_backward_compat_test.py | 77 output_c = keras.layers.Dense(3, activation='softmax', name='dense_2')(merged)
80 inputs=[input_a, input_b, input_m], outputs=[output_c, output_d])
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 4798 …tput, sequence_lengths tf.Output, output tf.Output, output_h tf.Output, output_c tf.Output, output…
4809 …input, input_h, input_c, params, sequence_lengths, output, output_h, output_c, output_backprop, ou…
4912 …ut_c tf.Output, params tf.Output, output tf.Output, output_h tf.Output, output_c tf.Output, output…
4923 …input, input_h, input_c, params, output, output_h, output_c, output_backprop, output_h_backprop, o…
5025 …put, optional ...CudnnRNNV2Attr) (output tf.Output, output_h tf.Output, output_c tf.Output, reserv…
17225 …put, optional ...CudnnRNNV3Attr) (output tf.Output, output_h tf.Output, output_c tf.Output, reserv…
21878 …ut_c tf.Output, params tf.Output, output tf.Output, output_h tf.Output, output_c tf.Output, output…
21889 …input, input_h, input_c, params, output, output_h, output_c, output_backprop, output_h_backprop, o…
25116 …utput, optional ...CudnnRNNAttr) (output tf.Output, output_h tf.Output, output_c tf.Output, reserv…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 737 …ut\', \'input_h\', \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backp…
741 …ut\', \'input_h\', \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backp…
745 …nput_c\', \'params\', \'sequence_lengths\', \'output\', \'output_h\', \'output_c\', \'output_backp…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 737 …ut\', \'input_h\', \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backp…
741 …ut\', \'input_h\', \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backp…
745 …nput_c\', \'params\', \'sequence_lengths\', \'output\', \'output_h\', \'output_c\', \'output_backp…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7249 name: "output_c"
7366 name: "output_c"
7504 name: "output_c"
7650 name: "output_c"
8120 name: "output_c"
8245 name: "output_c"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 16711 name: "output_c"
16828 name: "output_c"
16966 name: "output_c"
17112 name: "output_c"
17258 name: "output_c"
17728 name: "output_c"
17853 name: "output_c"
17978 name: "output_c"
|
| D | ops_history.v2.pbtxt | 16653 name: "output_c"
16770 name: "output_c"
16908 name: "output_c"
17054 name: "output_c"
17517 name: "output_c"
17642 name: "output_c"
|