Searched refs:input_gate_scratch (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| 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()
118 logger->LogTensorValue(intemediate_tensor_indexes[1], input_gate_scratch, in LstmStepWithAuxInput()
134 n_batch, input_gate_scratch); in LstmStepWithAuxInput()
152 n_batch, input_gate_scratch); in LstmStepWithAuxInput()
201 input_gate_scratch); in LstmStepWithAuxInput()
204 logger->LogTensorValue(intemediate_tensor_indexes[0], input_gate_scratch, in LstmStepWithAuxInput()
207 input_gate_scratch, input_gate_scratch, n_cell, n_batch); in LstmStepWithAuxInput()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 192 float* input_gate_scratch = scratch0; in LstmStepCalibration() local
211 /*activation=*/kTfLiteActSigmoid, input_gate_scratch, in LstmStepCalibration()
236 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepCalibration()
315 float* input_gate_scratch = nullptr; in EvalCalibration() local
324 input_gate_scratch = scratch_buffer_ptr; in EvalCalibration()
376 input_gate_scratch, forget_gate_scratch, cell_gate_scratch, in EvalCalibration()
405 input_gate_scratch ? input_gate_scratch + b * n_cell : nullptr; in EvalCalibration()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 841 float* input_gate_scratch = scratch0; in LstmStepFloat() local
860 /*activation=*/kTfLiteActSigmoid, input_gate_scratch, in LstmStepFloat()
882 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepFloat()
1041 float* input_gate_scratch = scratch0; in LstmStepHybrid() local
1151 input_gate_scratch, is_input_all_zeros, is_aux_input_all_zeros, in LstmStepHybrid()
1189 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepHybrid()
1382 int16_t* input_gate_scratch = scratch0; in LstmStepInteger8x8_16() local
1418 kTfLiteActSigmoid, input_gate_scratch, context, scratch5); in LstmStepInteger8x8_16()
1447 input_gate_scratch, forget_gate_scratch, in LstmStepInteger8x8_16()
1737 float* input_gate_scratch = nullptr; in EvalFloat() local
[all …]
|