Searched defs:output_h (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 794 Tensor* output, Tensor* output_h, Tensor* output_c, in DoForward()
853 const Tensor* output, const Tensor* output_h, const Tensor* output_c, in DoBackward()
1484 Tensor* output_h = nullptr; in ComputeAndReturnAlgorithm() local
1530 Tensor* output, Tensor* output_h, in MaybeAutoTune()
1546 Tensor** output, Tensor** output_h, in AllocateOutputs()
1636 const Tensor* params, Tensor* output, Tensor* output_h, in MaybeAutoTune()
1884 const Tensor* output_h = nullptr; in ComputeImpl() local
1944 const Tensor** output_h, const Tensor** output_c, in ExtractBackwardInputs()
|
| D | softmax_op_gpu.cu.cc | 128 Eigen::half* output_h = reinterpret_cast<Eigen::half*>(&output_d); in GenerateNormalizedProb() local
|