aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/lookup_table_op.cc
diff options
context:
space:
mode:
authorGravatar Vijay Vasudevan <vrv@google.com>2015-11-10 15:23:01 -0800
committerGravatar Vijay Vasudevan <vrv@google.com>2015-11-10 15:23:01 -0800
commitc61c39614a4780464037226d9d3c28101f7e8e46 (patch)
treee615bdecafe406289b0a6ed77b40e6d5865c9816 /tensorflow/core/kernels/lookup_table_op.cc
parent9274f5aa478879386870bb0cb889e432d4aee330 (diff)
TensorFlow: upstream changes to git (doc fixes).
Changes: - Fix typos across several files contributed by Erik Erwitt, and Michael R. Berstein - Fix bug in translate example (fr->en typo) by schuster - Updates to some documentation (mcoram,shlens,vrv,joshl) - Fix to Android camera demo app window size detection (andrewharp) - Fix to support lookup table of high rank tensors (yleon) - Fix invalid op names for parse_example (dga) Base CL: 107531031
Diffstat (limited to 'tensorflow/core/kernels/lookup_table_op.cc')
-rw-r--r--tensorflow/core/kernels/lookup_table_op.cc6
1 files changed, 1 insertions, 5 deletions
diff --git a/tensorflow/core/kernels/lookup_table_op.cc b/tensorflow/core/kernels/lookup_table_op.cc
index 2bab4df94f..2d647d8ed4 100644
--- a/tensorflow/core/kernels/lookup_table_op.cc
+++ b/tensorflow/core/kernels/lookup_table_op.cc
@@ -109,9 +109,6 @@ class LookupTableFindOp : public OpKernel {
OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs));
const Tensor& input = ctx->input(1);
- OP_REQUIRES(ctx, TensorShapeUtils::IsVector(input.shape()),
- errors::InvalidArgument("Input must be a vector, not ",
- input.shape().DebugString()));
const Tensor& default_value = ctx->input(2);
OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(default_value.shape()),
@@ -119,8 +116,7 @@ class LookupTableFindOp : public OpKernel {
default_value.shape().DebugString()));
Tensor* out;
- OP_REQUIRES_OK(ctx,
- ctx->allocate_output("output_values", input.shape(), &out));
+ OP_REQUIRES_OK(ctx, ctx->allocate_output("values", input.shape(), &out));
OP_REQUIRES_OK(ctx, table->Find(input, out, default_value));
}