diff options
Diffstat (limited to 'tensorflow/core/framework/op_kernel.h')
-rw-r--r-- | tensorflow/core/framework/op_kernel.h | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/tensorflow/core/framework/op_kernel.h b/tensorflow/core/framework/op_kernel.h index 6c4c3a2ac1..1fc5e9908e 100644 --- a/tensorflow/core/framework/op_kernel.h +++ b/tensorflow/core/framework/op_kernel.h @@ -1044,7 +1044,6 @@ class OpKernelContext { // For control flow. FrameAndIter frame_iter() const { return params_->frame_iter; } bool is_input_dead() const { return params_->is_input_dead; } - bool* is_output_dead() { return &is_output_dead_; } // May be used, e.g., to get GPU handles, etc. // TODO(tucker): Add example usage. @@ -1143,8 +1142,6 @@ class OpKernelContext { // Constructed only if <params->record_tensor_accesses>. ManualConstructor<UniqueTensorReferences> referenced_tensors_ GUARDED_BY(mu_); - bool is_output_dead_ = false; - // The following data members are only used when allocation tracking is // enabled. mutable mutex stats_mu_; @@ -1307,6 +1304,13 @@ void LogAllRegisteredKernels(); // Gets a list of all registered kernels. KernelList GetAllRegisteredKernels(); +// Gets a list of all registered kernels for which predicate returns true +KernelList GetFilteredRegisteredKernels( + const std::function<bool(const KernelDef&)>& predicate); + +// Gets a list of all registered kernels for a given op +KernelList GetRegisteredKernelsForOp(StringPiece op_name); + namespace kernel_factory { class OpKernelRegistrar { |