Add shape inference for outside_compilation graph rewrite. Pull out enough of...
Add shape inference for outside_compilation graph rewrite. Pull out enough of the graph to enable inference of the shape of a SendFromHost Op once the shape of corresponding RecvAtHost Ops are known. PiperOrigin-RevId: 184153187
Loading
Please sign in to comment