diff options
Diffstat (limited to 'examples/graph_alexnet.cpp')
| -rw-r--r-- | examples/graph_alexnet.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/examples/graph_alexnet.cpp b/examples/graph_alexnet.cpp index 418d1ea071..6a3c14b7bb 100644 --- a/examples/graph_alexnet.cpp +++ b/examples/graph_alexnet.cpp @@ -21,7 +21,7 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ -#include "arm_compute/graph2.h" +#include "arm_compute/graph.h" #include "support/ToolchainSupport.h" #include "utils/GraphUtils.h" #include "utils/Utils.h" @@ -31,7 +31,7 @@ #include <memory> using namespace arm_compute::utils; -using namespace arm_compute::graph2::frontend; +using namespace arm_compute::graph::frontend; using namespace arm_compute::graph_utils; /** Example demonstrating how to implement AlexNet's network using the Compute Library's graph API @@ -54,7 +54,7 @@ public: // Set target. 0 (NEON), 1 (OpenCL), 2 (OpenCL with Tuner). By default it is NEON const int target = argc > 1 ? std::strtol(argv[1], nullptr, 10) : 0; - Target target_hint = set_target_hint2(target); + Target target_hint = set_target_hint(target); // TODO (geopin01) : Get GPU target somehow and set gemm also for midgard ? const bool is_gemm_convolution5x5 = (target_hint == Target::NEON); |