Searched defs:input_h (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.h | 120 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator() local
|
| D | cudnn_rnn_ops.cc | 567 const Tensor** input, const Tensor** input_h, in ExtractForwardInput()
681 const Tensor** input, const Tensor** input_h, in ExtractForwardInput()
790 const Tensor* input, const Tensor* input_h, in DoForward()
850 const Tensor* input, const Tensor* input_h, const Tensor* input_c, in DoBackward()
1459 const Tensor* input_h = nullptr; in ComputeAndReturnAlgorithm() local
1528 const Tensor* input, const Tensor* input_h, in MaybeAutoTune()
1635 const Tensor* input_h, const Tensor* input_c, in MaybeAutoTune()
1859 const Tensor* input_h = nullptr; in ComputeImpl() local
|
| D | conv_grad_input_ops.cc | 78 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator ()() local
|