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
15 lines
443 B
C++
15 lines
443 B
C++
#include "caffe2/core/common.h"
|
|
#include "nomnigraph/Representations/NeuralNet.h"
|
|
|
|
namespace caffe2 {
|
|
namespace opt {
|
|
|
|
CAFFE2_API void insertCopies(
|
|
nom::repr::NNModule* nn,
|
|
std::function<bool(nom::repr::NNGraph::NodeRef)> supported,
|
|
std::function<nom::repr::NNGraph::NodeRef(nom::repr::NNGraph&)> copyToFn,
|
|
std::function<nom::repr::NNGraph::NodeRef(nom::repr::NNGraph&)> copyFromFn);
|
|
|
|
} // namespace opt
|
|
} // namespace caffe2
|