549 B
549 B
Making Custom Ops with C++
This is just an outline. Full instructions will be added later.
- Clone and build the repo by following the Getting Stated instructions here.
- Create a kernel for your custom op in the /ocos/kernels directory. Create a .cc and .hpp file.
- Write tests in the /test directory.
To use the op in a tensorflow model, follow the TF2ONNX Custom Ops Tutorial