Home
last modified time | relevance | path

Searched refs:rnn_mode_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc162 rnn_mode_(rnn_mode), in CudnnRnnParameters()
191 std::to_string(static_cast<int>(rnn_mode_)), in ToString()
204 rnn_mode_, rnn_input_mode_, dtype_); in get_data_as_tuple()
215 const RnnMode rnn_mode_; member in tensorflow::__anon454a70830111::CudnnRnnParameters
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1022 rnn_mode_(rnn_mode), in CudnnRnnDescriptor()
1114 cudnnRNNMode_t rnn_mode() const { return rnn_mode_; } in rnn_mode()
1145 cudnnRNNMode_t rnn_mode_; member in stream_executor::gpu::CudnnRnnDescriptor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1694 rnn_mode_(rnn_mode), in MIOpenRnnDescriptor()
1729 miopenRNNMode_t rnn_mode() const { return rnn_mode_; } in rnn_mode()
1754 miopenRNNMode_t rnn_mode_; member in stream_executor::gpu::MIOpenRnnDescriptor