diff --git a/compiler/circlechef/core/src/ModelChef.cpp b/compiler/circlechef/core/src/ModelChef.cpp index 177cdccb322..98eb00e5c94 100644 --- a/compiler/circlechef/core/src/ModelChef.cpp +++ b/compiler/circlechef/core/src/ModelChef.cpp @@ -135,7 +135,7 @@ gather_builtincode_map(const ::circlechef::ModelRecipe &model_recipe) for (const auto &operation : model_recipe.operation()) { - if (operation.type() == "Custom" || (operation.has_extype() && operation.extype() == "Custom")) + if (operation.type() == "Custom") continue; auto op_chef = op_chef_registry().lookup(operation.type()).create(&operation); @@ -151,8 +151,7 @@ gather_builtincode_map(const ::circlechef::ModelRecipe &model_recipe) const auto &graph = model_recipe.graph(g); for (const auto &operation : graph.operation()) { - if (operation.type() == "Custom" || - (operation.has_extype() && operation.extype() == "Custom")) + if (operation.type() == "Custom") continue; auto op_chef = op_chef_registry().lookup(operation.type()).create(&operation); @@ -172,7 +171,7 @@ std::set gather_customcode_set(const ::circlechef::ModelRecipe &mod std::set customcode_set; for (const auto &operation : model_recipe.operation()) { - if (operation.type() == "Custom" || (operation.has_extype() && operation.extype() == "Custom")) + if (operation.type() == "Custom") { assert(not operation.custom_code().empty()); customcode_set.insert(operation.custom_code()); @@ -185,8 +184,7 @@ std::set gather_customcode_set(const ::circlechef::ModelRecipe &mod const auto &graph = model_recipe.graph(g); for (const auto &operation : graph.operation()) { - if (operation.type() == "Custom" || - (operation.has_extype() && operation.extype() == "Custom")) + if (operation.type() == "Custom") { assert(not operation.custom_code().empty()); customcode_set.insert(operation.custom_code()); diff --git a/compiler/circlechef/proto/circlechef.proto b/compiler/circlechef/proto/circlechef.proto index c2db55a9d84..c01923a1a2e 100644 --- a/compiler/circlechef/proto/circlechef.proto +++ b/compiler/circlechef/proto/circlechef.proto @@ -93,7 +93,6 @@ message Operation { optional int32 version = 4 [default = 1]; optional string custom_code = 5; - optional string extype = 99; optional BatchMatMulOptions batch_matmul_options = 100; optional InstanceNormOptions instance_norm_options = 101; optional BCQFullyConnectedOptions bcq_fully_connected_options = 102;