blob: 0733eea8d562cf3def7c8bcdfb8b43207b80c8ab [file] [log] [blame]
#include "caffe2/image/image_input_op.h"
namespace caffe2 {
REGISTER_CPU_OPERATOR(ImageInput, ImageInputOp<CPUContext>);
OPERATOR_SCHEMA(ImageInput)
.NumInputs(0, 1).NumOutputs(2);
NO_GRADIENT(ImageInput);
} // namespace caffe2