Searched refs:output_gate_output (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | lstm_cell.h | 384 F0 output_gate_output = gemmlowp::logistic(output_gate_input); in LstmCell() local 403 F0 output_activ_int16 = output_gate_output * gemmlowp::tanh(new_state_f3); in LstmCell()
|
/external/ComputeLibrary/src/runtime/NEON/functions/ |
D | NELSTMLayerQuantized.cpp | 416 …const TensorInfo output_gate_output(output_gate_input.tensor_shape(), 1, DataType::QSYMM16, qsymm_… in validate() local 417 …ARM_COMPUTE_RETURN_ON_ERROR(NEActivationLayer::validate(&output_gate_input, &output_gate_output, A… in validate() 435 …const TensorInfo output_state_out_symm(output_gate_output.tensor_shape(), 1, DataType::QSYMM16, qs… in validate() 436 …N_ON_ERROR(NEPixelWiseMultiplication::validate(&output_state_tmp, &output_gate_output, &output_sta… in validate()
|
/external/ComputeLibrary/src/runtime/CL/functions/ |
D | CLLSTMLayerQuantized.cpp | 449 …const TensorInfo output_gate_output(output_gate_input.tensor_shape(), 1, DataType::QSYMM16, qsymm_… in validate() local 450 …ARM_COMPUTE_RETURN_ON_ERROR(CLActivationLayer::validate(&output_gate_input, &output_gate_output, A… in validate() 468 …const TensorInfo output_state_out_symm(output_gate_output.tensor_shape(), 1, DataType::QSYMM16, qs… in validate() 469 …N_ON_ERROR(CLPixelWiseMultiplication::validate(&output_state_tmp, &output_gate_output, &output_sta… in validate()
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | legacy_optimized_ops.h | 3298 F0 output_gate_output = gemmlowp::logistic(output_gate_input); in LstmCell() local 3318 F0 output_activ_int16 = output_gate_output * gemmlowp::tanh(new_state_f3); in LstmCell() 3368 F0 output_gate_output = gemmlowp::logistic(output_gate_input); in LstmCell() local 3387 F0 output_activ_int16 = output_gate_output * gemmlowp::tanh(new_state_f3); in LstmCell()
|
D | optimized_ops.h | 2737 F0 output_gate_output = gemmlowp::logistic(output_gate_input); in LstmCell() local 2757 F0 output_activ_int16 = output_gate_output * gemmlowp::tanh(new_state_f3); in LstmCell() 2807 F0 output_gate_output = gemmlowp::logistic(output_gate_input); in LstmCell() local 2826 F0 output_activ_int16 = output_gate_output * gemmlowp::tanh(new_state_f3); in LstmCell()
|