| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_ops_test.py | 66 num_units, argument
345 num_units, argument
387 def test_training(self, num_units, input_size, batch_size, time, num_layers, argument
411 def test_training_fp16(self, num_units, input_size, batch_size, time, argument
438 def test_inference(self, num_units, input_size, batch_size, time, num_layers, argument
470 def test_inference_fp16(self, num_units, input_size, batch_size, time, argument
507 def test_inference_with_dropout(self, num_units, input_size, batch_size, time, argument
553 num_units, argument
727 num_units, argument
769 def test_training(self, num_units, input_size, batch_size, time, num_layers, argument
[all …]
|
| D | cudnn_rnn_test.py | 87 num_units, argument
910 def _TestCudnnCompatibleRnnCells(self, num_layers, seq_length, num_units, 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
|
| /external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/ |
| D | grid_rnn_cell.py | 51 num_units, argument
330 def __init__(self, num_units, state_is_tuple=True, output_is_tuple=True): argument
354 num_units, argument
377 num_units, argument
406 num_units, argument
436 num_units, argument
467 num_units, argument
504 num_units, argument
540 num_units, argument
572 ls_non_recurrent_dims, non_recurrent_fn, tied, num_units): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops_test.cc | 44 int num_units = 4; in TEST() local
76 int num_units = 4; in TEST() local
108 int num_units = 4; in TEST() local
|
| D | cudnn_rnn_ops.cc | 87 auto num_units = c->Dim(input_h_shape, 2); in __anone1d60d010302() local
130 auto num_units = c->Dim(input_h_shape, 2); in __anone1d60d010402() local
176 auto num_units = c->Dim(input_h_shape, 2); in __anone1d60d010502() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 24 int input_size, int num_units, int batch_size, in RnnBatchStep()
40 int input_size, int aux_input_size, int num_units, in RnnBatchStep()
124 int num_units, int batch_size, int output_batch_leading_dim, in RnnBatchStep()
145 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep()
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | lstm_ops.py | 349 num_units, argument
455 def num_units(self): member in LSTMBlockWrapper
601 num_units, argument
635 def num_units(self): member in LSTMBlockFusedCell
|
| D | rnn_cell.py | 136 num_units, argument
338 num_units, argument
516 num_units, argument
976 num_units, argument
1363 num_units, argument
1482 def __init__(self, num_units, num_proj=None, use_bias=False, reuse=None, argument
1639 num_units, argument
1740 num_units, argument
1922 num_units, argument
2265 num_units, argument
[all …]
|
| D | gru_ops.py | 139 num_units=None, argument
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | rnn_common.py | 61 def _get_single_cell(cell_type, num_units): argument
81 def construct_rnn_cell(num_units, cell_type='basic_rnn', argument
|
| D | dynamic_rnn_estimator.py | 386 num_units, argument
561 num_units=None, argument
|
| D | state_saving_rnn_estimator.py | 391 num_units, argument
550 num_units=None, argument
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | rnn_cells.py | 56 num_units, argument
181 num_units, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_test.py | 811 def static_vs_dynamic_rnn_benchmark(batch_size, max_time, num_units, use_gpu): argument
873 def half_seq_len_vs_unroll_half_rnn_benchmark(batch_size, max_time, num_units, argument
941 def concat_state_vs_tuple_state_rnn_benchmark(batch_size, max_time, num_units, argument
1007 def dynamic_rnn_swap_memory_benchmark(batch_size, max_time, num_units): argument
1041 def rnn_long_sequence_benchmark(batch_size, seqlen, num_units, dynamic, argument
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 69 def __init__(self, num_units, reuse=None): argument
111 def __init__(self, num_units, reuse=None, kernel_initializer=None): argument
185 num_units, argument
635 num_units, argument
1317 num_units, argument
1378 num_units, argument
1438 num_units, argument
1506 num_units, argument
1558 def num_units(self): member in _CudnnRNN
1693 num_units, argument
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 75 const int num_units = input_weights->dims->data[0]; in Prepare() local
152 const int num_units = input_weights->dims->data[0]; in EvalFloat() local
184 const int num_units = input_weights->dims->data[0]; in EvalHybrid() local
|
| D | svdf.cc | 47 int batch_size, int memory_size, int num_filters, int num_units, int rank, in ApplyTimeWeightsBiasAndActivation()
150 const int num_units = num_filters / rank; in Prepare() local
257 const int num_units = num_filters / rank; in EvalFloat() local
295 const int num_units = num_filters / rank; in EvalHybrid() local
|
| D | unidirectional_sequence_rnn.cc | 75 const int num_units = input_weights->dims->data[0]; in Prepare() local
160 const int num_units = input_weights->dims->data[0]; in EvalFloat() local
215 const int num_units = input_weights->dims->data[0]; in EvalHybrid() local
|
| D | fully_connected.cc | 110 const int num_units = filter->dims->data[0]; in Prepare() local
184 const int num_units = filter->dims->data[0]; in EvalPie() local
227 const int num_units = filter->dims->data[0]; in EvalHybrid() local
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 48 num_units, argument
191 num_units, argument
|
| /external/tensorflow/tensorflow/contrib/timeseries/examples/ |
| D | lstm.py | 51 def __init__(self, num_units, num_features, exogenous_feature_columns=None, argument
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/ |
| D | cudnn_rnn.py | 159 num_units, argument
225 def num_units(self): member in _CudnnRNN
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/nets/ |
| D | resnet_v2.py | 240 def resnet_v2_block(scope, base_depth, num_units, stride): argument
|
| D | resnet_v1.py | 227 def resnet_v1_block(scope, base_depth, num_units, stride): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_cell_impl.py | 412 num_units, argument
509 num_units, argument
658 num_units, argument
831 def __init__(self, num_units, argument
|