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 …]
|