Home
last modified time | relevance | path

Searched defs:input_h (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py953 input_h, argument
1045 input_h, argument
1101 input_h, argument
1158 input_h, argument
1211 input_h, argument
1263 input_h, argument
1592 input_h, argument
1732 input_h, argument
1829 input_h, argument
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc563 const Tensor** input, const Tensor** input_h, in ExtractForwardInput()
640 const Tensor** input, const Tensor** input_h, in ExtractForwardInput()
732 const Tensor* input, const Tensor* input_h, in DoForward()
791 const Tensor* input, const Tensor* input_h, const Tensor* input_c, in DoBackward()
1278 const Tensor* input_h = nullptr; in ComputeAndReturnAlgorithm() local
1343 const Tensor* input, const Tensor* input_h, in MaybeAutoTune()
1449 const Tensor* input_h, const Tensor* input_c, in MaybeAutoTune()
1664 const Tensor* input_h = nullptr; in ComputeImpl() local
Dconv_grad_input_ops.cc692 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator ()() local
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent_v2.py785 def cudnn_lstm(inputs, input_h, input_c, kernel, recurrent_kernel, bias, argument