Learn more »
Push, build, and install
RubyGems
npm packages
Python packages
Maven artifacts
PHP packages
Go Modules
Bower components
Debian packages
RPM packages
NuGet packages
#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