| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | Lstm.cpp | 140 VectorBatchVectorAdd(*inputGateBiasTensor, in LstmImpl()
160 VectorBatchVectorAdd(*forgetGateBiasTensor, in LstmImpl()
174 VectorBatchVectorAdd(*cellBiasTensor, in LstmImpl()
219 VectorBatchVectorAdd(*outputGateBiasTensor, in LstmImpl()
|
| H A D | RefQLstmWorkload.cpp | 353 VectorBatchVectorAdd(*inputGateBiasDecoder, in Execute()
401 VectorBatchVectorAdd(*forgetGateBiasDecoder, in Execute()
436 VectorBatchVectorAdd(*cellGateBiasDecoder, in Execute()
496 …VectorBatchVectorAdd(*outputGateBiasDecoder, numUnits, *outputGateDecoder, numBatches, *outputGate… in Execute()
|
| H A D | LstmUtils.hpp | 15 void VectorBatchVectorAdd(armnn::Decoder<float>& vector,
|
| H A D | LstmUtils.cpp | 16 void VectorBatchVectorAdd(armnn::Decoder<float>& vector, in VectorBatchVectorAdd() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 218 tensor_utils::VectorBatchVectorAdd(input_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
240 tensor_utils::VectorBatchVectorAdd(forget_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
257 tensor_utils::VectorBatchVectorAdd(cell_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
291 tensor_utils::VectorBatchVectorAdd(output_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | svdf.h | 57 tensor_utils::VectorBatchVectorAdd(bias_ptr, num_units, batch_size, in ApplyTimeWeightsBiasAndActivation()
136 tensor_utils::VectorBatchVectorAdd(bias_data, n_unit, n_batch, in EvalIntegerSVDF()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 896 tflite::tensor_utils::VectorBatchVectorAdd(input_gate_bias_buffer, n_cell, n_batch, in LSTMStep()
915 tflite::tensor_utils::VectorBatchVectorAdd(forget_gate_bias_buffer, n_cell, n_batch, in LSTMStep()
926 tflite::tensor_utils::VectorBatchVectorAdd(cell_bias_buffer, n_cell, n_batch, cell_scratch); in LSTMStep()
959 tflite::tensor_utils::VectorBatchVectorAdd(output_gate_bias_buffer, n_cell, n_batch, in LSTMStep()
|
| D | SVDF.cpp | 212 tflite::tensor_utils::VectorBatchVectorAdd(biasData, num_units, batch_size, outputData); in EvalFloat32()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | portable_tensor_utils.h | 51 void VectorBatchVectorAdd(const T* vector, int v_size, int n_batch, in VectorBatchVectorAdd() function
|
| H A D | tensor_utils_test.cc | 1676 VectorBatchVectorAdd(input, kVectorSize, kBatchSize, output.data()); in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 91 tensor_utils::VectorBatchVectorAdd(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 233 tensor_utils::VectorBatchVectorAdd(gate_bias, n_cell, n_batch, output); in CalculateLstmGateFloat()
449 tensor_utils::VectorBatchVectorAdd(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateHybrid()
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | LstmTestImpl.cpp | 45 VectorBatchVectorAdd(*vecDecoder, vSize, *batchVecDecoder, nBatch, *batchVecEncoder); in LstmUtilsVectorBatchVectorAddTestImpl()
|