diff options
| -rw-r--r-- | src/profiling/test/FileOnlyProfilingDecoratorTests.cpp | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp b/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp index 31c1c7a0c5..91584e8787 100644 --- a/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp +++ b/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp @@ -80,8 +80,8 @@ BOOST_AUTO_TEST_CASE(TestFileOnlyProfiling) IConnectableLayer* input = net->AddInputLayer(0, "input"); - NormalizationDescriptor descriptor; - IConnectableLayer* normalize = net->AddNormalizationLayer(descriptor, "normalization"); + ElementwiseUnaryDescriptor descriptor(UnaryOperation::Sqrt); + IConnectableLayer* normalize = net->AddElementwiseUnaryLayer(descriptor, "normalization"); IConnectableLayer* output = net->AddOutputLayer(0, "output"); @@ -103,6 +103,11 @@ BOOST_AUTO_TEST_CASE(TestFileOnlyProfiling) // Creates structures for input & output. std::vector<float> inputData(16); std::vector<float> outputData(16); + for (unsigned int i = 0; i < 16; ++i) + { + inputData[i] = 9.0; + outputData[i] = 3.0; + } InputTensors inputTensors { |