| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops_test.cc | 44 int num_units = 4; in TEST() local
81 int num_units = 4; in TEST() local
118 int num_units = 4; in TEST() local
160 int num_units = 4; in TEST() local
|
| D | cudnn_rnn_ops.cc | 93 auto num_units = c->Dim(input_h_shape, 2); in __anon8b6eb66e0302() local
142 auto num_units = c->Dim(input_h_shape, 2); in __anon8b6eb66e0402() local
196 auto num_units = c->Dim(input_h_shape, 2); in __anon8b6eb66e0502() 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()
123 int num_units, int batch_size, int output_batch_leading_dim, in RnnBatchStep()
148 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep()
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | rnn_test.py | 450 def static_vs_dynamic_rnn_benchmark(batch_size, max_time, num_units, use_gpu): argument
512 def half_seq_len_vs_unroll_half_rnn_benchmark(batch_size, max_time, num_units, argument
580 def concat_state_vs_tuple_state_rnn_benchmark(batch_size, max_time, num_units, argument
646 def dynamic_rnn_swap_memory_benchmark(batch_size, max_time, num_units): argument
680 def rnn_long_sequence_benchmark(batch_size, seqlen, num_units, dynamic, argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | svdf.h | 38 int batch_size, int memory_size, int num_filters, int num_units, int rank, in ApplyTimeWeightsBiasAndActivation()
163 const int num_units = num_filters / rank; in EvalFloatSVDF() local
204 const int num_units = num_filters / rank; in EvalHybridSVDF() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 81 const int num_units = input_weights->dims->data[0]; in Prepare() local
207 const int num_units = input_weights->dims->data[0]; in EvalFloat() local
240 const int num_units = input_weights->dims->data[0]; in EvalHybrid() local
|
| D | unidirectional_sequence_rnn.cc | 87 const int num_units = input_weights->dims->data[0]; in Prepare() local
221 const int num_units = input_weights->dims->data[0]; in EvalFloat() local
281 const int num_units = input_weights->dims->data[0]; in EvalHybrid() local
|
| D | basic_rnn_test.cc | 216 int num_units() { return units_; } in num_units() function in tflite::__anond471c92d0111::RNNOpModel
|
| D | fully_connected.cc | 255 const int num_units = filter->dims->data[0]; in PrepareImpl() local
487 const int num_units = filter->dims->data[0]; in EvalPie() local
525 const int num_units = filter->dims->data[0]; in EvalHybridDense() local
|
| D | svdf.cc | 104 const int num_units = num_filters / rank; in Prepare() local
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| D | radeon_video.h | 29 unsigned num_units; member
|
| /external/libgav1/src/ |
| D | loop_restoration_info.h | 83 int num_units(Plane plane) const { return num_units_[plane]; } in num_units() function
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_impl.py | 421 num_units, argument
528 num_units, argument
680 num_units, argument
859 num_units, argument
|
| /external/ComputeLibrary/examples/ |
| D | graph_resnext50.cpp | 125 …unsigned int base_depth, unsigned int stage, unsigned int num_units, unsigned int stride_conv_unit… in add_residual_block()
|
| D | graph_resnet50.cpp | 140 unsigned int base_depth, unsigned int num_units, unsigned int stride) in add_residual_block()
|
| D | graph_resnet_v2_50.cpp | 143 unsigned int base_depth, unsigned int num_units, unsigned int stride) in add_residual_block()
|
| /external/marisa-trie/lib/marisa/grimoire/vector/ |
| D | flat-vector.h | 105 std::size_t num_units = values.empty() ? 0 : (64 / MARISA_WORD_SIZE); in build_() local
|
| /external/mesa3d/src/gallium/drivers/lima/ir/gp/ |
| D | disasm.c | 37 num_units enumerator
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonLstmFloatWorkload.cpp | 144 … const unsigned int num_units = armnn::numeric_cast<unsigned int>(inputTensorInfo.GetShape()[1]); in NeonLstmFloatWorkload() local
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClLstmFloatWorkload.cpp | 147 … const unsigned int num_units = armnn::numeric_cast<unsigned int>(inputTensorInfo.GetShape()[1]); in ClLstmFloatWorkload() local
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEQLSTMLayer.cpp | 272 const int num_units = input_to_output_weights->info()->dimension(1); in configure() local
692 const unsigned int num_units = input_to_output_weights->dimension(1); in validate() local
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLQLSTMLayer.cpp | 201 const int num_units = input_to_output_weights->info()->dimension(1); in configure() local
620 const unsigned int num_units = input_to_output_weights->dimension(1); in validate() local
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | FusedAdagradKernel.cpp | 160 size_t num_units = divup(param.numel(), cache_line_aligned_task_unit); in adagrad_fused_step_impl() local
|
| D | FusedSGDKernel.cpp | 205 size_t num_units = divup(param.numel(), cache_line_aligned_task_unit); in sgd_fused_step_impl() local
|
| /external/libaom/aom/ |
| D | aom_external_partition.h | 186 int num_units; ///< The number of units inside the current superblock member
201 int num_units; ///< The number of units inside the current superblock member
|