| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | unidirectional_sequence_lstm_test.cc | 36 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel()
252 const int n_output = 4; in TEST_F() local
314 const int n_output = 4; in TEST_F() local
381 const int n_output = 4; in TEST_P() local
444 const int n_output = 4; in TEST_P() local
555 const int n_output = 4; in TEST_F() local
617 const int n_output = 4; in TEST_P() local
679 const int n_output = 4; in TEST_P() local
1341 const int n_output = 16; in TEST_F() local
1408 const int n_output = 16; in TEST_P() local
[all …]
|
| H A D | bidirectional_sequence_lstm_test.cc | 32 BidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in BidirectionalLSTMOpModel()
434 const int n_output = 4; in TEST_P() local
600 const int n_output = 4; in TEST_P() local
766 const int n_output = 4; in TEST() local
927 const int n_output = 4; in TEST() local
1078 const int n_output = 4; in TEST() local
1229 const int n_output = 16; in TEST() local
1932 const int n_output = 16; in TEST() local
2634 const int n_output = 4; in TEST_P() local
2808 const int n_output = 4; in TEST_P() local
|
| H A D | lstm_test.cc | 39 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
417 const int n_output = 4; in TEST_P() local
494 const int n_output = 4; in TEST_P() local
564 const int n_output = 16; in TEST_P() local
1147 const int n_output = 3; in TEST_P() local
1242 const int n_output = 3; in TEST_P() local
1314 LSTMIntegerOpModel(int n_batch, int n_input, int n_cell, int n_output, in LSTMIntegerOpModel()
1601 const int n_output = 3; in TEST() local
1763 const int n_output = 3; in TEST() local
1936 const int n_output = 3; in TEST() local
[all …]
|
| H A D | lstm_eval.cc | 80 int n_input, int n_aux_input, int n_output, in ComputeRowSums()
188 const int n_output, const int n_cell, in CalculateLstmGateFloat()
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()
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()
767 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_8()
851 inline void LstmStepFloat( in LstmStepFloat()
[all …]
|
| H A D | unidirectional_sequence_gru_test.cc | 34 explicit GRUOpModel(int n_batch, int n_input, int n_output, in GRUOpModel()
103 const int n_output = 3; in TEST() local
|
| H A D | optional_tensor_test.cc | 31 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
233 const int n_output = 4; in TEST() local
|
| H A D | unidirectional_sequence_gru.cc | 42 const int n_output = output->dims->data[2]; in GruImpl() local
151 const int n_output = input_state->dims->data[1]; in Prepare() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| H A D | lstm_full_test.cc | 35 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
269 const int n_output = 4; in TEST_P() local
343 const int n_output = 4; in TEST_P() local
410 const int n_output = 16; in TEST_P() local
973 const int n_output = 3; in TEST_F() local
1059 const int n_output = 3; in TEST_F() local
1123 const int n_output = 4; in TEST_F() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTMTest.cpp | 78 LSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LSTMOpModel()
277 const int n_output = 4; in TEST() local
381 const int n_output = 4; in TEST() local
477 const int n_output = 16; in TEST() local
|
| D | LayerNormLSTMTest.cpp | 87 LayerNormLSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LayerNormLSTMOpModel()
291 const int n_output = 3; in TEST() local
|
| D | LSTM.cpp | 150 const RunTimeOperandInfo* output_layer_norm_weights, uint32_t n_input, uint32_t n_output, in CheckInputTensorDimensions()
357 const uint32_t n_output = SizeOfDimension(recurrent_to_output_weights_, 1); in Prepare() local
792 const uint32_t n_output = recurrent_to_output_weights_shape.dimensions[1]; in LSTMStep() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_lstm.mod.py | 26 n_output, argument
459 n_output = 4 variable
507 n_output = 4 variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 47 const int n_output, const int n_cell, in CalculateLstmGateFloat()
124 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputCalibration()
158 inline void LstmStepCalibration( in LstmStepCalibration()
308 const int n_output = recurrent_to_output_weights->dims->data[1]; in EvalCalibration() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | quantized_lstm.mod.py | 23 n_output = n_cell variable
110 n_output = n_cell variable
|
| D | layer_norm_lstm.mod.py | 25 n_output = 3 variable
197 n_output = 3 variable
|
| D | lstm3_state2_float16.mod.py | 25 n_output = 16 variable
|
| D | lstm2_state_float16.mod.py | 25 n_output = 4 variable
|
| D | lstm2_float16.mod.py | 25 n_output = 4 variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | neon_tensor_utils.h | 108 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
118 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
142 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
|
| H A D | sse_tensor_utils.h | 122 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
132 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
156 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
|
| H A D | neon_tensor_utils.cc | 770 int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyImpl()
879 int32_t multiplier, int32_t shift, int32_t n_batch, int32_t n_output, in NeonMatrixBatchVectorAccumulateImpl()
932 int32_t multiplier, int32_t shift, int32_t n_batch, int32_t n_output, in NeonMatrixBatchVectorAccumulateImpl()
1007 int32_t n_input, int32_t n_output, int32_t output_zp, in NeonCpuBackendGemm()
1040 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyAccumulate()
1056 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyAccumulate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | portable_tensor_utils.h | 141 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
151 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
181 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
|
| H A D | portable_tensor_utils.cc | 331 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulateImpl()
360 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulate()
370 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulate()
413 int32_t n_output, int32_t output_zp, int8_t* proj_output) { in PortableMatrixBatchVectorMultiply()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
385 const int n_output = recurrent_to_output_weights->dims->data[1]; in EvalFloat() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/ |
| D | lstm3_state.mod.py | 25 n_output = 16 variable
|