HomeSort by relevance Sort by last modified time
    Searched refs:RnnMode (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/
cudnn_rnn_ops.cc 109 using perftools::gputools::dnn::RnnMode;
113 Status ParseRNNMode(const string& str, RnnMode* rnn_mode) {
115 *rnn_mode = RnnMode::kRnnRelu;
118 *rnn_mode = RnnMode::kRnnTanh;
121 *rnn_mode = RnnMode::kRnnLstm;
124 *rnn_mode = RnnMode::kRnnGru;
341 RnnMode rnn_mode;
346 return rnn_mode == RnnMode::kRnnLstm;
508 RnnMode rnn_mode() const { return model_types_.rnn_mode; }
    [all...]
  /external/tensorflow/tensorflow/stream_executor/
stream_executor_pimpl.h 375 dnn::RnnMode rnn_mode, dnn::DataType data_type, float dropout,
    [all...]
dnn.h 117 enum class RnnMode {
    [all...]
stream_executor_pimpl.cc 347 dnn::RnnMode rnn_mode, dnn::DataType data_type, float dropout, uint64 seed,
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.cc     [all...]
cuda_dnn.h 53 dnn::RnnMode rnn_mode, dnn::DataType data_type, float dropout,
    [all...]

Completed in 193 milliseconds