Searched defs:output_h (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 736 Tensor* output, Tensor* output_h, Tensor* output_c, in DoForward()
794 const Tensor* output, const Tensor* output_h, const Tensor* output_c, in DoBackward()
1299 Tensor* output_h = nullptr; in ComputeAndReturnAlgorithm() local
1345 Tensor* output, Tensor* output_h, in MaybeAutoTune()
1361 Tensor** output, Tensor** output_h, in AllocateOutputs()
1450 const Tensor* params, Tensor* output, Tensor* output_h, in MaybeAutoTune()
1685 const Tensor* output_h = nullptr; in ComputeImpl() local
1745 const Tensor** output_h, const Tensor** output_c, in ExtractBackwardInputs()
|