Searched refs:rnn_desc_ptr (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 1325 RnnDescriptor* rnn_desc_ptr = nullptr; in ComputeAndReturnAlgorithm() local
1329 &rnn_state_cache_, &rnn_desc_ptr)); in ComputeAndReturnAlgorithm()
1331 context, *rnn_desc_ptr, model_types(), model_shapes, input, input_h, in ComputeAndReturnAlgorithm()
1714 RnnDescriptor* rnn_desc_ptr = nullptr; in ComputeImpl() local
1718 &rnn_desc_ptr)); in ComputeImpl()
1720 context, *rnn_desc_ptr, model_types(), model_shapes, input, input_h, in ComputeImpl()
|