diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-02-01 11:23:07 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-01 11:43:10 -0800 |
commit | 8fe32029f77b287a207c3bc3e0127fb35a3e23da (patch) | |
tree | cb4e0e5434e21f4ba2a8e980610503b78b3337b9 /tensorflow/examples/multibox_detector | |
parent | 287e845c52d0cb5fb7f5b64658ef513883f66fc5 (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.cc | 2 |
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; |