Searched refs:batchVector (Results 1 – 3 of 3) sorted by relevance
/external/armnn/src/backends/reference/workloads/ |
D | LstmUtils.cpp | 18 armnn::Decoder<float>& batchVector, in VectorBatchVectorAdd() argument 26 outResult.Set(batchVector.Get() + vector.Get()); in VectorBatchVectorAdd() 29 ++batchVector; in VectorBatchVectorAdd() 33 batchVector -= vSize * nBatch; in VectorBatchVectorAdd() 133 armnn::Decoder<float>& batchVector, in VectorBatchVectorCwiseProductAccumulate() argument 141 outResult.Set(outResult.Get() + vector.Get() * batchVector.Get()); in VectorBatchVectorCwiseProductAccumulate() 144 ++batchVector; in VectorBatchVectorCwiseProductAccumulate() 148 batchVector -= vSize * nBatch; in VectorBatchVectorCwiseProductAccumulate() 154 armnn::Decoder<float>& batchVector, in VectorBatchVectorCwiseProduct() argument 162 outResult.Set(vector.Get() * batchVector.Get()); in VectorBatchVectorCwiseProduct() [all …]
|
D | LstmUtils.hpp | 17 armnn::Decoder<float>& batchVector, 46 armnn::Decoder<float>& batchVector, 52 armnn::Decoder<float>& batchVector,
|
/external/armnn/src/backends/backendsCommon/test/layerTests/ |
D | LstmTestImpl.cpp | 2512 std::vector<float> batchVector = in LstmUtilsVectorBatchVectorCwiseProductTest() local 2557 …eturn LstmUtilsVectorBatchVectorCwiseProductTestImpl<armnn::DataType::Float32>(vector, batchVector, in LstmUtilsVectorBatchVectorCwiseProductTest() 2569 std::vector<float> batchVector = in LstmUtilsVectorBatchVectorAddTest() local 2581 return LstmUtilsVectorBatchVectorAddTestImpl<armnn::DataType::Float32>(vector, batchVector, in LstmUtilsVectorBatchVectorAddTest()
|