Skip to content

Latest commit

 

History

History
9 lines (6 loc) · 549 Bytes

cpp_custom_ops_tutorial.md

File metadata and controls

9 lines (6 loc) · 549 Bytes

Making Custom Ops with C++

This is just an outline. Full instructions will be added later.

  1. Clone and build the repo by following the Getting Stated instructions here.
  2. Create a kernel for your custom op in the /ocos/kernels directory. Create a .cc and .hpp file.
  3. Write tests in the /test directory.

To use the op in a tensorflow model, follow the TF2ONNX Custom Ops Tutorial

close