eliminate_concat.hpp 422 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
#ifndef MIGRAPH_GUARD_RTGLIB_ELIMINATE_CONCAT_HPP
#define MIGRAPH_GUARD_RTGLIB_ELIMINATE_CONCAT_HPP

#include <string>
#include <migraph/instruction_ref.hpp>
#include <migraph/concat_opt.hpp>

namespace migraph {

struct program;

struct eliminate_concat
{
Scott Thornton's avatar
Scott Thornton committed
14
    concat_optimization concat_opt;
15
16
17
18
19
20
21
    std::string name() const { return "eliminate_concat"; }
    void apply(program& p) const;
};

} // namespace migraph

#endif