(h)
| 269 | |
| 270 | @auto_handle() |
| 271 | def program(h): |
| 272 | h.constructor('create') |
| 273 | h.method('get_main_module', returns='migraphx::module*') |
| 274 | h.method('create_module', |
| 275 | api.params(name='const char*'), |
| 276 | returns='migraphx::module*') |
| 277 | h.method( |
| 278 | 'compile', |
| 279 | api.params(target='migraphx::target', |
| 280 | options='migraphx::compile_options')) |
| 281 | h.method('get_parameter_shapes', |
| 282 | returns='std::unordered_map<std::string, migraphx::shape>') |
| 283 | h.method('get_output_shapes', |
| 284 | invoke='migraphx::get_output_shapes($@)', |
| 285 | returns='std::vector<migraphx::shape>') |
| 286 | h.method('print', invoke='migraphx::print_program($@)', const=True) |
| 287 | h.method('sort') |
| 288 | h.method('run', |
| 289 | api.params( |
| 290 | params='std::unordered_map<std::string, migraphx::argument>'), |
| 291 | invoke='migraphx::run($@)', |
| 292 | returns='std::vector<migraphx::argument>') |
| 293 | h.method('run_async', |
| 294 | api.params( |
| 295 | params='std::unordered_map<std::string, migraphx::argument>', |
| 296 | s='void*', |
| 297 | name='const char *'), |
| 298 | invoke='migraphx::run_async($@)', |
| 299 | returns='std::vector<migraphx::argument>') |
| 300 | h.method('equal', |
| 301 | api.params(x='const migraphx::program&'), |
| 302 | invoke='migraphx::equal($@)', |
| 303 | returns='bool', |
| 304 | const=True) |
| 305 | h.method('experimental_get_context', |
| 306 | invoke='migraphx::get_context($@)', |
| 307 | const=True, |
| 308 | returns='migraphx::context') |
| 309 | |
| 310 | |
| 311 | @auto_handle() |
no test coverage detected