mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
d4832f1e7b
Summary: Some more `ATEN_API` additions for hidden visibility. Running CI tests to see what fails to link. cc Yangqing mingzhe09088 ezyang Pull Request resolved: https://github.com/pytorch/pytorch/pull/10624 Reviewed By: mingzhe09088 Differential Revision: D9392728 Pulled By: orionr fbshipit-source-id: e0f0861496b12c9a4e40c10b6e0c9e0df18e8726
17 lines
390 B
C++
17 lines
390 B
C++
#ifndef CAFFE2_OPT_MOBILE_H_
|
|
#define CAFFE2_OPT_MOBILE_H_
|
|
|
|
#include "caffe2/core/common.h"
|
|
#include "nomnigraph/Representations/NeuralNet.h"
|
|
|
|
namespace caffe2 {
|
|
namespace opt {
|
|
|
|
CAFFE2_API void addNNPACK(nom::repr::NNModule* nn, bool low_memory = false);
|
|
CAFFE2_API void fuseNNPACKConvRelu(nom::repr::NNModule* nn);
|
|
|
|
} // namespace opt
|
|
} // namespace caffe2
|
|
|
|
#endif // CAFFE2_OPT_MOBILE_H_
|