LSTM.cpp | 317 float* forget_gate_scratch = nullptr; local 321 forget_gate_scratch = cell_scratch + n_cell * n_batch; 326 forget_gate_scratch = input_gate_scratch + 2 * n_cell * n_batch; 336 n_cell, n_batch, forget_gate_scratch); local 350 GetBuffer<float>(input_), n_batch, forget_gate_scratch, /*result_stride*/1); local 366 GetBuffer<float>(output_state_in_), n_batch, forget_gate_scratch, /*result_stride*/1); local 390 GetBuffer<float>(cell_state_in_), n_batch, forget_gate_scratch); local 392 tflite::tensor_utils::ApplySigmoidToVector(forget_gate_scratch, 394 forget_gate_scratch); 398 forget_gate_scratch, GetBuffer<float>(cell_state_in_), n_batch * n_cell [all...] |