| // |
| // Copyright © 2017 Arm Ltd. All rights reserved. |
| // SPDX-License-Identifier: MIT |
| // |
| %include "stl.i" |
| %include "cstring.i" |
| %include "std_string.i" |
| %include "std_vector.i" |
| %include "std_unordered_set.i" |
| %include "std_pair.i" |
| %include "stdint.i" |
| %include "carrays.i" |
| %include "exception.i" |
| %include "typemaps.i" |
| %include "std_iostream.i" |
| |
| %ignore *::operator=; |
| %ignore *::operator[]; |
| |
| |
| // Define exception typemap to wrap armnn exception into python exception. |
| |
| %exception{ |
| try { |
| $action |
| } catch (armnn::Exception &e) { |
| SWIG_exception(SWIG_RuntimeError, const_cast<char*>(e.what())); |
| } |
| }; |
| |
| %exception __getitem__ { |
| try { |
| $action |
| } catch (armnn::InvalidArgumentException &e) { |
| SWIG_exception(SWIG_ValueError, const_cast<char*>(e.what())); |
| } catch (const std::out_of_range &e) { |
| SWIG_exception(SWIG_IndexError, const_cast<char*>(e.what())); |
| } catch (const std::exception &e) { |
| SWIG_exception(SWIG_RuntimeError, const_cast<char*>(e.what())); |
| } |
| }; |
| |
| %exception __setitem__ { |
| try { |
| $action |
| } catch (armnn::InvalidArgumentException &e) { |
| SWIG_exception(SWIG_ValueError, const_cast<char*>(e.what())); |
| } catch (const std::out_of_range &e) { |
| SWIG_exception(SWIG_IndexError, const_cast<char*>(e.what())); |
| } catch (const std::exception &e) { |
| SWIG_exception(SWIG_RuntimeError, const_cast<char*>(e.what())); |
| } |
| }; |