Searched refs:output_gate_scratch (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| 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 …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| 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()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 844 float* output_gate_scratch = scratch3; in LstmStepFloat() local
892 /*activation=*/kTfLiteActSigmoid, output_gate_scratch, is_input_all_zeros, in LstmStepFloat()
896 output_gate_scratch, params->activation, in LstmStepFloat()
1044 float* output_gate_scratch = scratch3; in LstmStepHybrid() local
1205 output_gate_scratch, is_input_all_zeros, is_aux_input_all_zeros, in LstmStepHybrid()
1210 n_batch, n_cell, n_output, cell_state_ptr, output_gate_scratch, in LstmStepHybrid()
1385 int16_t* output_gate_scratch = scratch3; in LstmStepInteger8x8_16() local
1460 n_output, n_cell, kTfLiteActSigmoid, output_gate_scratch, context, in LstmStepInteger8x8_16()
1465 output_gate_scratch, effective_hidden_scale_a, effective_hidden_scale_b, in LstmStepInteger8x8_16()
1630 int16_t* output_gate_scratch = scratch4; in LstmStepInteger8x8_8() local
[all …]
|