diff options
| author | Lester Solbakken <lesters@users.noreply.github.com> | 2018-01-22 17:04:52 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2018-01-22 17:04:52 +0100 |
| commit | b50b47bc22a011dc24f1906477faae21aeade1a5 (patch) | |
| tree | f999396b7a3cb34a614ff77f095fdb9fb4287610 /searchlib/src/main | |
| parent | 95efa074a0382ae6dcb6ebaf2256a9f5915e8a28 (diff) | |
| parent | e1ddfa9a9c862d353b7f6f1759786ce8b689b25c (diff) | |
Merge branch 'master' into lesters/add-batch-norm-tf-case
Diffstat (limited to 'searchlib/src/main')
| -rw-r--r-- | searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/tensorflow/OperationMapper.java | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/tensorflow/OperationMapper.java b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/tensorflow/OperationMapper.java index 2581cd46286..85452d16a77 100644 --- a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/tensorflow/OperationMapper.java +++ b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/tensorflow/OperationMapper.java @@ -126,7 +126,7 @@ class OperationMapper { String name = tfNode.getInput(0); Tensor defaultValue = getConstantTensor(model, name); result.constant(name, defaultValue); - result.macro(name, new RankingExpression(name, new ReferenceNode("constant('" + name + "')"))); + result.macro(name, new RankingExpression(name, new ReferenceNode("constant(\"" + name + "\")"))); // The default value will be provided by the macro. Users can override macro to change value. return new TypedTensorFunction(defaultValue.type(), new VariableTensor(name)); } @@ -159,7 +159,7 @@ class OperationMapper { Tensor constant = getConstantTensor(model, name); result.constant(name, constant); return new TypedTensorFunction(constant.type(), - new TensorFunctionNode.TensorFunctionExpressionNode(new ReferenceNode("constant('" + name + "')"))); + new TensorFunctionNode.TensorFunctionExpressionNode(new ReferenceNode("constant(\"" + name + "\")"))); } private Tensor getConstantTensor(SavedModelBundle model, String name) { |