1#pragma once
2
3#include <torch/csrc/jit/ir/ir.h>
4
5namespace torch {
6namespace jit {
7
8// Concats multiple linear ops with the same Tensor input
9// into a single linear op.
10TORCH_API bool FrozenConcatLinear(std::shared_ptr<Graph>& graph);
11
12} // namespace jit
13} // namespace torch
14