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

  /external/tensorflow/tensorflow/core/kernels/
cudnn_rnn_ops.cc 127 using se::dnn::RnnDirectionMode;
260 RnnDirectionMode* rnn_dir_mode) {
262 *rnn_dir_mode = RnnDirectionMode::kRnnUnidirectional;
265 *rnn_dir_mode = RnnDirectionMode::kRnnBidirectional;
482 RnnDirectionMode rnn_direction_mode;
586 (model_types.rnn_direction_mode == RnnDirectionMode::kRnnBidirectional)
    [all...]
  /external/tensorflow/tensorflow/stream_executor/
stream_executor_pimpl.h 409 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode,
    [all...]
dnn.h 140 enum class RnnDirectionMode {
    [all...]
stream_executor_pimpl.cc 383 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode,
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.cc     [all...]
cuda_dnn.h 52 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode,
    [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_dnn.cc     [all...]
rocm_dnn.h 50 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode,
    [all...]

Completed in 197 milliseconds