Sign in
android
/
platform
/
external
/
pytorch
/
2ed1077a83
/
.
/
caffe2
/
operators
/
depth_split_op.cc
blob: 9fc945a6fa5ecc317381e61b3378eb7bac75c8e9 [
file
] [
log
] [
blame
]
#include
"caffe2/operators/depth_split_op.h"
namespace
caffe2
{
namespace
{
REGISTER_CPU_OPERATOR
(
DepthSplit
,
DepthSplitOp
<
float
,
CPUContext
>)
REGISTER_CPU_OPERATOR
(
DepthConcat
,
DepthConcatOp
<
float
,
CPUContext
>)
}
// namespace
}
// namespace caffe2