From ff421f2100e0e9e532f5fe78585300546af61690 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 4 Oct 2017 16:53:58 +0100 Subject: COMPMID-601: Add GraphContext GraphContext hold all the information about the hints that need to be passed in the nodes. As these might expand, it serves as a centralized class for such information. Change-Id: I0b5527630fb97cc5fa500db0bac8307ff2ea36e6 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/90300 Tested-by: Kaizen Reviewed-by: Anthony Barbier --- examples/graph_alexnet.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'examples/graph_alexnet.cpp') diff --git a/examples/graph_alexnet.cpp b/examples/graph_alexnet.cpp index cf5f635d33..9c736c5df1 100644 --- a/examples/graph_alexnet.cpp +++ b/examples/graph_alexnet.cpp @@ -94,11 +94,11 @@ void main_graph_alexnet(int argc, const char **argv) } // Check if OpenCL is available and initialize the scheduler - Hint hint = Hint::NEON; + TargetHint hint = TargetHint::NEON; if(arm_compute::opencl_is_available()) { arm_compute::CLScheduler::get().default_init(); - hint = Hint::OPENCL; + hint = TargetHint::OPENCL; } Graph graph; @@ -116,6 +116,7 @@ void main_graph_alexnet(int argc, const char **argv) << NormalizationLayer(NormalizationLayerInfo(NormType::CROSS_MAP, 5, 0.0001f, 0.75f)) << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, 0, 0))) // Layer 2 + << ConvolutionMethodHint::DIRECT << ConvolutionLayer( 5U, 5U, 256U, get_accessor(data_path, "/cnn_data/alexnet_model/conv2_w.npy"), -- cgit v1.2.1