mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
71ca600af9
Summary: Since caffe2 and torch have been consolidated, CAFFE2_API should be merged with TORCH_API. Addresses a TODO. Manually edited some references of the removed `CAFFE2_API`: * `CONTRIBUTING.md` * `caffe2/proto/CMakeLists.txt` * `cmake/ProtoBuf.cmake` * `c10/macros/Export.h` * `torch/csrc/WindowsTorchApiMacro.h` Pull Request resolved: https://github.com/pytorch/pytorch/pull/49496 Reviewed By: malfet, samestep Differential Revision: D25600726 Pulled By: janeyx99 fbshipit-source-id: 7e068d959e397ac183c097d7e9a9afeca5ddd782
26 lines
584 B
C++
26 lines
584 B
C++
#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
|