Searched refs:VectorBatchVectorAdd (Results 1 – 11 of 11) sorted by relevance
| /external/armnn/src/backends/reference/workloads/ |
| D | Lstm.cpp | 140 VectorBatchVectorAdd(*inputGateBiasTensor, in LstmImpl()
160 VectorBatchVectorAdd(*forgetGateBiasTensor, in LstmImpl()
174 VectorBatchVectorAdd(*cellBiasTensor, in LstmImpl()
219 VectorBatchVectorAdd(*outputGateBiasTensor, in LstmImpl()
|
| 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()
|
| D | LstmUtils.hpp | 15 void VectorBatchVectorAdd(armnn::Decoder<float>& vector,
|
| D | LstmUtils.cpp | 16 void VectorBatchVectorAdd(armnn::Decoder<float>& vector, in VectorBatchVectorAdd() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| 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()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| 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()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | portable_tensor_utils.h | 51 void VectorBatchVectorAdd(const T* vector, int v_size, int n_batch, in VectorBatchVectorAdd() function
|
| D | tensor_utils_test.cc | 1676 VectorBatchVectorAdd(input, kVectorSize, kBatchSize, output.data()); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 91 tensor_utils::VectorBatchVectorAdd(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| 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()
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | LstmTestImpl.cpp | 45 VectorBatchVectorAdd(*vecDecoder, vSize, *batchVecDecoder, nBatch, *batchVecEncoder); in LstmUtilsVectorBatchVectorAddTestImpl()
|