aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/examples/multibox_detector
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-02-01 11:23:07 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-02-01 11:43:10 -0800
commit8fe32029f77b287a207c3bc3e0127fb35a3e23da (patch)
treecb4e0e5434e21f4ba2a8e980610503b78b3337b9 /tensorflow/examples/multibox_detector
parent287e845c52d0cb5fb7f5b64658ef513883f66fc5 (diff)
Add a mechanism for hiding, skipping, and modifying the generated op
functions for C++. A souped-up version of the hidden_ops mechanism in Python, the intent is to use this for most or all of the client languages, with a common list of changes to make in a common file and per-language overrides. Also: * include the documentation for outputs in the generated comments * several updates to C++ API to match Python * fix C++ shape function for ConcatV2 now that we use it by default * split op_gen_lib out of core:framework, since it is only used by the op generators, and I don't want to add another proto to mobile builds Change: 146267344
Diffstat (limited to 'tensorflow/examples/multibox_detector')
-rw-r--r--tensorflow/examples/multibox_detector/main.cc2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/examples/multibox_detector/main.cc b/tensorflow/examples/multibox_detector/main.cc
index 42972078e5..0d6875671b 100644
--- a/tensorflow/examples/multibox_detector/main.cc
+++ b/tensorflow/examples/multibox_detector/main.cc
@@ -184,7 +184,7 @@ Status GetTopDetections(const std::vector<Tensor>& outputs, int how_many_labels,
using namespace ::tensorflow::ops; // NOLINT(build/namespaces)
string output_name = "top_k";
- TopKV2(root.WithOpName(output_name), outputs[0], how_many_labels);
+ TopK(root.WithOpName(output_name), outputs[0], how_many_labels);
// This runs the GraphDef network definition that we've just constructed, and
// returns the results in the output tensors.
tensorflow::GraphDef graph;