Searched refs:rnn_direction_mode (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 482 RnnDirectionMode rnn_direction_mode; member
493 static_cast<int>(rnn_direction_mode)); in DebugString()
586 (model_types.rnn_direction_mode == RnnDirectionMode::kRnnBidirectional) in ExtractForwardInput()
916 context, ParseRNNDirectionMode(str, &model_types_.rnn_direction_mode)); in CudnnRNNKernelCommon()
926 RnnDirectionMode rnn_direction_mode() const { in rnn_direction_mode() function in tensorflow::CudnnRNNKernelCommon
927 return model_types_.rnn_direction_mode; in rnn_direction_mode()
966 rnn_direction_mode(), rnn_mode(), ToDataType<T>::value, algo_config, in ExtractCudnnRNNParamsInfo()
987 rnn_direction_mode(), rnn_mode(), data_type, algo_config, dropout(), in CreateRnnDescriptor()
1109 if (rnn_direction_mode() == RnnDirectionMode::kRnnBidirectional) { in Compute()
1134 if (rnn_direction_mode() == RnnDirectionMode::kRnnUnidirectional) { in Compute()
|