Searched refs:input_modulation_gate_input (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLLSTMLayerQuantized.cpp | 405 TensorInfo input_modulation_gate_input; in validate() local
417 …input_modulation_gate_input = TensorInfo(TensorShape(output_size, batch_size), 1, DataType::QSYMM1… in validate()
418 …ARM_COMPUTE_RETURN_ON_ERROR(CLSlice::validate(&output_lowp, &input_modulation_gate_input, { 2 * ou… in validate()
432 … input_modulation_gate_input = TensorInfo(TensorShape(output_size), 1, DataType::QSYMM16, qsymm_3); in validate()
433 …ARM_COMPUTE_RETURN_ON_ERROR(CLSlice::validate(&output_lowp, &input_modulation_gate_input, { 2 * ou… in validate()
446 …const TensorInfo input_modulation_gate_output(input_modulation_gate_input.tensor_shape(), 1, DataT… in validate()
447 …ARM_COMPUTE_RETURN_ON_ERROR(CLActivationLayer::validate(&input_modulation_gate_input, &input_modul… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NELSTMLayerQuantized.cpp | 372 TensorInfo input_modulation_gate_input; in validate() local
384 …input_modulation_gate_input = TensorInfo(TensorShape(output_size, batch_size), 1, DataType::QSYMM1… in validate()
385 …ARM_COMPUTE_RETURN_ON_ERROR(NESlice::validate(&output_lowp, &input_modulation_gate_input, { 2 * ou… in validate()
399 … input_modulation_gate_input = TensorInfo(TensorShape(output_size), 1, DataType::QSYMM16, qsymm_3); in validate()
400 …ARM_COMPUTE_RETURN_ON_ERROR(NESlice::validate(&output_lowp, &input_modulation_gate_input, { 2 * ou… in validate()
413 …const TensorInfo input_modulation_gate_output(input_modulation_gate_input.tensor_shape(), 1, DataT… in validate()
414 …ARM_COMPUTE_RETURN_ON_ERROR(NEActivationLayer::validate(&input_modulation_gate_input, &input_modul… in validate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 373 F3 input_modulation_gate_input = F3::FromRaw( in LstmCell() local
376 gemmlowp::tanh(input_modulation_gate_input); in LstmCell()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTM.cpp | 160 F3 input_modulation_gate_input = F3::FromRaw( in quantizedLstmStep() local
163 gemmlowp::tanh(input_modulation_gate_input); in quantizedLstmStep()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 3286 F3 input_modulation_gate_input = in LstmCell() local
3290 gemmlowp::tanh(input_modulation_gate_input); in LstmCell()
3359 F3 input_modulation_gate_input = in LstmCell() local
3362 gemmlowp::tanh(input_modulation_gate_input); in LstmCell()
|
| H A D | optimized_ops.h | 2725 F3 input_modulation_gate_input = in LstmCell() local
2729 gemmlowp::tanh(input_modulation_gate_input); in LstmCell()
2798 F3 input_modulation_gate_input = in LstmCell() local
2801 gemmlowp::tanh(input_modulation_gate_input); in LstmCell()
|