Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/runtime/NEON/functions/
DNELSTMLayerQuantized.cpp404 …const TensorInfo input_gate_output(input_gate_input.tensor_shape(), 1, DataType::QSYMM16, qsymm_0); in validate() local
405 …ARM_COMPUTE_RETURN_ON_ERROR(NEActivationLayer::validate(&input_gate_input, &input_gate_output, Act… in validate()
418 … const TensorInfo cell_state_tmp2(input_gate_output.tensor_shape(), 1, DataType::QSYMM16, qsymm_4); in validate()
419 …ARM_COMPUTE_RETURN_ON_ERROR(NEPixelWiseMultiplication::validate(&input_gate_output, &input_modulat… in validate()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayerQuantized.cpp433 …const TensorInfo input_gate_output(input_gate_input.tensor_shape(), 1, DataType::QSYMM16, qsymm_0); in validate() local
434 …ARM_COMPUTE_RETURN_ON_ERROR(CLActivationLayer::validate(&input_gate_input, &input_gate_output, Act… in validate()
447 … const TensorInfo cell_state_tmp2(input_gate_output.tensor_shape(), 1, DataType::QSYMM16, qsymm_4); in validate()
448 …ARM_COMPUTE_RETURN_ON_ERROR(CLPixelWiseMultiplication::validate(&input_gate_output, &input_modulat… in validate()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h784 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in LstmCell() local
801 input_gate_output * input_modulation_gate_output; in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3281 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in LstmCell() local
3299 input_gate_output * input_modulation_gate_output; in LstmCell()
3354 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in LstmCell() local
3369 input_gate_output * input_modulation_gate_output; in LstmCell()
Doptimized_ops.h3047 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in LstmCell() local
3065 input_gate_output * input_modulation_gate_output; in LstmCell()
3120 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in LstmCell() local
3135 input_gate_output * input_modulation_gate_output; in LstmCell()