diff options
Diffstat (limited to 'tensorflow/core/common_runtime/lower_if_op.h')
-rw-r--r-- | tensorflow/core/common_runtime/lower_if_op.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/core/common_runtime/lower_if_op.h b/tensorflow/core/common_runtime/lower_if_op.h index a9ef39ae5c..5ab1123e3f 100644 --- a/tensorflow/core/common_runtime/lower_if_op.h +++ b/tensorflow/core/common_runtime/lower_if_op.h @@ -29,8 +29,9 @@ class LowerIfOpPass : public GraphOptimizationPass { Status Run(const GraphOptimizationPassOptions& options) override; private: - // Rewrite the given If node `n` in graph `g` to use the switch-merge form. - Status RewriteNode(Node* n, Graph* g); + // Rewrite the given If node `n` in graph `g` to use the switch-merge + // form. `flib` should contain the branch functions referenced by `n`. + Status RewriteNode(Node* n, const FunctionLibraryDefinition& flib, Graph* g); }; } // namespace tensorflow |