| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | Lstm.cpp | 78 VectorBatchVectorAssign(*inputGateBiasTensor, in LstmImpl()
81 VectorBatchVectorAssign(*forgetGateBiasTensor, in LstmImpl()
83 VectorBatchVectorAssign(*cellBiasTensor, in LstmImpl()
85 VectorBatchVectorAssign(*outputGateBiasTensor, in LstmImpl()
240 VectorBatchVectorAssign(*projectionBiasTensor, in LstmImpl()
|
| H A D | LstmUtils.hpp | 39 void VectorBatchVectorAssign(armnn::Decoder<float>& vector,
|
| H A D | LstmUtils.cpp | 113 void VectorBatchVectorAssign(armnn::Decoder<float>& vector, in VectorBatchVectorAssign() function
|
| H A D | RefQLstmWorkload.cpp | 524 … VectorBatchVectorAssign(*projectionBiasDecoder, outputSize, numBatches, *outputInt16Encoder); in Execute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 87 tensor_utils::VectorBatchVectorAssign(input_gate_bias_ptr, n_cell, in LstmStepWithAuxInput()
90 tensor_utils::VectorBatchVectorAssign(forget_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
92 tensor_utils::VectorBatchVectorAssign(cell_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
94 tensor_utils::VectorBatchVectorAssign(output_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 814 tflite::tensor_utils::VectorBatchVectorAssign(input_gate_bias_buffer, n_cell, n_batch, in LSTMStep()
817 tflite::tensor_utils::VectorBatchVectorAssign(forget_gate_bias_buffer, n_cell, n_batch, in LSTMStep()
819 tflite::tensor_utils::VectorBatchVectorAssign(cell_bias_buffer, n_cell, n_batch, in LSTMStep()
821 tflite::tensor_utils::VectorBatchVectorAssign(output_gate_bias_buffer, n_cell, n_batch, in LSTMStep()
973 tflite::tensor_utils::VectorBatchVectorAssign(projection_bias_buffer, n_output, n_batch, in LSTMStep()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | kernel_utils.cc | 50 tensor_utils::VectorBatchVectorAssign(bias_ptr, num_units, batch_size, in RnnBatchStep()
185 tensor_utils::VectorBatchVectorAssign(bias_ptr, num_units, batch_size, in RnnBatchStep()
|
| H A D | portable_tensor_utils.h | 112 void VectorBatchVectorAssign(const T* vector, int v_size, int n_batch, in VectorBatchVectorAssign() function
|
| H A D | tensor_utils_test.cc | 1686 VectorBatchVectorAssign(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 | 60 tensor_utils::VectorBatchVectorAssign(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
143 tensor_utils::VectorBatchVectorAssign(projection_bias, n_output, n_batch, in CalculateLstmOutputCalibration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | fully_connected.cc | 491 tensor_utils::VectorBatchVectorAssign(GetTensorData<float>(bias), num_units, in EvalPie()
529 tensor_utils::VectorBatchVectorAssign(GetTensorData<float>(bias), num_units, in EvalHybridDense()
609 tensor_utils::VectorBatchVectorAssign(GetTensorData<float>(bias), in EvalSparseHybridImpl()
|
| H A D | lstm_eval.cc | 200 tensor_utils::VectorBatchVectorAssign(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
320 tensor_utils::VectorBatchVectorAssign(projection_bias, n_output, n_batch, in CalculateLstmOutputFloat()
385 tensor_utils::VectorBatchVectorAssign(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateHybrid()
493 tensor_utils::VectorBatchVectorAssign(projection_bias, n_output, n_batch, in CalculateLstmOutputHybrid()
|