diff options
| -rw-r--r-- | examples/graph_lenet.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/examples/graph_lenet.cpp b/examples/graph_lenet.cpp index 676fdb9ce2..1f04ff8a50 100644 --- a/examples/graph_lenet.cpp +++ b/examples/graph_lenet.cpp @@ -93,16 +93,18 @@ void main_graph_lenet(int argc, const char **argv) } // Check if OpenCL is available and initialize the scheduler + Hint hint = Hint::NEON; if(arm_compute::opencl_is_available()) { arm_compute::CLScheduler::get().default_init(); + hint = Hint::OPENCL; } Graph graph; graph.set_info_enablement(true); //conv1 << pool1 << conv2 << pool2 << fc1 << act1 << fc2 << smx - graph << Hint::OPENCL + graph << hint << Tensor(TensorInfo(TensorShape(28U, 28U, 1U, batches), 1, DataType::F32), DummyAccessor()) << ConvolutionLayer( 5U, 5U, 20U, |