Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlstm_cell.h380 F0 forget_gate_output = gemmlowp::logistic(forget_gate_input); in LstmCell() local
389 FS prev_state_times_forget_state = forget_gate_output * prev_state; in LstmCell()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNELSTMLayerQuantized.cpp407 …const TensorInfo forget_gate_output(forget_gate_input.tensor_shape(), 1, DataType::QSYMM16, qsymm_… in validate() local
408 …ARM_COMPUTE_RETURN_ON_ERROR(NEActivationLayer::validate(&forget_gate_input, &forget_gate_output, A… in validate()
420 …const TensorInfo cell_state_tmp1(forget_gate_output.tensor_shape(), 1, DataType::QSYMM16, qsymm_4); in validate()
421 …ARM_COMPUTE_RETURN_ON_ERROR(NEPixelWiseMultiplication::validate(&forget_gate_output, cell_state_in… in validate()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayerQuantized.cpp440 …const TensorInfo forget_gate_output(forget_gate_input.tensor_shape(), 1, DataType::QSYMM16, qsymm_… in validate() local
441 …ARM_COMPUTE_RETURN_ON_ERROR(CLActivationLayer::validate(&forget_gate_input, &forget_gate_output, A… in validate()
453 …const TensorInfo cell_state_tmp1(forget_gate_output.tensor_shape(), 1, DataType::QSYMM16, qsymm_4); in validate()
454 …ARM_COMPUTE_RETURN_ON_ERROR(CLPixelWiseMultiplication::validate(&forget_gate_output, cell_state_in… in validate()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3294 F0 forget_gate_output = gemmlowp::logistic(forget_gate_input); in LstmCell() local
3304 FS prev_state_times_forget_state = forget_gate_output * prev_state; in LstmCell()
3365 F0 forget_gate_output = gemmlowp::logistic(forget_gate_input); in LstmCell() local
3373 FS prev_state_times_forget_state = forget_gate_output * prev_state; in LstmCell()
Doptimized_ops.h2733 F0 forget_gate_output = gemmlowp::logistic(forget_gate_input); in LstmCell() local
2743 FS prev_state_times_forget_state = forget_gate_output * prev_state; in LstmCell()
2804 F0 forget_gate_output = gemmlowp::logistic(forget_gate_input); in LstmCell() local
2812 FS prev_state_times_forget_state = forget_gate_output * prev_state; in LstmCell()