Home
last modified time | relevance | path

Searched refs:output_gate_scratch (Results 1 – 1 of 1) sorted by relevance

/frameworks/ml/nn/common/operations/
DLSTM.cpp799 float* output_gate_scratch = nullptr; in LSTMStep() local
803 output_gate_scratch = cell_scratch + 2 * n_cell * n_batch; in LSTMStep()
808 output_gate_scratch = input_gate_scratch + 3 * n_cell * n_batch; in LSTMStep()
822 output_gate_scratch); in LSTMStep()
830 std::fill_n(output_gate_scratch, n_cell * n_batch, 0.0f); in LSTMStep()
847 output_gate_scratch, /*result_stride*/ 1); in LSTMStep()
866 output_gate_scratch, /*result_stride=*/1); in LSTMStep()
884 output_gate_scratch, /*result_stride*/ 1); in LSTMStep()
953 n_batch, output_gate_scratch); in LSTMStep()
956 tflite::tensor_utils::MeanStddevNormalization(output_gate_scratch, output_gate_scratch, in LSTMStep()
[all …]