diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-04-03 13:44:29 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:16 +0000 |
commit | d9eb27597eabe5b7c17520f4f9b3f8a282d72573 (patch) | |
tree | 9b2b7d74b0ef83623b18d6d4279a564e5b63d641 /examples/graph_mobilenet_qasymm8.cpp | |
parent | a8ca2b0cfe052c9a28b691317a674f28f495c139 (diff) | |
download | ComputeLibrary-d9eb27597eabe5b7c17520f4f9b3f8a282d72573.tar.gz |
COMPMID-797: Switch to new graph.
- Cleaned up build system
Change-Id: If2faa27ee5b31fa8b972836960ab3ef671059c8d
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/126435
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'examples/graph_mobilenet_qasymm8.cpp')
-rw-r--r-- | examples/graph_mobilenet_qasymm8.cpp | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/examples/graph_mobilenet_qasymm8.cpp b/examples/graph_mobilenet_qasymm8.cpp index 76b13dd851..cb49ffdfb4 100644 --- a/examples/graph_mobilenet_qasymm8.cpp +++ b/examples/graph_mobilenet_qasymm8.cpp @@ -21,14 +21,15 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ -#include "arm_compute/graph/Graph.h" -#include "arm_compute/graph/Nodes.h" +#include "arm_compute/graph.h" #include "support/ToolchainSupport.h" #include "utils/GraphUtils.h" #include "utils/Utils.h" -using namespace arm_compute; -using namespace arm_compute::graph; +#include <cstdlib> + +using namespace arm_compute::utils; +using namespace arm_compute::graph::frontend; using namespace arm_compute::graph_utils; /** Example demonstrating how to implement QASYMM8 MobileNet's network using the Compute Library's graph API @@ -36,9 +37,11 @@ using namespace arm_compute::graph_utils; * @param[in] argc Number of arguments * @param[in] argv Arguments ( [optional] Target (0 = NEON, 1 = OpenCL, 2 = OpenCL with Tuner), [optional] Path to the weights folder, [optional] npy_input, [optional] labels ) */ -class GraphMobileNetQASYMM8Example : public utils::Example +class GraphMobileNetQASYMM8Example : public Example { public: + //FIXME: Missing quantization info to the tensor descriptor (Giorgio is working on it) +#if 0 void do_setup(int argc, char **argv) override { std::string data_path; /* Path to the trainable data */ @@ -217,6 +220,8 @@ private: return BranchLayer(std::move(sg)); } +#endif /* if 0 */ + Stream graph { 0, "MobileNetV1_QASYMM8" }; }; /** Main program for MobileNetQASYMM8 * @@ -225,5 +230,5 @@ private: */ int main(int argc, char **argv) { - return utils::run_example<GraphMobileNetQASYMM8Example>(argc, argv); + return arm_compute::utils::run_example<GraphMobileNetQASYMM8Example>(argc, argv); } |