1#pragma once
2
3#include <torch/csrc/jit/ir/ir.h>
4
5namespace torch {
6namespace jit {
7
8// Transposes the weight matrix for frozen linear modules.
9// and converts it into a matmul
10TORCH_API bool FrozenLinearTranspose(std::shared_ptr<Graph>& graph);
11
12} // namespace jit
13} // namespace torch
14