#include "tensorflow/core/framework/common_shape_fns.h"
#include "tensorflow/core/framework/op.h"

namespace tensorflow {

REGISTER_OP("RunGraph")
.Input("feed: T1")
.Output("fetch: T2")
.Attr("T1: list(type) >= 0")
.Attr("T2: list(type) >= 0")
.Attr("feed_names: list(string) >=0")
.Attr("fetch_names: list(string) >=0")
.Attr("loc: string")
.Attr("graph_handle: string")
.Attr("ps_graph_count: int")
.SetShapeFn(shape_inference::UnknownShape);

}  // namespace tensorflow

