diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-10-04 16:53:58 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | ff421f2100e0e9e532f5fe78585300546af61690 (patch) | |
| tree | 9ba5a1bfe64b5b10f70c64a965f9c5ca14de9ce3 /examples/graph_alexnet.cpp | |
| parent | 925ca0f7402115da3bffb21c04fca0bc822c9b38 (diff) | |
| download | ComputeLibrary-ff421f2100e0e9e532f5fe78585300546af61690.tar.gz | |
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 <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'examples/graph_alexnet.cpp')
| -rw-r--r-- | examples/graph_alexnet.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
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"), |