Add a way to fuse a graph by remote graph executor so that users don't need to be aware of supported op types, node names, subgraph stracture etc.

PiperOrigin-RevId: 162411763

1 files changed, 7 insertions, 0 deletions

diff --git a/tensorflow/core/kernels/i_remote_fused_graph_executor.h b/tensorflow/core/kernels/i_remote_fused_graph_executor.h
index 09d1f43ff1..05b76172b2 100644
--- a/tensorflow/core/kernels/i_remote_fused_graph_executor.h
+++ b/tensorflow/core/kernels/i_remote_fused_graph_executor.h
@@ -59,6 +59,13 @@ class IRemoteFusedGraphExecutor {
   virtual bool ReadOutputNode(const string& node_name,
                               TensorAllocatorFunc tensor_allocator) = 0;
 
+  virtual Status FuseRemoteGraph(const GraphDef& original_graph_def,
+                                 const std::vector<string>& inputs,
+                                 const std::vector<string>& outputs,
+                                 GraphDef* fused_graph_def) = 0;
+
+  virtual bool IsEnabled() const = 0;
+
  private:
   TF_DISALLOW_COPY_AND_ASSIGN(IRemoteFusedGraphExecutor);
 };