Skip to content

Commit 9a105fa

Browse files
authored
replace inputDtypes with outputDtypes (#363)
1 parent dde35ea commit 9a105fa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,7 @@ public Map<String, Operand<?>> call(Scope scope, Map<String, Operand<?>> argumen
244244
PartitionedCall.create(
245245
scope,
246246
inputList,
247-
Arrays.stream(inputDtypes)
247+
Arrays.stream(outputDtypes)
248248
.map(x -> TensorTypeRegistry.find(x).type())
249249
.collect(Collectors.toList()),
250250
this)

0 commit comments

Comments
 (0)