Adapt to BatchMatMulV3

karllessard · karllessard · commit 1e28b11b084c · 2021-09-15T09:30:21.000-04:00
diff --git a/tensorflow-framework/src/main/java/org/tensorflow/framework/op/linalg/MatMul.java b/tensorflow-framework/src/main/java/org/tensorflow/framework/op/linalg/MatMul.java
@@ -241,7 +241,7 @@ public static <T extends TNumber> Operand<T> matmul(
         adjointB = true;
       }
       return BatchMatMul.create(
-          lscope, a, b, BatchMatMul.adjX(adjointA), BatchMatMul.adjY(adjointB));
+          lscope, a, b, a.type(), BatchMatMul.adjX(adjointA), BatchMatMul.adjY(adjointB));
     }
 
     // Neither matmul nor sparse_matmul support adjoint, so we conjugate

Original file line number	Diff line number	Diff line change
`@@ -241,7 +241,7 @@ public static <T extends TNumber> Operand<T> matmul(`
`241`	`241`	`adjointB = true;`
`242`	`242`	`}`
`243`	`243`	`return BatchMatMul.create(`
`244`		`- lscope, a, b, BatchMatMul.adjX(adjointA), BatchMatMul.adjY(adjointB));`
	`244`	`+ lscope, a, b, a.type(), BatchMatMul.adjX(adjointA), BatchMatMul.adjY(adjointB));`
`245`	`245`	`}`
`246`	`246`
`247`	`247`	`// Neither matmul nor sparse_matmul support adjoint, so we conjugate`