simplify_reshapes.hpp 463 Bytes
Newer Older
Paul's avatar
Paul committed
1
2
3
4
5
#ifndef MIGRAPH_GUARD_RTGLIB_SIMPLIFY_RESHAPES_HPP
#define MIGRAPH_GUARD_RTGLIB_SIMPLIFY_RESHAPES_HPP

#include <string>
#include <migraph/instruction_ref.hpp>
6
#include <migraph/config.hpp>
Paul's avatar
Paul committed
7

8
namespace migraph { inline namespace MIGRAPH_INLINE_NS {
Paul's avatar
Paul committed
9
10
11
12
13
14
15
16
17

struct program;

struct simplify_reshapes
{
    std::string name() const { return "simplify_reshapes"; }
    void apply(program& p) const;
};

18
} // inline namespace MIGRAPH_INLINE_NS
Paul's avatar
Paul committed
19
20
21
} // namespace migraph

#endif