Skip to content

Commit

Permalink
[tflchef] Introduce extype and custom_code for custom operators
Browse files Browse the repository at this point in the history
This introduces extype and custom_code for custom operators.

ONE-DCO-1.0-Signed-off-by: SeungHui Lee <[email protected]>
  • Loading branch information
Seunghui98 committed Oct 23, 2023
1 parent a8dda5b commit 3905e6d
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions compiler/tflchef/proto/tflchef.proto
Original file line number Diff line number Diff line change
Expand Up @@ -556,6 +556,8 @@ message Operation {
repeated string input = 2;
repeated string output = 3;
optional int32 version = 4 [default = 1];
optional string extype = 5;
optional string custom_code = 6;

optional Conv2DOptions conv2d_options = 100;
optional Pool2DOptions averagepool2d_options = 101;
Expand Down

0 comments on commit 3905e6d

Please sign in to comment.