Searched defs:output_c (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 796 Tensor* output, Tensor* output_h, Tensor* output_c, in DoForward()
875 const Tensor* output, const Tensor* output_h, const Tensor* output_c, in DoBackward()
1526 Tensor* output_c = nullptr; in ComputeAndReturnAlgorithm() local
1572 Tensor* output_c, in MaybeAutotune()
1588 Tensor** output_c) { in AllocateOutputs()
1678 Tensor* output_c, in MaybeAutotune()
1927 const Tensor* output_c = nullptr; in ComputeImpl() local
1986 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
|