summaryrefslogtreecommitdiffstats
path: root/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java
diff options
context:
space:
mode:
authorgjoranv <gjoranv@gmail.com>2017-12-17 21:44:49 +0100
committerGitHub <noreply@github.com>2017-12-17 21:44:49 +0100
commit03bce1fe1a494f2ac9d4268d4c90b08011b3f600 (patch)
tree180f294d2ac97d641f0266216ffdc328db9bfef8 /vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java
parentb72e55b87eecae006ed92976151137a80d75be0f (diff)
Revert "Bratseth/tensorflow models"
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java')
-rw-r--r--vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java9
1 files changed, 2 insertions, 7 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java
index 4071917c2b5..bb27e937699 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java
@@ -3,7 +3,6 @@ package com.yahoo.tensor.functions;
import com.google.common.annotations.Beta;
import com.google.common.collect.ImmutableList;
-import com.yahoo.tensor.TensorType;
import java.util.List;
@@ -15,17 +14,13 @@ public class Matmul extends CompositeTensorFunction {
private final TensorFunction argument1, argument2;
private final String dimension;
-
+
public Matmul(TensorFunction argument1, TensorFunction argument2, String dimension) {
this.argument1 = argument1;
this.argument2 = argument2;
this.dimension = dimension;
}
- public static TensorType outputType(TensorType a, TensorType b, String dimension) {
- return Join.outputType(a, b);
- }
-
@Override
public List<TensorFunction> functionArguments() { return ImmutableList.of(argument1, argument2); }
@@ -44,7 +39,7 @@ public class Matmul extends CompositeTensorFunction {
Reduce.Aggregator.sum,
dimension);
}
-
+
@Override
public String toString(ToStringContext context) {
return "matmul(" + argument1.toString(context) + ", " + argument2.toString(context) + ", " + dimension + ")";