diff options
author | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
---|---|---|
committer | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
commit | c577f2c6a3b4ddb6ba87a882723c53a248afbeba (patch) | |
tree | bd7d4c148df27f8be6649d313efb24f536b7cf34 /src/armnn/backends/RefWorkloads/FullyConnected.cpp | |
parent | 4c7098bfeab1ffe1cdc77f6c15548d3e73274746 (diff) | |
download | armnn-c577f2c6a3b4ddb6ba87a882723c53a248afbeba.tar.gz |
Release 18.08
Diffstat (limited to 'src/armnn/backends/RefWorkloads/FullyConnected.cpp')
-rw-r--r-- | src/armnn/backends/RefWorkloads/FullyConnected.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/backends/RefWorkloads/FullyConnected.cpp b/src/armnn/backends/RefWorkloads/FullyConnected.cpp index 8ba11d19c6..1a8263b9a1 100644 --- a/src/armnn/backends/RefWorkloads/FullyConnected.cpp +++ b/src/armnn/backends/RefWorkloads/FullyConnected.cpp @@ -18,11 +18,11 @@ void FullyConnected(const float* inputData, const float* biasData, bool transposeWeights) { - unsigned int N = outputTensorInfo.GetShape()[1]; // Output Vector Size + unsigned int N = outputTensorInfo.GetShape()[1]; // Outputs Vector Size. - BOOST_ASSERT(inputTensorInfo.GetNumDimensions() > 1); // Need some data + BOOST_ASSERT(inputTensorInfo.GetNumDimensions() > 1); // Needs some data. - unsigned int K = 1; // Total number of activations in the input + unsigned int K = 1; // Total number of activations in the input. for (unsigned int i = 1; i < inputTensorInfo.GetNumDimensions(); i++) { K *= inputTensorInfo.GetShape()[i]; |