diff options
author | Gian Marco <gianmarco.iodice@arm.com> | 2018-02-05 08:54:54 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:45:42 +0000 |
commit | 2d40555fff73ac8c23b9a21c3b4fc2a9fa01eb2e (patch) | |
tree | 519ecfd9df8df2701bb06ac16d220c9ac0236a93 /examples/graph_inception_v3.cpp | |
parent | bc4484acbbe3b6c1f06d25c1006ebd53f037767c (diff) | |
download | ComputeLibrary-2d40555fff73ac8c23b9a21c3b4fc2a9fa01eb2e.tar.gz |
COMPMID-765 - Used GEMM instead of Direct into the GoogleNet graph
example
Change-Id: Ic639d51fb5dd4f78912a9b11abc7df79d205a22b
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/118843
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'examples/graph_inception_v3.cpp')
-rw-r--r-- | examples/graph_inception_v3.cpp | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/examples/graph_inception_v3.cpp b/examples/graph_inception_v3.cpp index 9bc82977c7..6e31f5e669 100644 --- a/examples/graph_inception_v3.cpp +++ b/examples/graph_inception_v3.cpp @@ -53,8 +53,7 @@ public: constexpr float std = 255.f; /* Standard deviation value to divide from the channels */ // Set target. 0 (NEON), 1 (OpenCL). By default it is NEON - TargetHint target_hint = set_target_hint(argc > 1 ? std::strtol(argv[1], nullptr, 10) : 0); - ConvolutionMethodHint convolution_hint = ConvolutionMethodHint::DIRECT; + TargetHint target_hint = set_target_hint(argc > 1 ? std::strtol(argv[1], nullptr, 10) : 0); // Parse arguments if(argc < 2) @@ -88,10 +87,10 @@ public: label = argv[4]; } - graph << target_hint << convolution_hint << Tensor(TensorInfo(TensorShape(299U, 299U, 3U, 1U), 1, DataType::F32), - get_input_accessor(image, - mean, mean, mean, - std, std, std, false /* Do not convert to BGR */)) + graph << target_hint << Tensor(TensorInfo(TensorShape(299U, 299U, 3U, 1U), 1, DataType::F32), + get_input_accessor(image, + mean, mean, mean, + std, std, std, false /* Do not convert to BGR */)) << ConvolutionLayer(3U, 3U, 32U, get_weights_accessor(data_path, "/cnn_data/inceptionv3_model/Conv2d_1a_3x3_weights.npy"), |