1#pragma once
2
3#include <torch/csrc/jit/ir/ir.h>
4
5namespace torch {
6namespace jit {
7
8// Converts operators & their parameters to mkldnn if it is profitable
9// Currently encompassing Conv2d and Conv3d, and Linear
10// Op must be in float32 and mkldnn must be built
11// This pass only works on frozen graph
12TORCH_API void ConvertFrozenOpsToMKLDNN(std::shared_ptr<Graph>& graph);
13
14} // namespace jit
15} // namespace torch
16