Searched defs:output_gate_scratch (Results 1 - 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| H A D | LSTM.cpp | 318 float* output_gate_scratch = nullptr; local
322 output_gate_scratch = cell_scratch + 2 * n_cell * n_batch;
327 output_gate_scratch = input_gate_scratch + 3 * n_cell * n_batch;
340 n_cell, n_batch, output_gate_scratch); local
356 GetBuffer<float>(input_), n_batch, output_gate_scratch, /*result_stride*/1); local
372 GetBuffer<float>(output_state_in_), n_batch, output_gate_scratch, /*result_stride*/1); local
423 GetBuffer<float>(cell_state_out_), n_batch, output_gate_scratch); local
425 tflite::tensor_utils::ApplySigmoidToVector(output_gate_scratch, n_batch * n_cell,
426 output_gate_scratch);
431 tflite::tensor_utils::VectorVectorCwiseProduct(output_gate_scratch,
448 output_gate_scratch, n_batch, GetBuffer<float>(output_), local
[all...] |
Completed in 57 milliseconds