| /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 | 88 auto num_units = c->Dim(input_h_shape, 2); in __anon4fa58c430302() local
131 auto num_units = c->Dim(input_h_shape, 2); in __anon4fa58c430402() local
179 auto num_units = c->Dim(input_h_shape, 2); in __anon4fa58c430502() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 26 int input_size, int num_units, int batch_size, in RnnBatchStep()
42 int input_size, int aux_input_size, int num_units, in RnnBatchStep()
125 int num_units, int batch_size, int output_batch_leading_dim, in RnnBatchStep()
146 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | svdf_test.cc | 124 void ValidateSVDFGoldens(const int batch_size, const int num_units, in ValidateSVDFGoldens()
205 void ValidateIntegerSVDFGoldens(const int batch_size, const int num_units, in ValidateIntegerSVDFGoldens()
273 void TestSVDF(const int batch_size, const int num_units, const int input_size, in TestSVDF()
323 const int batch_size, const int num_units, const int input_size, in TestIntegerSVDF()
424 constexpr int num_units = 4; in TF_LITE_MICRO_TEST() local
471 constexpr int num_units = 4; in TF_LITE_MICRO_TEST() local
531 constexpr int num_units = 4; in TF_LITE_MICRO_TEST() local
|
| D | svdf.cc | 59 int batch_size, int memory_size, int num_filters, int num_units, int rank, in ApplyTimeWeightsBiasAndActivation()
155 const int num_units = num_filters / rank; in EvalFloatSVDF() local
389 const int num_units = num_filters / rank; in Prepare() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_test.py | 812 def static_vs_dynamic_rnn_benchmark(batch_size, max_time, num_units, use_gpu): argument
874 def half_seq_len_vs_unroll_half_rnn_benchmark(batch_size, max_time, num_units, argument
942 def concat_state_vs_tuple_state_rnn_benchmark(batch_size, max_time, num_units, argument
1008 def dynamic_rnn_swap_memory_benchmark(batch_size, max_time, num_units): argument
1042 def rnn_long_sequence_benchmark(batch_size, seqlen, num_units, dynamic, argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 64 const int num_units = input_weights->dims->data[0]; in Prepare() local
139 const int num_units = input_weights->dims->data[0]; in EvalFloat() local
170 const int num_units = input_weights->dims->data[0]; in EvalHybrid() local
|
| D | unidirectional_sequence_rnn.cc | 70 const int num_units = input_weights->dims->data[0]; in Prepare() local
153 const int num_units = input_weights->dims->data[0]; in EvalFloat() local
211 const int num_units = input_weights->dims->data[0]; in EvalHybrid() local
|
| D | fully_connected.cc | 162 const int num_units = filter->dims->data[0]; in PrepareImpl() local
287 const int num_units = filter->dims->data[0]; in EvalPie() local
324 const int num_units = filter->dims->data[0]; in EvalHybrid() local
|
| D | basic_rnn_test.cc | 217 int num_units() { return units_; } in num_units() function in tflite::__anon76108c220111::RNNOpModel
|
| D | fully_connected_test.cc | 190 int num_units() { return units_; } in num_units() function in tflite::__anon66fc9d710111::BaseFullyConnectedOpModel
327 int num_units() { return units_; } in num_units() function in tflite::__anon66fc9d710111::HybridFullyConnectedOpModel
|
| D | svdf.cc | 101 const int num_units = num_filters / rank; in Prepare() local
|
| D | unidirectional_sequence_rnn_test.cc | 229 int num_units() { return units_; } in num_units() function in tflite::__anon46a7a5720111::UnidirectionalRNNOpModel
|
| D | svdf_test.cc | 180 int num_units() { return units_; } in num_units() function in tflite::__anon3f7029260111::BaseSVDFOpModel
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 49 num_units, argument
193 num_units, argument
|
| /external/libgav1/libgav1/src/ |
| D | loop_restoration_info.h | 85 int num_units(Plane plane) const { return num_units_[plane]; } in num_units() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | svdf.h | 35 int batch_size, int memory_size, int num_filters, int num_units, int rank, in ApplyTimeWeightsBiasAndActivation()
212 const int num_units = num_filters / rank; in EvalFloatSVDF() local
251 const int num_units = num_filters / rank; in EvalHybridSVDF() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_cell_impl.py | 414 num_units, argument
518 num_units, argument
666 num_units, argument
842 num_units, argument
|
| /external/mesa3d/src/gallium/drivers/lima/ir/gp/ |
| D | disasm.c | 37 num_units enumerator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 150 CudnnRnnParameters(int num_layers, int input_size, int num_units, in CudnnRnnParameters()
271 Status ToRNNInputMode(TFRNNInputMode tf_input_mode, int num_units, in ToRNNInputMode()
498 int num_units; member
1018 int num_units = num_units_t->scalar<int>()(); in ExtractCudnnRNNParamsInfo() local
1207 int num_units = num_units_t->scalar<int>()(); in Compute() local
|
| D | clustering_ops.cc | 389 const int64 num_units = in Compute() local
|
| /external/libhevc/encoder/ |
| D | ihevce_dep_mngr.c | 208 iv_mem_rec_t *ps_mem_tab, WORD32 num_units, WORD32 num_threads, WORD32 i4_mem_space) in ihevce_dmgr_map_get_mem_recs()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa-hifimini/ |
| D | svdf.cc | 308 const int num_units = num_filters / rank; in Prepare() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 502 const int num_units = item->units.size(); in StartParallelExecutors() local
|
| /external/mesa3d/src/intel/compiler/ |
| D | brw_ir_performance.cpp | 62 num_units, enumerator
|