Searched refs:scratch_buffer_ptr (Results 1 – 3 of 3) sorted by relevance
392 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalFloat() local398 cell_scratch = scratch_buffer_ptr; in EvalFloat()399 forget_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalFloat()400 output_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalFloat()402 input_gate_scratch = scratch_buffer_ptr; in EvalFloat()403 cell_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalFloat()404 forget_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalFloat()405 output_gate_scratch = scratch_buffer_ptr + 3 * n_cell * n_batch; in EvalFloat()
315 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalCalibration() local321 cell_gate_scratch = scratch_buffer_ptr; in EvalCalibration()322 forget_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalCalibration()323 output_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalCalibration()325 input_gate_scratch = scratch_buffer_ptr; in EvalCalibration()326 cell_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalCalibration()327 forget_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()
1795 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalFloat() local1802 cell_gate_scratch = scratch_buffer_ptr; in EvalFloat()1803 forget_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalFloat()1804 output_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalFloat()1805 accumulation_scratch_buffer = scratch_buffer_ptr + 3 * n_cell * n_batch; in EvalFloat()1807 input_gate_scratch = scratch_buffer_ptr; in EvalFloat()1808 cell_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalFloat()1809 forget_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalFloat()1810 output_gate_scratch = scratch_buffer_ptr + 3 * n_cell * n_batch; in EvalFloat()1811 accumulation_scratch_buffer = scratch_buffer_ptr + 4 * n_cell * n_batch; in EvalFloat()[all …]