Searched refs:scratch_buffer_ptr (Results 1 – 3 of 3) sorted by relevance
314 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalCalibration() local320 cell_gate_scratch = scratch_buffer_ptr; in EvalCalibration()321 forget_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalCalibration()322 output_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalCalibration()324 input_gate_scratch = scratch_buffer_ptr; in EvalCalibration()325 cell_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalCalibration()326 forget_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalCalibration()327 output_gate_scratch = scratch_buffer_ptr + 3 * n_cell * n_batch; in EvalCalibration()
383 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalFloat() local389 cell_scratch = scratch_buffer_ptr; in EvalFloat()390 forget_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalFloat()391 output_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalFloat()393 input_gate_scratch = scratch_buffer_ptr; in EvalFloat()394 cell_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalFloat()395 forget_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalFloat()396 output_gate_scratch = scratch_buffer_ptr + 3 * n_cell * n_batch; in EvalFloat()
1736 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalFloat() local1742 cell_gate_scratch = scratch_buffer_ptr; in EvalFloat()1743 forget_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalFloat()1744 output_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalFloat()1746 input_gate_scratch = scratch_buffer_ptr; in EvalFloat()1747 cell_gate_scratch = scratch_buffer_ptr + n_cell * n_batch; in EvalFloat()1748 forget_gate_scratch = scratch_buffer_ptr + 2 * n_cell * n_batch; in EvalFloat()1749 output_gate_scratch = scratch_buffer_ptr + 3 * n_cell * n_batch; in EvalFloat()1933 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalHybrid() local1939 cell_gate_scratch = scratch_buffer_ptr; in EvalHybrid()[all …]