| <!-- HTML header for doxygen 1.8.9.1--> |
| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| <html xmlns="http://www.w3.org/1999/xhtml"> |
| <head> |
| <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| <meta http-equiv="X-UA-Compatible" content="IE=9"/> |
| <meta name="generator" content="Doxygen 1.8.6"/> |
| <meta name="robots" content="NOINDEX, NOFOLLOW" /> <!-- Prevent indexing by search engines --> |
| <title>Compute Library: examples/graph_lenet.cpp File Reference</title> |
| <link href="tabs.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="jquery.js"></script> |
| <script type="text/javascript" src="dynsections.js"></script> |
| <link href="navtree.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="resize.js"></script> |
| <script type="text/javascript" src="navtree.js"></script> |
| <script type="text/javascript"> |
| $(document).ready(initResizable); |
| $(window).load(resizeHeight); |
| </script> |
| <link href="search/search.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="search/search.js"></script> |
| <script type="text/javascript"> |
| $(document).ready(function() { searchBox.OnSelectItem(0); }); |
| </script> |
| <script type="text/x-mathjax-config"> |
| MathJax.Hub.Config({ |
| extensions: ["tex2jax.js"], |
| jax: ["input/TeX","output/HTML-CSS"], |
| }); |
| </script><script src="http://cdn.mathjax.org/mathjax/latest/MathJax.js"></script> |
| <link href="doxygen.css" rel="stylesheet" type="text/css" /> |
| </head> |
| <body> |
| <div id="top"><!-- do not remove this div, it is closed by doxygen! --> |
| <div id="titlearea"> |
| <table cellspacing="0" cellpadding="0"> |
| <tbody> |
| <tr style="height: 56px;"> |
| <td style="padding-left: 0.5em;"> |
| <div id="projectname">Compute Library |
|  <span id="projectnumber">17.10</span> |
| </div> |
| </td> |
| </tr> |
| </tbody> |
| </table> |
| </div> |
| <!-- end header part --> |
| <!-- Generated by Doxygen 1.8.6 --> |
| <script type="text/javascript"> |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| </script> |
| <div id="navrow1" class="tabs"> |
| <ul class="tablist"> |
| <li><a href="index.xhtml"><span>Main Page</span></a></li> |
| <li><a href="pages.xhtml"><span>Related Pages</span></a></li> |
| <li><a href="namespaces.xhtml"><span>Namespaces</span></a></li> |
| <li><a href="annotated.xhtml"><span>Data Structures</span></a></li> |
| <li class="current"><a href="files.xhtml"><span>Files</span></a></li> |
| <li> |
| <div id="MSearchBox" class="MSearchBoxInactive"> |
| <span class="left"> |
| <img id="MSearchSelect" src="search/mag_sel.png" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| alt=""/> |
| <input type="text" id="MSearchField" value="Search" accesskey="S" |
| onfocus="searchBox.OnSearchFieldFocus(true)" |
| onblur="searchBox.OnSearchFieldFocus(false)" |
| onkeyup="searchBox.OnSearchFieldChange(event)"/> |
| </span><span class="right"> |
| <a id="MSearchClose" href="javascript:searchBox.CloseResultsWindow()"><img id="MSearchCloseImg" border="0" src="search/close.png" alt=""/></a> |
| </span> |
| </div> |
| </li> |
| </ul> |
| </div> |
| <div id="navrow2" class="tabs2"> |
| <ul class="tablist"> |
| <li><a href="files.xhtml"><span>File List</span></a></li> |
| <li><a href="globals.xhtml"><span>Globals</span></a></li> |
| </ul> |
| </div> |
| </div><!-- top --> |
| <div id="side-nav" class="ui-resizable side-nav-resizable"> |
| <div id="nav-tree"> |
| <div id="nav-tree-contents"> |
| <div id="nav-sync" class="sync"></div> |
| </div> |
| </div> |
| <div id="splitbar" style="-moz-user-select:none;" |
| class="ui-resizable-handle"> |
| </div> |
| </div> |
| <script type="text/javascript"> |
| $(document).ready(function(){initNavTree('graph__lenet_8cpp.xhtml','');}); |
| </script> |
| <div id="doc-content"> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| <a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(0)"><span class="SelectionMark"> </span>All</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(1)"><span class="SelectionMark"> </span>Data Structures</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(2)"><span class="SelectionMark"> </span>Namespaces</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(3)"><span class="SelectionMark"> </span>Files</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(4)"><span class="SelectionMark"> </span>Functions</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(5)"><span class="SelectionMark"> </span>Variables</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(6)"><span class="SelectionMark"> </span>Typedefs</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(7)"><span class="SelectionMark"> </span>Enumerations</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(8)"><span class="SelectionMark"> </span>Enumerator</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(9)"><span class="SelectionMark"> </span>Friends</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(10)"><span class="SelectionMark"> </span>Macros</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(11)"><span class="SelectionMark"> </span>Pages</a></div> |
| |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe src="javascript:void(0)" frameborder="0" |
| name="MSearchResults" id="MSearchResults"> |
| </iframe> |
| </div> |
| |
| <div class="header"> |
| <div class="summary"> |
| <a href="#func-members">Functions</a> </div> |
| <div class="headertitle"> |
| <div class="title">graph_lenet.cpp File Reference</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| <div class="textblock"><code>#include "<a class="el" href="_logger_8h_source.xhtml">arm_compute/core/Logger.h</a>"</code><br/> |
| <code>#include "<a class="el" href="_graph_8h_source.xhtml">arm_compute/graph/Graph.h</a>"</code><br/> |
| <code>#include "<a class="el" href="_nodes_8h_source.xhtml">arm_compute/graph/Nodes.h</a>"</code><br/> |
| <code>#include "<a class="el" href="_c_l_scheduler_8h_source.xhtml">arm_compute/runtime/CL/CLScheduler.h</a>"</code><br/> |
| <code>#include "<a class="el" href="_scheduler_8h_source.xhtml">arm_compute/runtime/Scheduler.h</a>"</code><br/> |
| <code>#include "<a class="el" href="_toolchain_support_8h_source.xhtml">support/ToolchainSupport.h</a>"</code><br/> |
| <code>#include "<a class="el" href="_graph_utils_8h_source.xhtml">utils/GraphUtils.h</a>"</code><br/> |
| <code>#include "<a class="el" href="utils_2_utils_8h_source.xhtml">utils/Utils.h</a>"</code><br/> |
| <code>#include <cstdlib></code><br/> |
| <code>#include <iostream></code><br/> |
| <code>#include <memory></code><br/> |
| </div> |
| <p><a href="graph__lenet_8cpp_source.xhtml">Go to the source code of this file.</a></p> |
| <table class="memberdecls"> |
| <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a> |
| Functions</h2></td></tr> |
| <tr class="memitem:acbea98d13e0adbf27ecc036feeb610f0"><td class="memItemLeft" align="right" valign="top">std::unique_ptr< <a class="el" href="classarm__compute_1_1graph_1_1_i_tensor_accessor.xhtml">ITensorAccessor</a> > </td><td class="memItemRight" valign="bottom"><a class="el" href="graph__lenet_8cpp.xhtml#acbea98d13e0adbf27ecc036feeb610f0">get_accessor</a> (const std::string &path, const std::string &data_file)</td></tr> |
| <tr class="memdesc:acbea98d13e0adbf27ecc036feeb610f0"><td class="mdescLeft"> </td><td class="mdescRight">Generates appropriate accessor according to the specified path. <a href="#acbea98d13e0adbf27ecc036feeb610f0">More...</a><br/></td></tr> |
| <tr class="separator:acbea98d13e0adbf27ecc036feeb610f0"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a8b6f84d005166799e5a371a3d3e072b3"><td class="memItemLeft" align="right" valign="top">void </td><td class="memItemRight" valign="bottom"><a class="el" href="graph__lenet_8cpp.xhtml#a8b6f84d005166799e5a371a3d3e072b3">main_graph_lenet</a> (int argc, const char **argv)</td></tr> |
| <tr class="memdesc:a8b6f84d005166799e5a371a3d3e072b3"><td class="mdescLeft"> </td><td class="mdescRight">Example demonstrating how to implement LeNet's network using the Compute Library's graph API. <a href="#a8b6f84d005166799e5a371a3d3e072b3">More...</a><br/></td></tr> |
| <tr class="separator:a8b6f84d005166799e5a371a3d3e072b3"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a217dbf8b442f20279ea00b898af96f52"><td class="memItemLeft" align="right" valign="top">int </td><td class="memItemRight" valign="bottom"><a class="el" href="graph__lenet_8cpp.xhtml#a217dbf8b442f20279ea00b898af96f52">main</a> (int argc, const char **argv)</td></tr> |
| <tr class="memdesc:a217dbf8b442f20279ea00b898af96f52"><td class="mdescLeft"> </td><td class="mdescRight">Main program for LeNet. <a href="#a217dbf8b442f20279ea00b898af96f52">More...</a><br/></td></tr> |
| <tr class="separator:a217dbf8b442f20279ea00b898af96f52"><td class="memSeparator" colspan="2"> </td></tr> |
| </table> |
| <h2 class="groupheader">Function Documentation</h2> |
| <a class="anchor" id="acbea98d13e0adbf27ecc036feeb610f0"></a> |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">std::unique_ptr<<a class="el" href="classarm__compute_1_1graph_1_1_i_tensor_accessor.xhtml">ITensorAccessor</a>> get_accessor </td> |
| <td>(</td> |
| <td class="paramtype">const std::string & </td> |
| <td class="paramname"><em>path</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::string & </td> |
| <td class="paramname"><em>data_file</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Generates appropriate accessor according to the specified path. </p> |
| <dl class="section note"><dt>Note</dt><dd>If path is empty will generate a DummyAccessor else will generate a NumPyBinLoader</dd></dl> |
| <dl class="params"><dt>Parameters</dt><dd> |
| <table class="params"> |
| <tr><td class="paramname">path</td><td>Path to the data files </td></tr> |
| <tr><td class="paramname">data_file</td><td>Relative path to the data files from path</td></tr> |
| </table> |
| </dd> |
| </dl> |
| <dl class="section return"><dt>Returns</dt><dd>An appropriate tensor accessor </dd></dl> |
| |
| <p>Definition at line <a class="el" href="graph__lenet_8cpp_source.xhtml#l00053">53</a> of file <a class="el" href="graph__lenet_8cpp_source.xhtml">graph_lenet.cpp</a>.</p> |
| |
| <p>Referenced by <a class="el" href="graph__lenet_8cpp_source.xhtml#l00070">main_graph_lenet()</a>.</p> |
| <div class="fragment"><div class="line"><a name="l00054"></a><span class="lineno"> 54</span> {</div> |
| <div class="line"><a name="l00055"></a><span class="lineno"> 55</span>  <span class="keywordflow">if</span>(path.empty())</div> |
| <div class="line"><a name="l00056"></a><span class="lineno"> 56</span>  {</div> |
| <div class="line"><a name="l00057"></a><span class="lineno"> 57</span>  <span class="keywordflow">return</span> arm_compute::support::cpp14::make_unique<DummyAccessor>();</div> |
| <div class="line"><a name="l00058"></a><span class="lineno"> 58</span>  }</div> |
| <div class="line"><a name="l00059"></a><span class="lineno"> 59</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00060"></a><span class="lineno"> 60</span>  {</div> |
| <div class="line"><a name="l00061"></a><span class="lineno"> 61</span>  <span class="keywordflow">return</span> arm_compute::support::cpp14::make_unique<NumPyBinLoader>(path + data_file);</div> |
| <div class="line"><a name="l00062"></a><span class="lineno"> 62</span>  }</div> |
| <div class="line"><a name="l00063"></a><span class="lineno"> 63</span> }</div> |
| </div><!-- fragment --> |
| </div> |
| </div> |
| <a class="anchor" id="a217dbf8b442f20279ea00b898af96f52"></a> |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">int main </td> |
| <td>(</td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>argc</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const char ** </td> |
| <td class="paramname"><em>argv</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Main program for LeNet. </p> |
| <dl class="params"><dt>Parameters</dt><dd> |
| <table class="params"> |
| <tr><td class="paramdir">[in]</td><td class="paramname">argc</td><td>Number of arguments </td></tr> |
| <tr><td class="paramdir">[in]</td><td class="paramname">argv</td><td>Arguments ( [optional] Path to the weights folder, [optional] batches ) </td></tr> |
| </table> |
| </dd> |
| </dl> |
| |
| <p>Definition at line <a class="el" href="graph__lenet_8cpp_source.xhtml#l00142">142</a> of file <a class="el" href="graph__lenet_8cpp_source.xhtml">graph_lenet.cpp</a>.</p> |
| |
| <p>References <a class="el" href="graph__lenet_8cpp_source.xhtml#l00070">main_graph_lenet()</a>, and <a class="el" href="utils_2_utils_8cpp_source.xhtml#l00069">arm_compute::utils::run_example()</a>.</p> |
| <div class="fragment"><div class="line"><a name="l00143"></a><span class="lineno"> 143</span> {</div> |
| <div class="line"><a name="l00144"></a><span class="lineno"> 144</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearm__compute_1_1utils.xhtml#a4c9395db2c8b8d0c336656a7b58fca3e">arm_compute::utils::run_example</a>(argc, argv, <a class="code" href="graph__lenet_8cpp.xhtml#a8b6f84d005166799e5a371a3d3e072b3">main_graph_lenet</a>);</div> |
| <div class="line"><a name="l00145"></a><span class="lineno"> 145</span> }</div> |
| <div class="ttc" id="namespacearm__compute_1_1utils_xhtml_a4c9395db2c8b8d0c336656a7b58fca3e"><div class="ttname"><a href="namespacearm__compute_1_1utils.xhtml#a4c9395db2c8b8d0c336656a7b58fca3e">arm_compute::utils::run_example</a></div><div class="ttdeci">int run_example(int argc, const char **argv, example &func)</div><div class="ttdoc">Run an example and handle the potential exceptions it throws. </div><div class="ttdef"><b>Definition:</b> <a href="utils_2_utils_8cpp_source.xhtml#l00069">Utils.cpp:69</a></div></div> |
| <div class="ttc" id="graph__lenet_8cpp_xhtml_a8b6f84d005166799e5a371a3d3e072b3"><div class="ttname"><a href="graph__lenet_8cpp.xhtml#a8b6f84d005166799e5a371a3d3e072b3">main_graph_lenet</a></div><div class="ttdeci">void main_graph_lenet(int argc, const char **argv)</div><div class="ttdoc">Example demonstrating how to implement LeNet's network using the Compute Library's graph API...</div><div class="ttdef"><b>Definition:</b> <a href="graph__lenet_8cpp_source.xhtml#l00070">graph_lenet.cpp:70</a></div></div> |
| </div><!-- fragment --> |
| </div> |
| </div> |
| <a class="anchor" id="a8b6f84d005166799e5a371a3d3e072b3"></a> |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">void main_graph_lenet </td> |
| <td>(</td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>argc</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const char ** </td> |
| <td class="paramname"><em>argv</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Example demonstrating how to implement LeNet's network using the Compute Library's graph API. </p> |
| <dl class="params"><dt>Parameters</dt><dd> |
| <table class="params"> |
| <tr><td class="paramdir">[in]</td><td class="paramname">argc</td><td>Number of arguments </td></tr> |
| <tr><td class="paramdir">[in]</td><td class="paramname">argv</td><td>Arguments ( [optional] Path to the weights folder, [optional] batches ) </td></tr> |
| </table> |
| </dd> |
| </dl> |
| <p>Path to the trainable data</p> |
| <p>Number of batches </p> |
| |
| <p>Definition at line <a class="el" href="graph__lenet_8cpp_source.xhtml#l00070">70</a> of file <a class="el" href="graph__lenet_8cpp_source.xhtml">graph_lenet.cpp</a>.</p> |
| |
| <p>References <a class="el" href="_c_l_scheduler_8h_source.xhtml#l00083">CLScheduler::default_init()</a>, <a class="el" href="namespacearm__compute.xhtml#ab4e88c89b3b7ea1735996cc4def22d58a44ad4ef5a76e6aa6fb3e3fa079a54fda">arm_compute::F32</a>, <a class="el" href="classarm__compute_1_1_logger.xhtml#a6b2499aec6ae645d88670be75dcef769">Logger::get()</a>, <a class="el" href="classarm__compute_1_1_c_l_scheduler.xhtml#a60f9a6836b628a7171914c4afe43b4a7">CLScheduler::get()</a>, <a class="el" href="graph__lenet_8cpp_source.xhtml#l00053">get_accessor()</a>, <a class="el" href="namespacearm__compute.xhtml#afb2e0528558bbb0131d2cb41a66c13d7a551b723eafd6a31d444fcb2f5920fbd3">arm_compute::INFO</a>, <a class="el" href="namespacearm__compute.xhtml#adf2ced65e536375a1c96425d9fced858a26a4b44a837bf97b972628509912b4a5">arm_compute::MAX</a>, <a class="el" href="namespacearm__compute_1_1graph.xhtml#a8d5e69e9a697c2007e241eb413c9833bacaf162e9233294cadf62d2a71a14ca09">arm_compute::graph::NEON</a>, <a class="el" href="namespacearm__compute_1_1graph.xhtml#a8d5e69e9a697c2007e241eb413c9833ba542f952490e2db695a1d544338a70cda">arm_compute::graph::OPENCL</a>, <a class="el" href="namespacearm__compute.xhtml#aa4f4d7a58287017588fc338965873f14">arm_compute::opencl_is_available()</a>, <a class="el" href="classarm__compute_1_1_activation_layer_info.xhtml#a56297e0f7b215eea46c818cb7528d9eaad346bb4679d29be241279f15d7795c1c">ActivationLayerInfo::RELU</a>, <a class="el" href="classarm__compute_1_1graph_1_1_graph.xhtml#a13a43e6d814de94978c515cb084873b1">Graph::run()</a>, and <a class="el" href="classarm__compute_1_1_logger.xhtml#a8ce097d129855b3ce7f3fcd6c30551ba">Logger::set_logger()</a>.</p> |
| |
| <p>Referenced by <a class="el" href="graph__lenet_8cpp_source.xhtml#l00142">main()</a>.</p> |
| <div class="fragment"><div class="line"><a name="l00071"></a><span class="lineno"> 71</span> {</div> |
| <div class="line"><a name="l00072"></a><span class="lineno"> 72</span>  std::string data_path; </div> |
| <div class="line"><a name="l00073"></a><span class="lineno"> 73</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> batches = 4; </div> |
| <div class="line"><a name="l00075"></a><span class="lineno"> 75</span>  <span class="comment">// Parse arguments</span></div> |
| <div class="line"><a name="l00076"></a><span class="lineno"> 76</span>  <span class="keywordflow">if</span>(argc < 2)</div> |
| <div class="line"><a name="l00077"></a><span class="lineno"> 77</span>  {</div> |
| <div class="line"><a name="l00078"></a><span class="lineno"> 78</span>  <span class="comment">// Print help</span></div> |
| <div class="line"><a name="l00079"></a><span class="lineno"> 79</span>  std::cout << <span class="stringliteral">"Usage: "</span> << argv[0] << <span class="stringliteral">" [path_to_data] [batches]\n\n"</span>;</div> |
| <div class="line"><a name="l00080"></a><span class="lineno"> 80</span>  std::cout << <span class="stringliteral">"No data folder provided: using random values\n\n"</span>;</div> |
| <div class="line"><a name="l00081"></a><span class="lineno"> 81</span>  }</div> |
| <div class="line"><a name="l00082"></a><span class="lineno"> 82</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span>(argc == 2)</div> |
| <div class="line"><a name="l00083"></a><span class="lineno"> 83</span>  {</div> |
| <div class="line"><a name="l00084"></a><span class="lineno"> 84</span>  <span class="comment">//Do something with argv[1]</span></div> |
| <div class="line"><a name="l00085"></a><span class="lineno"> 85</span>  data_path = argv[1];</div> |
| <div class="line"><a name="l00086"></a><span class="lineno"> 86</span>  std::cout << <span class="stringliteral">"Usage: "</span> << argv[0] << <span class="stringliteral">" [path_to_data] [batches]\n\n"</span>;</div> |
| <div class="line"><a name="l00087"></a><span class="lineno"> 87</span>  std::cout << <span class="stringliteral">"No number of batches where specified, thus will use the default : "</span> << batches << <span class="stringliteral">"\n\n"</span>;</div> |
| <div class="line"><a name="l00088"></a><span class="lineno"> 88</span>  }</div> |
| <div class="line"><a name="l00089"></a><span class="lineno"> 89</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00090"></a><span class="lineno"> 90</span>  {</div> |
| <div class="line"><a name="l00091"></a><span class="lineno"> 91</span>  <span class="comment">//Do something with argv[1] and argv[2]</span></div> |
| <div class="line"><a name="l00092"></a><span class="lineno"> 92</span>  data_path = argv[1];</div> |
| <div class="line"><a name="l00093"></a><span class="lineno"> 93</span>  batches = std::strtol(argv[2], <span class="keyword">nullptr</span>, 0);</div> |
| <div class="line"><a name="l00094"></a><span class="lineno"> 94</span>  }</div> |
| <div class="line"><a name="l00095"></a><span class="lineno"> 95</span> </div> |
| <div class="line"><a name="l00096"></a><span class="lineno"> 96</span>  <span class="comment">// Check if OpenCL is available and initialize the scheduler</span></div> |
| <div class="line"><a name="l00097"></a><span class="lineno"> 97</span>  <a class="code" href="namespacearm__compute_1_1graph.xhtml#a8d5e69e9a697c2007e241eb413c9833b">TargetHint</a> hint = TargetHint::NEON;</div> |
| <div class="line"><a name="l00098"></a><span class="lineno"> 98</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearm__compute.xhtml#aa4f4d7a58287017588fc338965873f14">arm_compute::opencl_is_available</a>())</div> |
| <div class="line"><a name="l00099"></a><span class="lineno"> 99</span>  {</div> |
| <div class="line"><a name="l00100"></a><span class="lineno"> 100</span>  <a class="code" href="classarm__compute_1_1_c_l_scheduler.xhtml#a60f9a6836b628a7171914c4afe43b4a7">arm_compute::CLScheduler::get</a>().<a class="code" href="classarm__compute_1_1_c_l_scheduler.xhtml#a46ecf9ef0fe80ba2ed35acfc29856b7d">default_init</a>();</div> |
| <div class="line"><a name="l00101"></a><span class="lineno"> 101</span>  hint = TargetHint::OPENCL;</div> |
| <div class="line"><a name="l00102"></a><span class="lineno"> 102</span>  }</div> |
| <div class="line"><a name="l00103"></a><span class="lineno"> 103</span> </div> |
| <div class="line"><a name="l00104"></a><span class="lineno"> 104</span>  <a class="code" href="classarm__compute_1_1graph_1_1_graph.xhtml">Graph</a> graph;</div> |
| <div class="line"><a name="l00105"></a><span class="lineno"> 105</span>  <a class="code" href="classarm__compute_1_1_logger.xhtml#a6b2499aec6ae645d88670be75dcef769">arm_compute::Logger::get</a>().<a class="code" href="classarm__compute_1_1_logger.xhtml#a8ce097d129855b3ce7f3fcd6c30551ba">set_logger</a>(std::cout, <a class="code" href="namespacearm__compute.xhtml#afb2e0528558bbb0131d2cb41a66c13d7a551b723eafd6a31d444fcb2f5920fbd3">arm_compute::LoggerVerbosity::INFO</a>);</div> |
| <div class="line"><a name="l00106"></a><span class="lineno"> 106</span> </div> |
| <div class="line"><a name="l00107"></a><span class="lineno"> 107</span>  <span class="comment">//conv1 << pool1 << conv2 << pool2 << fc1 << act1 << fc2 << smx</span></div> |
| <div class="line"><a name="l00108"></a><span class="lineno"> 108</span>  graph << hint</div> |
| <div class="line"><a name="l00109"></a><span class="lineno"> 109</span>  << <a class="code" href="classarm__compute_1_1graph_1_1_tensor.xhtml">Tensor</a>(<a class="code" href="classarm__compute_1_1_tensor_info.xhtml">TensorInfo</a>(<a class="code" href="classarm__compute_1_1_tensor_shape.xhtml">TensorShape</a>(28U, 28U, 1U, batches), 1, DataType::F32), <a class="code" href="classarm__compute_1_1graph__utils_1_1_dummy_accessor.xhtml">DummyAccessor</a>())</div> |
| <div class="line"><a name="l00110"></a><span class="lineno"> 110</span>  << <a class="code" href="classarm__compute_1_1graph_1_1_convolution_layer.xhtml">ConvolutionLayer</a>(</div> |
| <div class="line"><a name="l00111"></a><span class="lineno"> 111</span>  5U, 5U, 20U,</div> |
| <div class="line"><a name="l00112"></a><span class="lineno"> 112</span>  <a class="code" href="graph__lenet_8cpp.xhtml#acbea98d13e0adbf27ecc036feeb610f0">get_accessor</a>(data_path, <span class="stringliteral">"/cnn_data/lenet_model/conv1_w.npy"</span>),</div> |
| <div class="line"><a name="l00113"></a><span class="lineno"> 113</span>  <a class="code" href="graph__lenet_8cpp.xhtml#acbea98d13e0adbf27ecc036feeb610f0">get_accessor</a>(data_path, <span class="stringliteral">"/cnn_data/lenet_model/conv1_b.npy"</span>),</div> |
| <div class="line"><a name="l00114"></a><span class="lineno"> 114</span>  <a class="code" href="classarm__compute_1_1_pad_stride_info.xhtml">PadStrideInfo</a>(1, 1, 0, 0))</div> |
| <div class="line"><a name="l00115"></a><span class="lineno"> 115</span>  << <a class="code" href="classarm__compute_1_1graph_1_1_pooling_layer.xhtml">PoolingLayer</a>(<a class="code" href="classarm__compute_1_1_pooling_layer_info.xhtml">PoolingLayerInfo</a>(PoolingType::MAX, 2, <a class="code" href="classarm__compute_1_1_pad_stride_info.xhtml">PadStrideInfo</a>(2, 2, 0, 0)))</div> |
| <div class="line"><a name="l00116"></a><span class="lineno"> 116</span>  << <a class="code" href="classarm__compute_1_1graph_1_1_convolution_layer.xhtml">ConvolutionLayer</a>(</div> |
| <div class="line"><a name="l00117"></a><span class="lineno"> 117</span>  5U, 5U, 50U,</div> |
| <div class="line"><a name="l00118"></a><span class="lineno"> 118</span>  <a class="code" href="graph__lenet_8cpp.xhtml#acbea98d13e0adbf27ecc036feeb610f0">get_accessor</a>(data_path, <span class="stringliteral">"/cnn_data/lenet_model/conv2_w.npy"</span>),</div> |
| <div class="line"><a name="l00119"></a><span class="lineno"> 119</span>  <a class="code" href="graph__lenet_8cpp.xhtml#acbea98d13e0adbf27ecc036feeb610f0">get_accessor</a>(data_path, <span class="stringliteral">"/cnn_data/lenet_model/conv2_b.npy"</span>),</div> |
| <div class="line"><a name="l00120"></a><span class="lineno"> 120</span>  <a class="code" href="classarm__compute_1_1_pad_stride_info.xhtml">PadStrideInfo</a>(1, 1, 0, 0))</div> |
| <div class="line"><a name="l00121"></a><span class="lineno"> 121</span>  << <a class="code" href="classarm__compute_1_1graph_1_1_pooling_layer.xhtml">PoolingLayer</a>(<a class="code" href="classarm__compute_1_1_pooling_layer_info.xhtml">PoolingLayerInfo</a>(PoolingType::MAX, 2, <a class="code" href="classarm__compute_1_1_pad_stride_info.xhtml">PadStrideInfo</a>(2, 2, 0, 0)))</div> |
| <div class="line"><a name="l00122"></a><span class="lineno"> 122</span>  << <a class="code" href="classarm__compute_1_1graph_1_1_fully_connected_layer.xhtml">FullyConnectedLayer</a>(</div> |
| <div class="line"><a name="l00123"></a><span class="lineno"> 123</span>  500U,</div> |
| <div class="line"><a name="l00124"></a><span class="lineno"> 124</span>  <a class="code" href="graph__lenet_8cpp.xhtml#acbea98d13e0adbf27ecc036feeb610f0">get_accessor</a>(data_path, <span class="stringliteral">"/cnn_data/lenet_model/ip1_w.npy"</span>),</div> |
| <div class="line"><a name="l00125"></a><span class="lineno"> 125</span>  <a class="code" href="graph__lenet_8cpp.xhtml#acbea98d13e0adbf27ecc036feeb610f0">get_accessor</a>(data_path, <span class="stringliteral">"/cnn_data/lenet_model/ip1_b.npy"</span>))</div> |
| <div class="line"><a name="l00126"></a><span class="lineno"> 126</span>  << <a class="code" href="classarm__compute_1_1graph_1_1_activation_layer.xhtml">ActivationLayer</a>(<a class="code" href="classarm__compute_1_1_activation_layer_info.xhtml">ActivationLayerInfo</a>(ActivationLayerInfo::ActivationFunction::RELU))</div> |
| <div class="line"><a name="l00127"></a><span class="lineno"> 127</span>  << <a class="code" href="classarm__compute_1_1graph_1_1_fully_connected_layer.xhtml">FullyConnectedLayer</a>(</div> |
| <div class="line"><a name="l00128"></a><span class="lineno"> 128</span>  10U,</div> |
| <div class="line"><a name="l00129"></a><span class="lineno"> 129</span>  <a class="code" href="graph__lenet_8cpp.xhtml#acbea98d13e0adbf27ecc036feeb610f0">get_accessor</a>(data_path, <span class="stringliteral">"/cnn_data/lenet_model/ip2_w.npy"</span>),</div> |
| <div class="line"><a name="l00130"></a><span class="lineno"> 130</span>  <a class="code" href="graph__lenet_8cpp.xhtml#acbea98d13e0adbf27ecc036feeb610f0">get_accessor</a>(data_path, <span class="stringliteral">"/cnn_data/lenet_model/ip2_b.npy"</span>))</div> |
| <div class="line"><a name="l00131"></a><span class="lineno"> 131</span>  << <a class="code" href="classarm__compute_1_1graph_1_1_softmax_layer.xhtml">SoftmaxLayer</a>()</div> |
| <div class="line"><a name="l00132"></a><span class="lineno"> 132</span>  << <a class="code" href="classarm__compute_1_1graph_1_1_tensor.xhtml">Tensor</a>(<a class="code" href="classarm__compute_1_1graph__utils_1_1_dummy_accessor.xhtml">DummyAccessor</a>());</div> |
| <div class="line"><a name="l00133"></a><span class="lineno"> 133</span> </div> |
| <div class="line"><a name="l00134"></a><span class="lineno"> 134</span>  graph.<a class="code" href="classarm__compute_1_1graph_1_1_graph.xhtml#a13a43e6d814de94978c515cb084873b1">run</a>();</div> |
| <div class="line"><a name="l00135"></a><span class="lineno"> 135</span> }</div> |
| <div class="ttc" id="classarm__compute_1_1_tensor_shape_xhtml"><div class="ttname"><a href="classarm__compute_1_1_tensor_shape.xhtml">arm_compute::TensorShape</a></div><div class="ttdoc">Shape of a tensor. </div><div class="ttdef"><b>Definition:</b> <a href="_tensor_shape_8h_source.xhtml#l00038">TensorShape.h:38</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1graph_1_1_fully_connected_layer_xhtml"><div class="ttname"><a href="classarm__compute_1_1graph_1_1_fully_connected_layer.xhtml">arm_compute::graph::FullyConnectedLayer</a></div><div class="ttdoc">Fully connected layer node. </div><div class="ttdef"><b>Definition:</b> <a href="arm__compute_2graph_2nodes_2_fully_connected_layer_8h_source.xhtml#l00037">FullyConnectedLayer.h:37</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1graph_1_1_graph_xhtml_a13a43e6d814de94978c515cb084873b1"><div class="ttname"><a href="classarm__compute_1_1graph_1_1_graph.xhtml#a13a43e6d814de94978c515cb084873b1">arm_compute::graph::Graph::run</a></div><div class="ttdeci">void run()</div><div class="ttdoc">Executes the graph. </div></div> |
| <div class="ttc" id="classarm__compute_1_1graph__utils_1_1_dummy_accessor_xhtml"><div class="ttname"><a href="classarm__compute_1_1graph__utils_1_1_dummy_accessor.xhtml">arm_compute::graph_utils::DummyAccessor</a></div><div class="ttdoc">Dummy accessor class. </div><div class="ttdef"><b>Definition:</b> <a href="_graph_utils_8h_source.xhtml#l00060">GraphUtils.h:60</a></div></div> |
| <div class="ttc" id="graph__lenet_8cpp_xhtml_acbea98d13e0adbf27ecc036feeb610f0"><div class="ttname"><a href="graph__lenet_8cpp.xhtml#acbea98d13e0adbf27ecc036feeb610f0">get_accessor</a></div><div class="ttdeci">std::unique_ptr< ITensorAccessor > get_accessor(const std::string &path, const std::string &data_file)</div><div class="ttdoc">Generates appropriate accessor according to the specified path. </div><div class="ttdef"><b>Definition:</b> <a href="graph__lenet_8cpp_source.xhtml#l00053">graph_lenet.cpp:53</a></div></div> |
| <div class="ttc" id="namespacearm__compute_xhtml_aa4f4d7a58287017588fc338965873f14"><div class="ttname"><a href="namespacearm__compute.xhtml#aa4f4d7a58287017588fc338965873f14">arm_compute::opencl_is_available</a></div><div class="ttdeci">bool opencl_is_available()</div></div> |
| <div class="ttc" id="classarm__compute_1_1_activation_layer_info_xhtml"><div class="ttname"><a href="classarm__compute_1_1_activation_layer_info.xhtml">arm_compute::ActivationLayerInfo</a></div><div class="ttdoc">Activation Layer Information class. </div><div class="ttdef"><b>Definition:</b> <a href="arm__compute_2core_2_types_8h_source.xhtml#l00511">Types.h:511</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1_c_l_scheduler_xhtml_a60f9a6836b628a7171914c4afe43b4a7"><div class="ttname"><a href="classarm__compute_1_1_c_l_scheduler.xhtml#a60f9a6836b628a7171914c4afe43b4a7">arm_compute::CLScheduler::get</a></div><div class="ttdeci">static CLScheduler & get()</div><div class="ttdoc">Access the scheduler singleton. </div></div> |
| <div class="ttc" id="classarm__compute_1_1_pad_stride_info_xhtml"><div class="ttname"><a href="classarm__compute_1_1_pad_stride_info.xhtml">arm_compute::PadStrideInfo</a></div><div class="ttdoc">Padding and stride information class. </div><div class="ttdef"><b>Definition:</b> <a href="arm__compute_2core_2_types_8h_source.xhtml#l00406">Types.h:406</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1graph_1_1_softmax_layer_xhtml"><div class="ttname"><a href="classarm__compute_1_1graph_1_1_softmax_layer.xhtml">arm_compute::graph::SoftmaxLayer</a></div><div class="ttdoc">Softmax layer node. </div><div class="ttdef"><b>Definition:</b> <a href="arm__compute_2graph_2nodes_2_softmax_layer_8h_source.xhtml#l00036">SoftmaxLayer.h:36</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1graph_1_1_graph_xhtml"><div class="ttname"><a href="classarm__compute_1_1graph_1_1_graph.xhtml">arm_compute::graph::Graph</a></div><div class="ttdoc">Graph class. </div><div class="ttdef"><b>Definition:</b> <a href="_graph_8h_source.xhtml#l00041">Graph.h:41</a></div></div> |
| <div class="ttc" id="namespacearm__compute_xhtml_afb2e0528558bbb0131d2cb41a66c13d7a551b723eafd6a31d444fcb2f5920fbd3"><div class="ttname"><a href="namespacearm__compute.xhtml#afb2e0528558bbb0131d2cb41a66c13d7a551b723eafd6a31d444fcb2f5920fbd3">arm_compute::LoggerVerbosity::INFO</a></div><div class="ttdoc">Log info. </div></div> |
| <div class="ttc" id="classarm__compute_1_1graph_1_1_pooling_layer_xhtml"><div class="ttname"><a href="classarm__compute_1_1graph_1_1_pooling_layer.xhtml">arm_compute::graph::PoolingLayer</a></div><div class="ttdoc">Pooling layer node. </div><div class="ttdef"><b>Definition:</b> <a href="arm__compute_2graph_2nodes_2_pooling_layer_8h_source.xhtml#l00037">PoolingLayer.h:37</a></div></div> |
| <div class="ttc" id="namespacearm__compute_1_1graph_xhtml_a8d5e69e9a697c2007e241eb413c9833b"><div class="ttname"><a href="namespacearm__compute_1_1graph.xhtml#a8d5e69e9a697c2007e241eb413c9833b">arm_compute::graph::TargetHint</a></div><div class="ttdeci">TargetHint</div><div class="ttdoc">< Execution hint to the graph executor </div><div class="ttdef"><b>Definition:</b> <a href="arm__compute_2graph_2_types_8h_source.xhtml#l00050">Types.h:50</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1_tensor_info_xhtml"><div class="ttname"><a href="classarm__compute_1_1_tensor_info.xhtml">arm_compute::TensorInfo</a></div><div class="ttdoc">Store the tensor's metadata. </div><div class="ttdef"><b>Definition:</b> <a href="_tensor_info_8h_source.xhtml#l00042">TensorInfo.h:42</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1graph_1_1_activation_layer_xhtml"><div class="ttname"><a href="classarm__compute_1_1graph_1_1_activation_layer.xhtml">arm_compute::graph::ActivationLayer</a></div><div class="ttdoc">Activation Layer node. </div><div class="ttdef"><b>Definition:</b> <a href="arm__compute_2graph_2nodes_2_activation_layer_8h_source.xhtml#l00037">ActivationLayer.h:37</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1graph_1_1_convolution_layer_xhtml"><div class="ttname"><a href="classarm__compute_1_1graph_1_1_convolution_layer.xhtml">arm_compute::graph::ConvolutionLayer</a></div><div class="ttdoc">Convolution layer node. </div><div class="ttdef"><b>Definition:</b> <a href="arm__compute_2graph_2nodes_2_convolution_layer_8h_source.xhtml#l00041">ConvolutionLayer.h:41</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1_pooling_layer_info_xhtml"><div class="ttname"><a href="classarm__compute_1_1_pooling_layer_info.xhtml">arm_compute::PoolingLayerInfo</a></div><div class="ttdoc">Pooling Layer Information class. </div><div class="ttdef"><b>Definition:</b> <a href="arm__compute_2core_2_types_8h_source.xhtml#l00445">Types.h:445</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1_logger_xhtml_a6b2499aec6ae645d88670be75dcef769"><div class="ttname"><a href="classarm__compute_1_1_logger.xhtml#a6b2499aec6ae645d88670be75dcef769">arm_compute::Logger::get</a></div><div class="ttdeci">static Logger & get()</div></div> |
| <div class="ttc" id="classarm__compute_1_1_logger_xhtml_a8ce097d129855b3ce7f3fcd6c30551ba"><div class="ttname"><a href="classarm__compute_1_1_logger.xhtml#a8ce097d129855b3ce7f3fcd6c30551ba">arm_compute::Logger::set_logger</a></div><div class="ttdeci">void set_logger(std::ostream &ostream, LoggerVerbosity verbosity)</div></div> |
| <div class="ttc" id="classarm__compute_1_1_c_l_scheduler_xhtml_a46ecf9ef0fe80ba2ed35acfc29856b7d"><div class="ttname"><a href="classarm__compute_1_1_c_l_scheduler.xhtml#a46ecf9ef0fe80ba2ed35acfc29856b7d">arm_compute::CLScheduler::default_init</a></div><div class="ttdeci">void default_init(ICLTuner *cl_tuner=nullptr)</div><div class="ttdoc">Initialises the context and command queue used by the scheduler to default values and sets a default ...</div><div class="ttdef"><b>Definition:</b> <a href="_c_l_scheduler_8h_source.xhtml#l00083">CLScheduler.h:83</a></div></div> |
| <div class="ttc" id="classarm__compute_1_1graph_1_1_tensor_xhtml"><div class="ttname"><a href="classarm__compute_1_1graph_1_1_tensor.xhtml">arm_compute::graph::Tensor</a></div><div class="ttdoc">Tensor class. </div><div class="ttdef"><b>Definition:</b> <a href="graph_2_tensor_8h_source.xhtml#l00038">Tensor.h:38</a></div></div> |
| </div><!-- fragment --> |
| </div> |
| </div> |
| </div><!-- contents --> |
| </div><!-- doc-content --> |
| <!-- start footer part --> |
| <div id="nav-path" class="navpath"><!-- id is needed for treeview function! --> |
| <ul> |
| <li class="navelem"><a class="el" href="dir_1253bad92dedae5edd993ead924afb7b.xhtml">examples</a></li><li class="navelem"><a class="el" href="graph__lenet_8cpp.xhtml">graph_lenet.cpp</a></li> |
| <li class="footer">Generated on Thu Oct 12 2017 14:26:35 for Compute Library by |
| <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.6 </li> |
| </ul> |
| </div> |
| </body> |
| </html> |