Keep functions disabled for now

Signed-off-by: Ryan Nett <[email protected]>

Author: rnett

tensorflow-core/tensorflow-core-api/src/bazel/op_generator/op_generator.cc

@@ -548,6 +548,11 @@ bool CanGenerateOp(const OpDef& op_def, const ApiDef& api_def) {
   if (api_def.visibility() == ApiDef::SKIP) {
     return false;
   }
+  for (const auto& attr : op_def.attr()) {
+    if (attr.type() == "func" || attr.type() == "list(func)") {
+      return false;  // TODO function attributes are supported, just need to enable
+    }
+  }
   return true;
 }
 

tensorflow-core/tensorflow-core-api/src/bazel/op_generator/op_specs.cc

@@ -190,6 +190,15 @@ string SnakeToCamelCase(const string& str, bool upper = false) {
       result += c;
     }
   }
+
+  if(result == "for"){
+    return "for_";
+  } else if(result == "if"){
+    return "if_";
+  } else if(result == "while"){
+    return "while_";
+  }
+
   return result;
 }