| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 244 rnn_mode=self._rnn_mode,
265 rnn_mode=self._rnn_mode,
905 def _check_rnn_mode(rnn_mode): argument
906 if rnn_mode not in (CUDNN_LSTM, CUDNN_GRU, CUDNN_RNN_TANH, CUDNN_RNN_RELU):
908 (rnn_mode, CUDNN_LSTM, CUDNN_GRU, CUDNN_RNN_TANH,
934 def _get_num_params(rnn_mode, num_layers, direction): argument
936 if rnn_mode == CUDNN_LSTM:
938 elif rnn_mode == CUDNN_GRU:
940 elif rnn_mode == CUDNN_RNN_RELU:
942 elif rnn_mode == CUDNN_RNN_TANH:
[all …]
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_test.py | 85 rnn_mode, argument
106 if rnn_mode == CUDNN_LSTM:
109 elif rnn_mode == CUDNN_GRU:
112 elif rnn_mode == CUDNN_RNN_TANH:
115 elif rnn_mode == CUDNN_RNN_RELU:
119 raise ValueError("Invalid rnn_mode: %s" % rnn_mode)
177 if self._rnn.rnn_mode == CUDNN_LSTM:
193 if self._rnn.rnn_mode == CUDNN_LSTM:
233 mode = rnn.rnn_mode
479 def _CompareBiases(self, lhs, rhs, rnn_mode, num_layers, direction): argument
[all …]
|
| D | cudnn_rnn_ops_test.py | 156 rnn_mode=cudnn_rnn_ops.CUDNN_LSTM)
654 rnn_mode=cudnn_rnn_ops.CUDNN_GRU)
1071 rnn_mode, argument
1078 rnn_mode, num_layers, num_units, input_size, direction=direction)
1083 def _create_saveable(self, opaque_param, rnn_mode, num_units, input_size, argument
1085 if rnn_mode == CUDNN_LSTM:
1087 elif rnn_mode == CUDNN_GRU:
1089 elif rnn_mode == CUDNN_RNN_TANH:
1091 elif rnn_mode == CUDNN_RNN_RELU:
1117 def test_save_restore_variable(self, rnn_mode, num_units, input_size, argument
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops.cc | 90 string rnn_mode; in __anone1d60d010302() local
91 TF_RETURN_IF_ERROR(c->GetAttr("rnn_mode", &rnn_mode)); in __anone1d60d010302()
98 (rnn_mode == "lstm") ? output_h_shape : c->MakeShape({}); in __anone1d60d010302()
133 string rnn_mode; in __anone1d60d010402() local
134 TF_RETURN_IF_ERROR(c->GetAttr("rnn_mode", &rnn_mode)); in __anone1d60d010402()
141 (rnn_mode == "lstm") ? output_h_shape : c->MakeShape({}); in __anone1d60d010402()
179 string rnn_mode; in __anone1d60d010502() local
180 TF_RETURN_IF_ERROR(c->GetAttr("rnn_mode", &rnn_mode)); in __anone1d60d010502()
188 (rnn_mode == "lstm") ? output_h_shape : c->MakeShape({}); in __anone1d60d010502()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cudnn_rnn_grad.py | 45 rnn_mode=op.get_attr("rnn_mode"),
71 rnn_mode=op.get_attr("rnn_mode"),
101 rnn_mode=op.get_attr("rnn_mode"),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 152 bool has_dropout, bool is_training, RnnMode rnn_mode, in CudnnRnnParameters() argument
162 rnn_mode_(rnn_mode), in CudnnRnnParameters()
168 static_cast<int>(is_training), static_cast<int>(rnn_mode), in CudnnRnnParameters()
228 Status ParseRNNMode(const string& str, RnnMode* rnn_mode) { in ParseRNNMode() argument
230 *rnn_mode = RnnMode::kRnnRelu; in ParseRNNMode()
233 *rnn_mode = RnnMode::kRnnTanh; in ParseRNNMode()
236 *rnn_mode = RnnMode::kRnnLstm; in ParseRNNMode()
239 *rnn_mode = RnnMode::kRnnGru; in ParseRNNMode()
480 RnnMode rnn_mode; member
486 return rnn_mode == RnnMode::kRnnLstm; in HasInputC()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 901 cudnnRNNMode_t ToCudnnRnnMode(dnn::RnnMode rnn_mode) { in ToCudnnRnnMode() argument
902 switch (rnn_mode) { in ToCudnnRnnMode()
907 return static_cast<cudnnRNNMode_t>(rnn_mode); in ToCudnnRnnMode()
909 LOG(FATAL) << "Invalid RNN Mode: " << static_cast<int>(rnn_mode); in ToCudnnRnnMode()
980 cudnnRNNDescriptor_t rnn_desc, cudnnRNNMode_t rnn_mode,
1008 cudnnRNNMode_t rnn_mode, cudnnDataType_t data_type, in CudnnRnnDescriptor() argument
1022 rnn_mode_(rnn_mode), in CudnnRnnDescriptor()
1035 cudnnDirectionMode_t direction_mode, cudnnRNNMode_t rnn_mode, in Create() argument
1051 /*mode=*/rnn_mode, /*algo=*/rnn_algo, in Create()
1081 rnn_mode, direction_mode, num_layers)); in Create()
[all …]
|
| D | cuda_dnn.h | 53 dnn::RnnMode rnn_mode, dnn::DataType data_type,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CudnnRNNParamsSize.pbtxt | 11 rnn_mode: Indicates the type of the RNN model.
|
| D | api_def_CudnnRNN.pbtxt | 8 rnn_mode: Indicates the type of the RNN model.
|
| D | api_def_CudnnRNNCanonicalToParams.pbtxt | 24 rnn_mode: Indicates the type of the RNN model.
|
| D | api_def_CudnnRNNParamsToCanonical.pbtxt | 24 rnn_mode: Indicates the type of the RNN model.
|
| D | api_def_CudnnRNNV2.pbtxt | 9 rnn_mode: Indicates the type of the RNN model.
|
| D | api_def_CudnnRNNBackprop.pbtxt | 7 rnn_mode: Indicates the type of the RNN model.
|
| D | api_def_CudnnRNNV3.pbtxt | 9 rnn_mode: Indicates the type of the RNN model.
|
| D | api_def_CudnnRNNBackpropV2.pbtxt | 10 rnn_mode: Indicates the type of the RNN model.
|
| D | api_def_CudnnRNNBackpropV3.pbtxt | 9 rnn_mode: Indicates the type of the RNN model.
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/ |
| D | cudnn_rnn.py | 253 def rnn_mode(self): member in _CudnnRNN
487 rnn_mode=self._rnn_mode,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1576 miopenRNNMode_t ToMIOpenRnnMode(dnn::RnnMode rnn_mode) { in ToMIOpenRnnMode() argument
1577 switch (rnn_mode) { in ToMIOpenRnnMode()
1587 LOG(FATAL) << "Invalid RNN Mode: " << static_cast<int>(rnn_mode); in ToMIOpenRnnMode()
1685 miopenRNNMode_t rnn_mode, miopenDataType_t data_type, in MIOpenRnnDescriptor() argument
1694 rnn_mode_(rnn_mode), in MIOpenRnnDescriptor()
1702 direction_mode /*direction*/, rnn_mode /*mode*/, 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
1767 switch (rnn_mode) { in GetRegionCountPerLayer()
1776 LOG(FATAL) << "Invalid RNN Mode: " << static_cast<int>(rnn_mode); in GetRegionCountPerLayer()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 384 dnn::RnnMode rnn_mode, dnn::DataType data_type, in createRnnDescriptor() argument
394 direction_mode, rnn_mode, data_type, algorithm_config, dropout, seed, in createRnnDescriptor()
|
| D | stream_executor_pimpl.h | 410 dnn::RnnMode rnn_mode, dnn::DataType data_type,
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | cudnn_recurrent.py | 303 rnn_mode='gru')
|
| D | recurrent_v2.py | 426 rnn_mode='gru')
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 733 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'rnn_mode\', \'input_mode\', \'d…
737 …p\', \'output_h_backprop\', \'output_c_backprop\', \'reserve_space\', \'rnn_mode\', \'input_mode\'…
741 …kprop\', \'output_c_backprop\', \'reserve_space\', \'host_reserved\', \'rnn_mode\', \'input_mode\'…
745 …kprop\', \'output_c_backprop\', \'reserve_space\', \'host_reserved\', \'rnn_mode\', \'input_mode\'…
749 …num_layers\', \'num_units\', \'input_size\', \'weights\', \'biases\', \'rnn_mode\', \'input_mode\'…
753 …argspec: "args=[\'num_layers\', \'num_units\', \'input_size\', \'T\', \'S\', \'rnn_mode\', \'input…
757 …_layers\', \'num_units\', \'input_size\', \'params\', \'num_params\', \'rnn_mode\', \'input_mode\'…
761 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'rnn_mode\', \'input_mode\', \'d…
765 …'input\', \'input_h\', \'input_c\', \'params\', \'sequence_lengths\', \'rnn_mode\', \'input_mode\'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 733 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'rnn_mode\', \'input_mode\', \'d…
737 …p\', \'output_h_backprop\', \'output_c_backprop\', \'reserve_space\', \'rnn_mode\', \'input_mode\'…
741 …kprop\', \'output_c_backprop\', \'reserve_space\', \'host_reserved\', \'rnn_mode\', \'input_mode\'…
745 …kprop\', \'output_c_backprop\', \'reserve_space\', \'host_reserved\', \'rnn_mode\', \'input_mode\'…
749 …num_layers\', \'num_units\', \'input_size\', \'weights\', \'biases\', \'rnn_mode\', \'input_mode\'…
753 …argspec: "args=[\'num_layers\', \'num_units\', \'input_size\', \'T\', \'S\', \'rnn_mode\', \'input…
757 …_layers\', \'num_units\', \'input_size\', \'params\', \'num_params\', \'rnn_mode\', \'input_mode\'…
761 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'rnn_mode\', \'input_mode\', \'d…
765 …'input\', \'input_h\', \'input_c\', \'params\', \'sequence_lengths\', \'rnn_mode\', \'input_mode\'…
|