mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
91797c0672
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/10946 ``` codemod -d . --extensions cc,cpp,cu,cuh,h caffe2/proto/caffe2.pb.h caffe2/proto/caffe2_pb.h ``` Reviewed By: houseroad Differential Revision: D9539945 fbshipit-source-id: 497d04720e8e7e61c05ffe1b23733d0cb774de7e
20 lines
464 B
C++
20 lines
464 B
C++
#include "caffe2/core/operator.h"
|
|
#include "caffe2/proto/caffe2_pb.h"
|
|
|
|
CAFFE2_DEFINE_bool(
|
|
caffe2_mkl_memonger_in_use,
|
|
false,
|
|
"Turn on if memonger is used to force reallocate intermediate "
|
|
"and output buffers within each op");
|
|
|
|
namespace caffe2 {
|
|
|
|
CAFFE_DEFINE_REGISTRY(
|
|
MKLOperatorRegistry,
|
|
OperatorBase,
|
|
const OperatorDef&,
|
|
Workspace*);
|
|
CAFFE_REGISTER_DEVICE_TYPE(DeviceType::MKLDNN, MKLOperatorRegistry);
|
|
|
|
} // namespace caffe2
|