| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_test.py | 85 rnn_mode, argument
479 def _CompareBiases(self, lhs, rhs, rnn_mode, num_layers, direction): argument
522 def _TestSaveRestoreVariable(self, rnn_mode, direction, dtype): argument
564 def _TestSaveRestoreTwoVariables(self, rnn_mode, direction, dtype): argument
624 def _TestSaveRestoreOutput(self, rnn_mode, direction, dtype): argument
664 def _TestSaveRestoreHelper(self, rnn_mode): argument
873 def _TestCudnnCompatibleRnnCellsHelper(self, rnn_mode): argument
911 input_size, batch_size, rnn_mode, direction): argument
1034 def _TestOpaqueParamsSize(self, rnn_mode, num_layers, num_units, input_size, argument
1203 def _TestOneSimpleTraining(self, rnn_mode, num_layers, num_units, input_size, argument
[all …]
|
| D | cudnn_rnn_ops_test.py | 1071 rnn_mode, argument
1083 def _create_saveable(self, opaque_param, rnn_mode, num_units, input_size, argument
1117 def test_save_restore_variable(self, rnn_mode, num_units, input_size, argument
1169 def test_save_restore_multi_variables(self, rnn_mode, num_units, input_size, argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops.cc | 90 string rnn_mode; in __anone1d60d010302() local
133 string rnn_mode; in __anone1d60d010402() local
179 string rnn_mode; in __anone1d60d010502() local
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 905 def _check_rnn_mode(rnn_mode): argument
934 def _get_num_params(rnn_mode, num_layers, direction): argument
957 rnn_mode, argument
1104 rnn_mode, argument
1315 def cudnn_rnn_opaque_params_to_canonical(rnn_mode, argument
1376 def cudnn_rnn_canonical_to_opaque_params(rnn_mode, argument
1436 def cudnn_rnn_opaque_params_size(rnn_mode, argument
1504 rnn_mode, argument
1566 def rnn_mode(self): member in _CudnnRNN
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/ |
| D | cudnn_rnn.py | 253 def rnn_mode(self): member in _CudnnRNN
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 152 bool has_dropout, bool is_training, RnnMode rnn_mode, in CudnnRnnParameters()
228 Status ParseRNNMode(const string& str, RnnMode* rnn_mode) { in ParseRNNMode()
480 RnnMode rnn_mode; member
924 RnnMode rnn_mode() const { return model_types_.rnn_mode; } in rnn_mode() function in tensorflow::CudnnRNNKernelCommon
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 901 cudnnRNNMode_t ToCudnnRnnMode(dnn::RnnMode rnn_mode) { in ToCudnnRnnMode()
1008 cudnnRNNMode_t rnn_mode, cudnnDataType_t data_type, in CudnnRnnDescriptor()
1035 cudnnDirectionMode_t direction_mode, cudnnRNNMode_t rnn_mode, in Create()
1114 cudnnRNNMode_t rnn_mode() const { return rnn_mode_; } in rnn_mode() function in stream_executor::gpu::CudnnRnnDescriptor
1158 cudnnRNNDescriptor_t rnn_desc, cudnnRNNMode_t rnn_mode, in Create()
1819 dnn::RnnMode rnn_mode, dnn::DataType data_type, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 384 dnn::RnnMode rnn_mode, dnn::DataType data_type, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1576 miopenRNNMode_t ToMIOpenRnnMode(dnn::RnnMode rnn_mode) { in ToMIOpenRnnMode()
1685 miopenRNNMode_t rnn_mode, miopenDataType_t data_type, in MIOpenRnnDescriptor()
1729 miopenRNNMode_t rnn_mode() const { return rnn_mode_; } in rnn_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
1766 auto rnn_mode = rnn_desc_->rnn_mode(); in GetRegionCountPerLayer() local
2278 dnn::RnnMode rnn_mode, dnn::DataType data_type, in createRnnDescriptor()
|