| /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 __anon3a5366b80302() local
131 auto num_units = c->Dim(input_h_shape, 2); in __anon3a5366b80402() local
179 auto num_units = c->Dim(input_h_shape, 2); in __anon3a5366b80502() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | svdf_test.cc | 483 void ValidateSVDFGoldens(const int batch_size, const int num_units, in ValidateSVDFGoldens()
536 void TestSVDF(const int batch_size, const int num_units, const int input_size, in TestSVDF()
591 const int batch_size, const int num_units, const int input_size, in TestIntegerSVDF()
667 constexpr int num_units = 4; in TF_LITE_MICRO_TEST() local
703 constexpr int num_units = 4; in TF_LITE_MICRO_TEST() local
761 constexpr int num_units = 64; in TF_LITE_MICRO_TEST() local
792 constexpr int num_units = 64; in TF_LITE_MICRO_TEST() local
824 constexpr int num_units = 64; in TF_LITE_MICRO_TEST() local
877 constexpr int num_units = 64; in TF_LITE_MICRO_TEST() local
|
| D | svdf.cc | 56 int batch_size, int memory_size, int num_filters, int num_units, int rank, in ApplyTimeWeightsBiasAndActivation()
127 const int num_units = num_filters / rank; in EvalFloatSVDF() local
226 const int num_units = num_filters / rank; in Prepare() 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/ |
| D | rnn_test.py | 455 def static_vs_dynamic_rnn_benchmark(batch_size, max_time, num_units, use_gpu): argument
517 def half_seq_len_vs_unroll_half_rnn_benchmark(batch_size, max_time, num_units, argument
585 def concat_state_vs_tuple_state_rnn_benchmark(batch_size, max_time, num_units, argument
651 def dynamic_rnn_swap_memory_benchmark(batch_size, max_time, num_units): argument
685 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/experimental/examples/lstm/ |
| D | rnn_cell.py | 50 num_units, argument
198 num_units, argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 81 const int num_units = input_weights->dims->data[0]; in Prepare() local
206 const int num_units = input_weights->dims->data[0]; in EvalFloat() local
239 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 | fully_connected_test.cc | 200 int num_units() { return units_; } in num_units() function in tflite::__anonc847e4c60111::BaseFullyConnectedOpModel
344 int num_units() { return units_; } in num_units() function in tflite::__anonc847e4c60111::HybridFullyConnectedOpModel
1223 int num_units() { return units_; } in num_units() function in tflite::__anonc847e4c60111::SparseFullyConnectedOpModel
|
| D | basic_rnn_test.cc | 216 int num_units() { return units_; } in num_units() function in tflite::__anonaa6237b70111::RNNOpModel
|
| D | svdf.cc | 103 const int num_units = num_filters / rank; in Prepare() local
|
| D | unidirectional_sequence_rnn_test.cc | 230 int num_units() { return units_; } in num_units() function in tflite::__anona6a394470111::UnidirectionalRNNOpModel
|
| D | fully_connected.cc | 227 const int num_units = filter->dims->data[0]; in PrepareImpl() local
410 const int num_units = filter->dims->data[0]; in EvalPie() local
|
| D | svdf_test.cc | 181 int num_units() { return units_; } in num_units() function in tflite::__anon2a1e039b0111::BaseSVDFOpModel
|
| /external/libgav1/libgav1/src/ |
| D | loop_restoration_info.h | 85 int num_units(Plane plane) const { return num_units_[plane]; } in num_units() function
|
| D | decoder_impl.cc | 1328 const int num_units = in DecodeTiles() local
1347 const int num_units = in DecodeTiles() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | svdf.cc | 72 int batch_size, int memory_size, int num_filters, int num_units, int rank, in ApplyTimeWeightsBiasAndActivation()
143 const int num_units = num_filters / rank; in EvalFloatSVDF() local
313 const int num_units = num_filters / rank; in Prepare() local
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_impl.py | 421 num_units, argument
527 num_units, argument
677 num_units, argument
855 num_units, argument
|
| /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/llvm-project/lldb/source/Plugins/SymbolFile/DWARF/ |
| D | DWARFDebugInfo.cpp | 57 const size_t num_units = GetNumUnits(); in GetCompileUnitAranges() local
|
| /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
1022 int num_units = num_units_t->scalar<int>()(); in ExtractCudnnRNNParamsInfo() local
1211 int num_units = num_units_t->scalar<int>()(); in Compute() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | svdf.cc | 357 const int num_units = num_filters / rank; in Prepare() local
|