Searched refs:input_gate_scratch (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 65 float* output_state_ptr, float* cell_state_ptr, float* input_gate_scratch, in LstmStepWithAuxInput()
80 std::fill_n(input_gate_scratch, n_cell * n_batch, 0.0f); in LstmStepWithAuxInput()
88 n_batch, input_gate_scratch); in LstmStepWithAuxInput()
102 input_gate_scratch); in LstmStepWithAuxInput()
119 input_gate_scratch, n_cell * n_batch, in LstmStepWithAuxInput()
137 n_batch, input_gate_scratch); in LstmStepWithAuxInput()
155 n_batch, input_gate_scratch); in LstmStepWithAuxInput()
207 input_gate_scratch); in LstmStepWithAuxInput()
211 input_gate_scratch, n_cell * n_batch, in LstmStepWithAuxInput()
214 input_gate_scratch, input_gate_scratch, n_cell, n_batch); in LstmStepWithAuxInput()
[all …]
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 796 float* input_gate_scratch = nullptr; in LSTMStep() local
805 input_gate_scratch = scratch_buffer_buffer; in LSTMStep()
806 cell_scratch = input_gate_scratch + n_cell * n_batch; in LSTMStep()
807 forget_gate_scratch = input_gate_scratch + 2 * n_cell * n_batch; in LSTMStep()
808 output_gate_scratch = input_gate_scratch + 3 * n_cell * n_batch; in LSTMStep()
815 input_gate_scratch); in LSTMStep()
826 std::fill_n(input_gate_scratch, n_cell * n_batch, 0.0f); in LSTMStep()
837 n_batch, input_gate_scratch); in LSTMStep()
853 n_batch, input_gate_scratch); in LSTMStep()
871 n_batch, input_gate_scratch); in LSTMStep()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 193 float* input_gate_scratch = scratch0; in LstmStepCalibration() local
212 /*activation=*/kTfLiteActSigmoid, input_gate_scratch, in LstmStepCalibration()
237 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepCalibration()
316 float* input_gate_scratch = nullptr; in EvalCalibration() local
325 input_gate_scratch = scratch_buffer_ptr; in EvalCalibration()
377 input_gate_scratch, forget_gate_scratch, cell_gate_scratch, in EvalCalibration()
406 input_gate_scratch ? input_gate_scratch + b * n_cell : nullptr; in EvalCalibration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 885 float* input_gate_scratch = scratch0; in LstmStepFloat() local
908 /*activation=*/kTfLiteActSigmoid, input_gate_scratch, in LstmStepFloat()
935 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepFloat()
1097 float* input_gate_scratch = scratch0; in LstmStepHybrid() local
1207 input_gate_scratch, is_input_all_zeros, is_aux_input_all_zeros, in LstmStepHybrid()
1245 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepHybrid()
1438 int16_t* input_gate_scratch = scratch0; in LstmStepInteger8x8_16() local
1474 kTfLiteActSigmoid, input_gate_scratch, context, scratch5); in LstmStepInteger8x8_16()
1503 input_gate_scratch, forget_gate_scratch, in LstmStepInteger8x8_16()
1796 float* input_gate_scratch = nullptr; in EvalFloat() local
[all …]
|