Searched refs:output_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 | 66 float* forget_gate_scratch, float* cell_scratch, float* output_gate_scratch, in LstmStepWithAuxInput()
84 std::fill_n(output_gate_scratch, n_cell * n_batch, 0.0f); in LstmStepWithAuxInput()
95 output_gate_scratch); in LstmStepWithAuxInput()
113 output_gate_scratch); in LstmStepWithAuxInput()
128 output_gate_scratch, n_cell * n_batch, in LstmStepWithAuxInput()
148 n_batch, output_gate_scratch); in LstmStepWithAuxInput()
165 n_batch, output_gate_scratch); in LstmStepWithAuxInput()
280 output_gate_scratch); in LstmStepWithAuxInput()
284 output_gate_scratch, n_cell * n_batch, in LstmStepWithAuxInput()
286 tensor_utils::MeanStddevNormalization(output_gate_scratch, in LstmStepWithAuxInput()
[all …]
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 799 float* output_gate_scratch = nullptr; in LSTMStep() local
803 output_gate_scratch = cell_scratch + 2 * n_cell * n_batch; in LSTMStep()
808 output_gate_scratch = input_gate_scratch + 3 * n_cell * n_batch; in LSTMStep()
822 output_gate_scratch); in LSTMStep()
830 std::fill_n(output_gate_scratch, n_cell * n_batch, 0.0f); in LSTMStep()
846 n_batch, output_gate_scratch); in LSTMStep()
864 output_gate_scratch); in LSTMStep()
881 output_gate_scratch); in LSTMStep()
951 n_batch, output_gate_scratch); in LSTMStep()
954 tflite::tensor_utils::MeanStddevNormalization(output_gate_scratch, output_gate_scratch, in LSTMStep()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 888 float* output_gate_scratch = scratch3; in LstmStepFloat() local
947 /*activation=*/kTfLiteActSigmoid, output_gate_scratch, is_input_all_zeros, in LstmStepFloat()
951 output_gate_scratch, params->activation, in LstmStepFloat()
1100 float* output_gate_scratch = scratch3; in LstmStepHybrid() local
1261 output_gate_scratch, is_input_all_zeros, is_aux_input_all_zeros, in LstmStepHybrid()
1266 n_batch, n_cell, n_output, cell_state_ptr, output_gate_scratch, in LstmStepHybrid()
1441 int16_t* output_gate_scratch = scratch3; in LstmStepInteger8x8_16() local
1516 n_output, n_cell, kTfLiteActSigmoid, output_gate_scratch, context, in LstmStepInteger8x8_16()
1521 output_gate_scratch, effective_hidden_scale_a, effective_hidden_scale_b, in LstmStepInteger8x8_16()
1688 int16_t* output_gate_scratch = scratch4; in LstmStepInteger8x8_8() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 196 float* output_gate_scratch = scratch3; in LstmStepCalibration() local
247 /*activation=*/kTfLiteActSigmoid, output_gate_scratch, is_input_all_zeros, in LstmStepCalibration()
252 n_batch, n_cell, n_output, cell_state_ptr, output_gate_scratch, in LstmStepCalibration()
319 float* output_gate_scratch = nullptr; in EvalCalibration() local
323 output_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalCalibration()
328 output_gate_scratch = scratch_buffer_ptr + 3 * n_cell * n_batch; in EvalCalibration()
378 output_gate_scratch, output_ptr_time, logger, in EvalCalibration()
409 float* output_gate_scratch_ptr = output_gate_scratch + b * n_cell; in EvalCalibration()
|