diff options
author | 2016-09-29 06:28:56 -0800 | |
---|---|---|
committer | 2016-09-29 07:35:19 -0700 | |
commit | 783e8388db67f3fefc6c714d479615821a9dc5e7 (patch) | |
tree | a25e65ce1552ca593f7cf72d61b0f35bfd18613f /tensorflow/core/framework/selective_registration.h | |
parent | 7a0f6252d7f93b8cc78e8cc61c88f45d98642b30 (diff) |
Add python/tools/print_selective_registration_header utility for printing a
header for use with 'selective registration' (see
core/framework/selective_registration.h), given an input model.
Change: 134659033
Diffstat (limited to 'tensorflow/core/framework/selective_registration.h')
-rw-r--r-- | tensorflow/core/framework/selective_registration.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/framework/selective_registration.h b/tensorflow/core/framework/selective_registration.h index 1227500782..751e2cde84 100644 --- a/tensorflow/core/framework/selective_registration.h +++ b/tensorflow/core/framework/selective_registration.h @@ -31,7 +31,8 @@ limitations under the License. // functions should be defined as valid constexpr functions, so that they are // evaluated at compile time: this is needed to make symbols referenced by // un-registered objects unused, and therefore allow the linker to strip them -// out. +// out. See tools/print_required_ops/print_selective_registration_header.py +// for a tool that can be used to generate ops_to_register.h. #include "ops_to_register.h" // Op kernel classes for which ShouldRegisterOpKernel returns false will not be |