pytorch/caffe2/db/create_db_op.cc

10 lines
214 B
C++

#include "caffe2/db/create_db_op.h"
namespace caffe2 {
REGISTER_CPU_OPERATOR(CreateDB, CreateDBOp<CPUContext>);
OPERATOR_SCHEMA(CreateDB).NumInputs(0).NumOutputs(1);
NO_GRADIENT(CreateDB);
} // namespace caffe2