models.hpp 328 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14

#include <migraphx/program.hpp>

namespace migraphx {
namespace driver {
inline namespace MIGRAPHX_INLINE_NS {

migraphx::program resnet50(unsigned batch);
migraphx::program inceptionv3(unsigned batch);
migraphx::program alexnet(unsigned batch);

} // namespace MIGRAPHX_INLINE_NS
} // namespace driver
} // namespace migraphx