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