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
|
#pragma once
#include <string>
#include <unordered_map>
#include <vector>
#include <caffe2/core/net.h>
#include <caffe2/core/workspace.h>
#include <caffe2/proto/caffe2_pb.h>
namespace caffe2 {
namespace opt {
// Mapping from fp32 ops to fakefp16 ops
TORCH_API std::unordered_map<std::string, std::string> getFakeFp16OpMapping(
bool use_fp16_acc = false,
bool use_nnpi = false);
TORCH_API void fakeFp16FuseOps(NetDef* net);
// Transform normal fp32 operators to fakefp16 operators.
TORCH_API void fakeFp16Transform(NetDef* net);
} // namespace opt
} // namespace caffe2
|
