[res] Introduce extype and custom_code for custom operators (#11764)

This introduces extype and custom_code for custom operators. ONE-DCO-1.0-Signed-off-by: SeungHui Lee <[email protected]>
Samsung · Oct 25, 2023 · 6fc769a · 6fc769a
1 parent b52c094
commit 6fc769a
Show file tree

Hide file tree

Showing 15 changed files with 34 additions and 0 deletions.
diff --git a/res/TensorFlowLiteRecipes/All_000/test.recipe b/res/TensorFlowLiteRecipes/All_000/test.recipe
@@ -25,6 +25,8 @@ operation {
   input: "ifm"
   input: "All/reduction_indices"
   output: "ofm"
+  custom_code: "All"
+  extype: "Custom"
 }
 input: "ifm"
 output: "ofm"
diff --git a/res/TensorFlowLiteRecipes/BatchMatMulV2_000/test.recipe b/res/TensorFlowLiteRecipes/BatchMatMulV2_000/test.recipe
@@ -18,6 +18,8 @@ operation {
   input: "ifm1"
   input: "ifm2"
   output: "ofm"
+  custom_code: "BatchMatMulV2"
+  extype: "Custom"
 }
 input: "ifm1"
 input: "ifm2"

diff --git a/res/TensorFlowLiteRecipes/BatchMatMulV2_001/test.recipe b/res/TensorFlowLiteRecipes/BatchMatMulV2_001/test.recipe
@@ -21,6 +21,8 @@ operation {
   input: "ifm1"
   input: "ifm2"
   output: "ofm"
+  custom_code: "BatchMatMulV2"
+  extype: "Custom"
 }
 input: "ifm1"
 input: "ifm2"

diff --git a/res/TensorFlowLiteRecipes/BroadcastTo_000/test.recipe b/res/TensorFlowLiteRecipes/BroadcastTo_000/test.recipe
@@ -19,6 +19,8 @@ operation {
   input: "bc_input"
   input: "bc_shape"
   output: "bc_ofm"
+  custom_code: "BroadcastTo"
+  extype: "Custom"
 }
 input: "bc_input"
 output: "bc_ofm"
diff --git a/res/TensorFlowLiteRecipes/MatMul_000/test.recipe b/res/TensorFlowLiteRecipes/MatMul_000/test.recipe
@@ -22,6 +22,8 @@ operation {
     transpose_a: true
     transpose_b: false
   }
+  custom_code: "MatMul"
+  extype: "Custom"
 }
 input: "ifm1"
 input: "ifm2"

diff --git a/res/TensorFlowLiteRecipes/MatrixBandPart_000/test.recipe b/res/TensorFlowLiteRecipes/MatrixBandPart_000/test.recipe
@@ -32,6 +32,8 @@ operation {
   input: "MatrixBandPart/num_lower"
   input: "MatrixBandPart/num_upper"
   output: "ofm"
+  custom_code: "MatrixBandPart"
+  extype: "Custom"
 }
 input: "ifm"
 output: "ofm"
diff --git a/res/TensorFlowLiteRecipes/MaxPoolWithArgmax_000/test.recipe b/res/TensorFlowLiteRecipes/MaxPoolWithArgmax_000/test.recipe
@@ -27,6 +27,8 @@ operation {
     output_type: INT64
     include_batch_in_index: false
   }
+  custom_code: "MaxPoolWithArgmax"
+  extype: "Custom"
 }
 input: "ifm"
 output: "ofm"

diff --git a/res/TensorFlowLiteRecipes/MaxPoolWithArgmax_001/test.recipe b/res/TensorFlowLiteRecipes/MaxPoolWithArgmax_001/test.recipe
@@ -27,6 +27,8 @@ operation {
     output_type: INT32
     include_batch_in_index: false
   }
+  custom_code: "MaxPoolWithArgmax"
+  extype: "Custom"
 }
 input: "ifm"
 output: "ofm"

diff --git a/res/TensorFlowLiteRecipes/MaxPoolWithArgmax_002/test.recipe b/res/TensorFlowLiteRecipes/MaxPoolWithArgmax_002/test.recipe
@@ -27,6 +27,8 @@ operation {
     output_type: INT64
     include_batch_in_index: false
   }
+  custom_code: "MaxPoolWithArgmax"
+  extype: "Custom"
 }
 input: "ifm"
 output: "ofm"

diff --git a/res/TensorFlowLiteRecipes/Net_BroadcastTo_AddV2_000/test.recipe b/res/TensorFlowLiteRecipes/Net_BroadcastTo_AddV2_000/test.recipe
@@ -19,6 +19,8 @@ operation {
   input: "bc_input"
   input: "bc_shape"
   output: "bc_ofm"
+  custom_code: "BroadcastTo"
+  extype: "Custom"
 }
 operand {
   name: "reshape_data"
@@ -57,6 +59,8 @@ operation {
   input: "bc_ofm"
   input: "reshape_ofm"
   output: "ofm"
+  custom_code: "AddV2"
+  extype: "Custom"
 }
 input: "bc_input"
 input: "reshape_data"

diff --git a/res/TensorFlowLiteRecipes/Net_BroadcastTo_AddV2_001/test.recipe b/res/TensorFlowLiteRecipes/Net_BroadcastTo_AddV2_001/test.recipe
@@ -19,6 +19,8 @@ operation {
   input: "bc_input"
   input: "bc_shape"
   output: "bc_ofm"
+  custom_code: "BroadcastTo"
+  extype: "Custom"
 }
 operand {
   name: "reshape_data"
@@ -57,6 +59,8 @@ operation {
   input: "bc_ofm"
   input: "reshape_ofm"
   output: "ofm"
+  custom_code: "AddV2"
+  extype: "Custom"
 }
 input: "bc_input"
 input: "reshape_data"

diff --git a/res/TensorFlowLiteRecipes/Net_FC_Gelu_FC_000/test.recipe b/res/TensorFlowLiteRecipes/Net_FC_Gelu_FC_000/test.recipe
@@ -104,6 +104,8 @@ operation {
   type: "Erf"
   input: "fc2"
   output: "erf"
+  custom_code: "Erf"
+  extype: "Custom"
 }
 operation {
   type: "Add"

diff --git a/res/TensorFlowLiteRecipes/Net_Gather_SparseToDense_AddV2_000/test.recipe b/res/TensorFlowLiteRecipes/Net_Gather_SparseToDense_AddV2_000/test.recipe
@@ -103,6 +103,8 @@ operation {
   input: "ofm_sparse"
   input: "add_v2_2"
   output: "ofm_add_v2"
+  custom_code: "AddV2"
+  extype: "Custom"
 }
 operation {
   type: "Cast"

diff --git a/res/TensorFlowLiteRecipes/Net_Gelu_000/test.recipe b/res/TensorFlowLiteRecipes/Net_Gelu_000/test.recipe
@@ -68,6 +68,8 @@ operation {
   type: "Erf"
   input: "mul_sqrt"
   output: "erf"
+  custom_code: "Erf"
+  extype: "Custom"
 }
 operation {
   type: "Add"

diff --git a/res/TensorFlowLiteRecipes/Net_Gelu_001/test.recipe b/res/TensorFlowLiteRecipes/Net_Gelu_001/test.recipe
@@ -68,6 +68,8 @@ operation {
   type: "Erf"
   input: "mul_sqrt"
   output: "erf"
+  custom_code: "Erf"
+  extype: "Custom"
 }
 operation {
   type: "Add"