caffe2_cpp_tutorial icon indicating copy to clipboard operation
caffe2_cpp_tutorial copied to clipboard
verified publisher icon leovandriel

Creating a Custom Operator

Open deeptibhegde opened this issue 7 years ago • 0 comments

I would like to create a custom multi - dimensional convolution operator using gemm functions in the math_cpu.cpp source code. Would it be enough to just create an op.cc and header file and include in the operators directory? I am aware of the official documentation on custom Caffe2 operators, but am hoping for some more information. Will I need to build it separately?

deeptibhegde avatar Oct 04 '18 17:10 deeptibhegde