Searched refs:input_modulation_gate_output (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 375 F0 input_modulation_gate_output = in LstmCell() local
387 input_gate_output * input_modulation_gate_output; in LstmCell()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLLSTMLayerQuantized.cpp | 446 …const TensorInfo input_modulation_gate_output(input_modulation_gate_input.tensor_shape(), 1, DataT… in validate() local
447 …OR(CLActivationLayer::validate(&input_modulation_gate_input, &input_modulation_gate_output, Activa… in validate()
458 …RROR(CLPixelWiseMultiplication::validate(&input_gate_output, &input_modulation_gate_output, &cell_… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NELSTMLayerQuantized.cpp | 413 …const TensorInfo input_modulation_gate_output(input_modulation_gate_input.tensor_shape(), 1, DataT… in validate() local
414 …OR(NEActivationLayer::validate(&input_modulation_gate_input, &input_modulation_gate_output, Activa… in validate()
425 …RROR(NEPixelWiseMultiplication::validate(&input_gate_output, &input_modulation_gate_output, &cell_… in validate()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTM.cpp | 162 F0 input_modulation_gate_output = 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 | 3289 F0 input_modulation_gate_output = in LstmCell() local
3301 input_gate_output * input_modulation_gate_output; in LstmCell()
3361 F0 input_modulation_gate_output = in LstmCell() local
3371 input_gate_output * input_modulation_gate_output; in LstmCell()
|
| H A D | optimized_ops.h | 2728 F0 input_modulation_gate_output = in LstmCell() local
2740 input_gate_output * input_modulation_gate_output; in LstmCell()
2800 F0 input_modulation_gate_output = in LstmCell() local
2810 input_gate_output * input_modulation_gate_output; in LstmCell()
|