Searched defs:output_c (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()
1485 Tensor* output_c = nullptr; in ComputeAndReturnAlgorithm() local
1531 Tensor* output_c, in MaybeAutoTune()
1547 Tensor** output_c) { in AllocateOutputs()
1637 Tensor* output_c, in MaybeAutoTune()
1885 const Tensor* output_c = nullptr; in ComputeImpl() local
1944 const Tensor** output_h, const Tensor** output_c, in ExtractBackwardInputs()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | convert.cc | 196 int output_c = p * kPiohw4ChannelsInPlane + c; in ConvertToPIOHW4() local
|