1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
#pragma once
#include <ATen/Config.h>
#include <torch/csrc/jit/api/module.h>
#include <torch/csrc/jit/ir/ir.h>
#include <torch/csrc/jit/passes/subgraph_rewrite.h>
#if AT_MKLDNN_ENABLED()
#include <ideep/tensor.hpp>
#endif // AT_MKLDNN_ENABLED()
namespace torch {
namespace jit {
#if AT_MKLDNN_ENABLED()
namespace mkldnn {
const static std::map<std::string, std::vector<torch::jit::MatchFilter>>
fusion_rewrite_map = {
{"none", {}},
{"relu", {}},
};
} // namespace mkldnn
#endif // AT_MKLDNN_ENABLED()
void FuseConvWithEltwise(std::shared_ptr<Graph>& graph);
} // namespace jit
} // namespace torch
|
