Searched defs:output_c (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 797 Tensor* output, Tensor* output_h, Tensor* output_c, in DoForward()
876 const Tensor* output, const Tensor* output_h, const Tensor* output_c, in DoBackward()
1527 Tensor* output_c = nullptr; in ComputeAndReturnAlgorithm() local
1574 Tensor* output_c, in MaybeAutotune()
1590 Tensor** output_c) { in AllocateOutputs()
1680 Tensor* output_c, in MaybeAutotune()
1929 const Tensor* output_c = nullptr; in ComputeImpl() local
1988 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
|