Caffe2 - C++ API
A deep learning, cross platform ML framework
conv_transpose_op_mobile.cc
1 
17 #include "caffe2/operators/conv_transpose_op_mobile.h"
18 #include "caffe2/operators/conv_transpose_op_mobile_impl.h"
19 
20 namespace caffe2 {
21 
22 #ifndef CAFFE2_MOBILE
23 #error "mobile build state not defined"
24 #endif
25 
26 #if CAFFE2_MOBILE
27 // mobile-only implementation (tiled + vectorized + multithreaded)
28 REGISTER_CPU_OPERATOR_WITH_ENGINE(
29  ConvTranspose,
30  MOBILE,
31  ConvTransposeMobileOp<float, CPUContext>);
32 #endif
33 
34 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.