aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/java
diff options
context:
space:
mode:
authorGravatar karl@kubx.ca <karl@kubx.ca>2018-06-13 17:16:06 -0400
committerGravatar karl@kubx.ca <karl@kubx.ca>2018-06-13 17:16:06 -0400
commit377815f6aa7871e428b98624db44f537875daf06 (patch)
tree279ed39e478d97c6deb783404cd64bd9a42844cc /tensorflow/java
parent6131e85cd75510b37cea781da6da21b74ed6aa7d (diff)
Nit
Diffstat (limited to 'tensorflow/java')
-rw-r--r--tensorflow/java/src/gen/java/org/tensorflow/processor/OperatorProcessor.java8
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/java/src/gen/java/org/tensorflow/processor/OperatorProcessor.java b/tensorflow/java/src/gen/java/org/tensorflow/processor/OperatorProcessor.java
index aa624a9e83..3524160d87 100644
--- a/tensorflow/java/src/gen/java/org/tensorflow/processor/OperatorProcessor.java
+++ b/tensorflow/java/src/gen/java/org/tensorflow/processor/OperatorProcessor.java
@@ -215,21 +215,21 @@ public final class OperatorProcessor extends AbstractProcessor {
AnnotationMirror am = getAnnotationMirror(opClass, annotation);
String groupName = getAnnotationElementValueAsString("group", am);
String methodName = getAnnotationElementValueAsString("name", am);
+ ClassName opClassName = ClassName.get(opClass);
if (Strings.isNullOrEmpty(methodName)) {
- methodName = CaseFormat.UPPER_CAMEL.to(CaseFormat.LOWER_CAMEL, ClassName.get(opClass).simpleName());
+ methodName = CaseFormat.UPPER_CAMEL.to(CaseFormat.LOWER_CAMEL, opClassName.simpleName());
}
// Build a method for each @Operator found in the class path. There should be one method per operation factory called
// "create", which takes in parameter a scope and, optionally, a list of arguments
for (ExecutableElement opMethod : ElementFilter.methodsIn(opClass.getEnclosedElements())) {
if (opMethod.getModifiers().contains(Modifier.STATIC) && opMethod.getSimpleName().contentEquals("create")) {
- MethodSpec method = buildOpMethod(methodName, opClass, opMethod);
+ MethodSpec method = buildOpMethod(methodName, opClassName, opMethod);
groupedMethods.put(groupName, method);
}
}
}
- private MethodSpec buildOpMethod(String methodName, TypeElement opClass, ExecutableElement factoryMethod) {
- ClassName opClassName = ClassName.get(opClass);
+ private MethodSpec buildOpMethod(String methodName, ClassName opClassName, ExecutableElement factoryMethod) {
MethodSpec.Builder builder =
MethodSpec.methodBuilder(methodName)
.addModifiers(Modifier.PUBLIC)