Skip to content

Commit 759a754

Browse files
committed
Rename withDevice's parameter
Signed-off-by: Ryan Nett <JNett96@gmail.com>
1 parent 2ddbb6c commit 759a754

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/GradientScope.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,8 @@ public GradientScope withNameAsSubScope(String defaultName) {
5252
}
5353

5454
@Override
55-
public GradientScope withDevice(DeviceSpec deviceSpec) {
56-
return new GradientScope(nativeScope.WithDevice(deviceSpec.toString()), graph, deviceSpec.toString());
55+
public GradientScope withDevice(DeviceSpec newDevice) {
56+
return new GradientScope(nativeScope.WithDevice(newDevice.toString()), graph, newDevice.toString());
5757
}
5858

5959
@Override

tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/OpScope.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,8 +64,8 @@ public OpScope withNameAsSubScope(String defaultName) {
6464
}
6565

6666
@Override
67-
public OpScope withDevice(DeviceSpec deviceSpec) {
68-
return new OpScope(env, nameScope, controlDependencies, deviceSpec, isInit);
67+
public OpScope withDevice(DeviceSpec newDevice) {
68+
return new OpScope(env, nameScope, controlDependencies, newDevice, isInit);
6969
}
7070

7171
@Override

tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/Scope.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -136,10 +136,10 @@ public interface Scope {
136136
* <p>Operations created within this scope will place the created operations on the device(s)
137137
* matching the provided spec.
138138
*
139-
* @param deviceSpec device specification for an operator in the returned scope
139+
* @param newDevice device specification for an operator in the returned scope
140140
* @return a new Scope that uses opName for operations.
141141
*/
142-
Scope withDevice(DeviceSpec deviceSpec);
142+
Scope withDevice(DeviceSpec newDevice);
143143

144144
/** Get an extension of this scope that generates initialization ops. */
145145
Scope withInitScope();

0 commit comments

Comments
 (0)