| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A 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
|
| H A D | cudnn_rnn_ops.cc | 93 auto num_units = c->Dim(input_h_shape, 2); in __anonb0821bee0302() local
142 auto num_units = c->Dim(input_h_shape, 2); in __anonb0821bee0402() local
196 auto num_units = c->Dim(input_h_shape, 2); in __anonb0821bee0502() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A 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
|
| H A 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
|
| H A 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
|
| H A D | fully_connected_test.cc | 245 int num_units() { return units_; } in num_units() function in tflite::__anonaa112c3c0111::BaseFullyConnectedOpModel
452 int num_units() { return units_; } in num_units() function in tflite::__anonaa112c3c0111::HybridFullyConnectedOpModel
1597 int num_units() { return units_; } in num_units() function in tflite::__anonaa112c3c0111::SparseFullyConnectedOpModel
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| H A D | rnn_cell_impl.py | 421 num_units, argument
528 num_units, argument
680 num_units, argument
859 num_units, argument
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | SVDF.cpp | 86 const uint32_t num_units = num_filters / rank; in Prepare() local
173 const int num_units = num_filters / rank; in EvalFloat32() local
|
| D | RNN.cpp | 62 const uint32_t num_units = SizeOfDimension(input_weights, 0); in Prepare() local
146 const uint32_t num_units = weightsShape.dimensions[0]; in RNNStep() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | cudnn_rnn_ops.cc | 153 CudnnRnnParameters(int num_layers, int input_size, int num_units, in CudnnRnnParameters()
274 Status ToRNNInputMode(TFRNNInputMode tf_input_mode, int num_units, in ToRNNInputMode()
501 int num_units; member
1064 int num_units = num_units_t->scalar<int>()(); in ExtractCudnnRNNParamsInfo() local
1253 int num_units = num_units_t->scalar<int>()(); in Compute() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A 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
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A 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
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/radeonsi/ |
| H A D | radeon_video.h | 29 unsigned num_units; member
|
| /aosp_15_r20/external/libaom/aom/ |
| H A 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
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | FullyConnected.cpp | 43 uint32_t num_units = getSizeOfDimension(weights, 0u); in validateShapes() local
|
| /aosp_15_r20/external/libgav1/src/ |
| H A D | loop_restoration_info.h | 83 int num_units(Plane plane) const { return num_units_[plane]; } in num_units() function
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | unidirectional_sequence_rnn.mod.py | 47 num_units = 16 variable
|
| /aosp_15_r20/external/mesa3d/src/freedreno/vulkan/ |
| H A D | tu_cmd_buffer.cc | 4789 unsigned num_units = const_state->push_consts.dwords; in tu6_user_consts_size() local
4810 unsigned num_units = const_state->push_consts.dwords; in tu6_emit_per_stage_push_consts() local
4928 unsigned num_units = shared_consts->dwords; in tu6_emit_shared_consts() local
5198 unsigned num_units) in tu_emit_fdm_params()
5239 unsigned num_units = fs_params_size(cmd); in tu6_emit_fs_params() local
5280 unsigned num_units = DIV_ROUND_UP(fs->const_state.fdm_ubo.size, 4); in tu7_emit_fs_params() local
|
| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A 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()
|
| H A D | graph_resnet50.cpp | 140 unsigned int base_depth, unsigned int num_units, unsigned int stride) in add_residual_block()
|
| H A D | graph_resnet_v2_50.cpp | 143 unsigned int base_depth, unsigned int num_units, unsigned int stride) in add_residual_block()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | unidirectional_sequence_rnn.mod.py | 50 num_units = 16 variable
|