| /external/tensorflow/tensorflow/python/ops/ |
| cudnn_rnn_grad.py | 36 output_h=op.outputs[1],
61 output_h=op.outputs[1],
90 output_h=op.outputs[1],
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/ |
| cudnn_rnn.py | 433 outputs, (output_h, output_c) = self._forward(
436 return outputs, (output_h, output_c)
438 return outputs, (output_h,)
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/kernels/ |
| cudnn_rnn_ops.cc | 736 Tensor* output, Tensor* output_h, Tensor* output_c,
763 auto output_h_data = AsDeviceMemory<T>(output_h);
794 const Tensor* output, const Tensor* output_h, const Tensor* output_c,
824 auto output_h_data = AsDeviceMemory<T>(output_h);
1299 Tensor* output_h = nullptr; local
1685 const Tensor* output_h = nullptr; local
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_ops_benchmark.py | 107 output, output_h, output_c = model(
114 [output, output_h, output_c],
|
| cudnn_rnn_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| cudnn_rnn_ops.py | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
