diff options
Diffstat (limited to 'test/Convolution2D.hpp')
| -rw-r--r-- | test/Convolution2D.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/Convolution2D.hpp b/test/Convolution2D.hpp index 46b46549..180f57e2 100644 --- a/test/Convolution2D.hpp +++ b/test/Convolution2D.hpp @@ -102,7 +102,7 @@ void PaddingTestImpl(android::nn::PaddingScheme paddingScheme, bool fp16Enabled AddPoolAndSetData(6, request, indata); // add memory for the output - android::sp<IMemory> outMemory = AddPoolAndGetData(outSize, request); + android::sp<IMemory> outMemory = AddPoolAndGetData<float>(outSize, request); float* outdata = reinterpret_cast<float*>(static_cast<void*>(outMemory->getPointer())); // run the execution |