Home
last modified time | relevance | path

Searched defs:n_cell (Results 1 – 25 of 74) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunidirectional_sequence_lstm_test.cc36 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel()
251 const int n_cell = 4; in TEST_F() local
313 const int n_cell = 4; in TEST_F() local
380 const int n_cell = 4; in TEST_P() local
443 const int n_cell = 4; in TEST_P() local
554 const int n_cell = 4; in TEST_F() local
616 const int n_cell = 4; in TEST_P() local
678 const int n_cell = 4; in TEST_P() local
1340 const int n_cell = 20; in TEST_F() local
1407 const int n_cell = 20; in TEST_P() local
[all …]
H A Dbidirectional_sequence_lstm_test.cc32 BidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in BidirectionalLSTMOpModel()
433 const int n_cell = 4; in TEST_P() local
599 const int n_cell = 4; in TEST_P() local
765 const int n_cell = 4; in TEST() local
926 const int n_cell = 4; in TEST() local
1077 const int n_cell = 4; in TEST() local
1228 const int n_cell = 20; in TEST() local
1931 const int n_cell = 20; in TEST() local
2633 const int n_cell = 4; in TEST_P() local
2807 const int n_cell = 4; in TEST_P() local
H A Dlstm_eval.cc79 int32_t* projection_weights_row_sums, int32_t* row_sums, int n_cell, in ComputeRowSums()
188 const int n_output, const int n_cell, in CalculateLstmGateFloat()
255 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat()
302 void CalculateLstmOutputFloat(int n_batch, int n_cell, int n_output, in CalculateLstmOutputFloat()
340 void CalculateLstmGateHybrid( in CalculateLstmGateHybrid()
475 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputHybrid()
548 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_16()
613 void UpdateLstmCellInteger(int n_batch, int n_cell, int16_t* cell_state, in UpdateLstmCellInteger()
659 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_16()
713 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_8()
[all …]
H A Dlstm_test.cc39 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
416 const int n_cell = 4; in TEST_P() local
493 const int n_cell = 4; in TEST_P() local
563 const int n_cell = 20; in TEST_P() local
1146 const int n_cell = 4; in TEST_P() local
1241 const int n_cell = 4; in TEST_P() local
1314 LSTMIntegerOpModel(int n_batch, int n_input, int n_cell, int n_output, in LSTMIntegerOpModel()
1600 const int n_cell = 4; in TEST() local
1762 const int n_cell = 4; in TEST() local
1935 const int n_cell = 4; in TEST() local
[all …]
H A Doptional_tensor_test.cc31 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
232 const int n_cell = 4; in TEST() local
H A Dbidirectional_sequence_lstm.cc177 int n_cell, int input_to_input_weights_tensor, in CheckLstmTensorDimensionsAndTypes()
385 int n_output, int n_cell) { in CheckInputTensorDimensions()
839 int n_cell = std::max(n_fw_cell, n_bw_cell); in Prepare() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
H A Dlstm_full_test.cc35 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
268 const int n_cell = 4; in TEST_P() local
342 const int n_cell = 4; in TEST_P() local
409 const int n_cell = 20; in TEST_P() local
972 const int n_cell = 4; in TEST_F() local
1058 const int n_cell = 4; in TEST_F() local
1122 const int n_cell = 4; in TEST_F() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTMTest.cpp78 LSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LSTMOpModel()
276 const int n_cell = 4; in TEST() local
380 const int n_cell = 4; in TEST() local
476 const int n_cell = 20; in TEST() local
DLayerNormLSTMTest.cpp87 LayerNormLSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LayerNormLSTMOpModel()
290 const int n_cell = 4; in TEST() local
DLSTM.cpp151 uint32_t n_cell, LSTMParams* params) { in CheckInputTensorDimensions()
351 const uint32_t n_cell = SizeOfDimension(input_to_output_weights_, 0); in Prepare() local
791 const uint32_t n_cell = input_to_output_weights_shape.dimensions[0]; in LSTMStep() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc47 const int n_output, const int n_cell, in CalculateLstmGateFloat()
100 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat()
124 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputCalibration()
181 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepCalibration()
307 const int n_cell = input_to_output_weights->dims->data[0]; in EvalCalibration() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
Dbidirectional_sequence_lstm.mod.py25 n_cell, argument
458 n_cell = 4 variable
506 n_cell = 4 variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dquantized_lstm.mod.py22 n_cell = 4 variable
109 n_cell = 4 variable
Dlayer_norm_lstm.mod.py24 n_cell = 4 variable
196 n_cell = 4 variable
Dlstm3_state2_float16.mod.py24 n_cell = 20 variable
Dlstm2_state_float16.mod.py24 n_cell = 4 variable
Dlstm2_float16.mod.py24 n_cell = 4 variable
Dlstm3_float16.mod.py24 n_cell = 20 variable
Dlstm3_state3_float16.mod.py24 n_cell = 20 variable
Dlstm2_state2_float16.mod.py24 n_cell = 4 variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc63 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput()
384 const int n_cell = input_to_output_weights->dims->data[0]; in EvalFloat() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
Dlstm3_state.mod.py24 n_cell = 20 variable
Dlstm2_state.mod.py24 n_cell = 4 variable
Dlstm2.mod.py24 n_cell = 4 variable
Dlstm3_state3.mod.py24 n_cell = 20 variable

123