Skip to content

Commit 463036d

Browse files
committed
fix style
Signed-off-by: Ryan Nett <[email protected]>
1 parent a42ed5b commit 463036d

File tree

2 files changed

+17
-11
lines changed

2 files changed

+17
-11
lines changed

tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/EagerOperationBuilder.java

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,9 @@
5656
final class EagerOperationBuilder implements OperationBuilder {
5757

5858
EagerOperationBuilder(EagerSession session, String type, String name) {
59-
if(!session.isOpEnabled(type))
59+
if (!session.isOpEnabled(type)) {
6060
throw new IllegalArgumentException("Op " + type + " is not valid in eager mode.");
61+
}
6162

6263
this.session = session;
6364
this.type = type;
@@ -78,15 +79,15 @@ public EagerOperation build() {
7879

7980
@Override
8081
public EagerOperationBuilder addInput(Output<?> input) {
81-
addInput(opHandle, (TFE_TensorHandle)input.getUnsafeNativeHandle());
82+
addInput(opHandle, (TFE_TensorHandle) input.getUnsafeNativeHandle());
8283
return this;
8384
}
8485

8586
@Override
8687
public EagerOperationBuilder addInputList(Output<?>[] inputs) {
8788
TFE_TensorHandle[] inputHandles = new TFE_TensorHandle[inputs.length];
8889
for (int i = 0; i < inputs.length; ++i) {
89-
inputHandles[i] = (TFE_TensorHandle)inputs[i].getUnsafeNativeHandle();
90+
inputHandles[i] = (TFE_TensorHandle) inputs[i].getUnsafeNativeHandle();
9091
}
9192
addInputList(opHandle, inputHandles);
9293
return this;
@@ -229,7 +230,9 @@ public EagerOperationBuilder setAttr(String name, Shape[] values) {
229230
private final String type;
230231
private final String name;
231232

232-
/** This value should be >= to the maximum number of outputs in any op */
233+
/**
234+
* This value should be >= to the maximum number of outputs in any op
235+
*/
233236
private static final int MAX_OUTPUTS_PER_OP = 1000;
234237

235238
private static void requireOp(TFE_Op handle) {
@@ -361,7 +364,7 @@ private static void setAttrFloatList(TFE_Op opHandle, String name, float[] value
361364

362365
private static void setAttrBool(TFE_Op opHandle, String name, boolean value) {
363366
requireOp(opHandle);
364-
TFE_OpSetAttrBool(opHandle, name, (byte)(value ? 1 : 0));
367+
TFE_OpSetAttrBool(opHandle, name, (byte) (value ? 1 : 0));
365368
}
366369

367370
private static void setAttrBoolList(TFE_Op opHandle, String name, boolean[] values) {
@@ -413,7 +416,7 @@ private static void setAttrShapeList(TFE_Op opHandle, String name, long[] shapes
413416
}
414417
TF_Status status = TF_Status.newStatus();
415418
TFE_OpSetAttrShapeList(opHandle, new BytePointer(name), shapesPointers, new IntPointer(numDims),
416-
numDims.length, status);
419+
numDims.length, status);
417420
}
418421
}
419422
}

tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/GraphOperationBuilder.java

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -54,12 +54,15 @@
5454
import org.tensorflow.ndarray.Shape;
5555
import org.tensorflow.proto.framework.DataType;
5656

57-
/** An {@link OperationBuilder} for adding {@link GraphOperation}s to a {@link Graph}. */
57+
/**
58+
* An {@link OperationBuilder} for adding {@link GraphOperation}s to a {@link Graph}.
59+
*/
5860
public final class GraphOperationBuilder implements OperationBuilder {
5961

6062
GraphOperationBuilder(Graph graph, String type, String name) {
61-
if(!graph.isOpEnabled(type))
63+
if (!graph.isOpEnabled(type)) {
6264
throw new IllegalArgumentException("Op " + type + " is not valid in graph mode.");
65+
}
6366

6467
this.graph = graph;
6568
Graph.Reference r = graph.ref();
@@ -106,7 +109,7 @@ public GraphOperationBuilder addControlInput(Operation control) {
106109
public GraphOperationBuilder addInput(Output<?> input) {
107110
Graph.Reference r = graph.ref();
108111
try {
109-
addInput(unsafeNativeHandle, (TF_Operation)input.getUnsafeNativeHandle(), input.index());
112+
addInput(unsafeNativeHandle, (TF_Operation) input.getUnsafeNativeHandle(), input.index());
110113
} finally {
111114
r.close();
112115
}
@@ -120,7 +123,7 @@ public GraphOperationBuilder addInputList(Output<?>[] inputs) {
120123
TF_Operation[] opHandles = new TF_Operation[inputs.length];
121124
int[] indices = new int[inputs.length];
122125
for (int i = 0; i < inputs.length; ++i) {
123-
opHandles[i] = (TF_Operation)inputs[i].getUnsafeNativeHandle();
126+
opHandles[i] = (TF_Operation) inputs[i].getUnsafeNativeHandle();
124127
indices[i] = inputs[i].index();
125128
}
126129
addInputList(unsafeNativeHandle, opHandles, indices);
@@ -447,7 +450,7 @@ private static void setAttrFloatList(TF_OperationDescription handle, String name
447450

448451
private static void setAttrBool(TF_OperationDescription handle, String name, boolean value) {
449452
requireHandle(handle);
450-
TF_SetAttrBool(handle, name, (byte)(value ? 1 : 0));
453+
TF_SetAttrBool(handle, name, (byte) (value ? 1 : 0));
451454
}
452455

453456
private static void setAttrBoolList(TF_OperationDescription handle, String name, boolean[] value) {

0 commit comments

Comments
 (0)