Searched refs:input_gate_output (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLLSTMLayerQuantized.cpp | 443 …const TensorInfo input_gate_output(input_gate_input.tensor_shape(), 1, DataType::QSYMM16, qsymm_0); in validate() local
444 …ARM_COMPUTE_RETURN_ON_ERROR(CLActivationLayer::validate(&input_gate_input, &input_gate_output, Act… in validate()
457 … const TensorInfo cell_state_tmp2(input_gate_output.tensor_shape(), 1, DataType::QSYMM16, qsymm_4); in validate()
458 …ARM_COMPUTE_RETURN_ON_ERROR(CLPixelWiseMultiplication::validate(&input_gate_output, &input_modulat… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NELSTMLayerQuantized.cpp | 410 …const TensorInfo input_gate_output(input_gate_input.tensor_shape(), 1, DataType::QSYMM16, qsymm_0); in validate() local
411 …ARM_COMPUTE_RETURN_ON_ERROR(NEActivationLayer::validate(&input_gate_input, &input_gate_output, Act… in validate()
424 … const TensorInfo cell_state_tmp2(input_gate_output.tensor_shape(), 1, DataType::QSYMM16, qsymm_4); in validate()
425 …ARM_COMPUTE_RETURN_ON_ERROR(NEPixelWiseMultiplication::validate(&input_gate_output, &input_modulat… in validate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 370 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in LstmCell() local
387 input_gate_output * input_modulation_gate_output; in LstmCell()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTM.cpp | 157 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in quantizedLstmStep() local
174 input_gate_output * input_modulation_gate_output; in quantizedLstmStep()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 3283 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in LstmCell() local
3301 input_gate_output * input_modulation_gate_output; in LstmCell()
3356 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in LstmCell() local
3371 input_gate_output * input_modulation_gate_output; in LstmCell()
|
| H A D | optimized_ops.h | 2722 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in LstmCell() local
2740 input_gate_output * input_modulation_gate_output; in LstmCell()
2795 F0 input_gate_output = gemmlowp::logistic(input_gate_input); in LstmCell() local
2810 input_gate_output * input_modulation_gate_output; in LstmCell()
|