Searched refs:input_gate_scratch_ptr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 473 float* input_gate_scratch_ptr = in EvalFloat() local
506 activation_state_ptr, cell_state_ptr, input_gate_scratch_ptr, in EvalFloat()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 404 float* input_gate_scratch_ptr = in EvalCalibration() local
437 output_state_ptr, cell_state_ptr, input_gate_scratch_ptr, in EvalCalibration()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 1825 float* input_gate_scratch_ptr = in EvalFloat() local
1858 output_state_ptr, cell_state_ptr, input_gate_scratch_ptr, in EvalFloat()
2070 float* input_gate_scratch_ptr = in EvalHybrid() local
2128 output_batch_leading_dim, input_gate_scratch_ptr, in EvalHybrid()
|