Searched refs:cell_gate_scratch (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 194 float* cell_gate_scratch = scratch2; in LstmStepCalibration() local
232 params->activation, cell_gate_scratch, in LstmStepCalibration()
237 forget_gate_scratch, cell_gate_scratch, use_cifg, in LstmStepCalibration()
316 float* cell_gate_scratch = nullptr; in EvalCalibration() local
320 cell_gate_scratch = scratch_buffer_ptr; in EvalCalibration()
325 cell_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalCalibration()
376 input_gate_scratch, forget_gate_scratch, cell_gate_scratch, in EvalCalibration()
407 float* cell_gate_scratch_ptr = cell_gate_scratch + b * n_cell; in EvalCalibration()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 843 float* cell_gate_scratch = scratch2; in LstmStepFloat() local
879 params->activation, cell_gate_scratch, in LstmStepFloat()
883 forget_gate_scratch, cell_gate_scratch, use_cifg, in LstmStepFloat()
1043 float* cell_gate_scratch = scratch2; in LstmStepHybrid() local
1184 params->activation, cell_gate_scratch, is_input_all_zeros, in LstmStepHybrid()
1190 forget_gate_scratch, cell_gate_scratch, use_cifg, in LstmStepHybrid()
1384 int16_t* cell_gate_scratch = scratch2; in LstmStepInteger8x8_16() local
1444 cell_gate_scratch, context, scratch5); in LstmStepInteger8x8_16()
1448 cell_gate_scratch, use_cifg, quantized_cell_clip); in LstmStepInteger8x8_16()
1629 int16_t* cell_gate_scratch = scratch3; in LstmStepInteger8x8_8() local
[all …]
|