Add padding between main, in-extract and out-extract graphs.

Change: 117444098

1 files changed, 5 insertions, 2 deletions

diff --git a/tensorflow/tensorboard/components/tf-graph-common/lib/scene/scene.ts b/tensorflow/tensorboard/components/tf-graph-common/lib/scene/scene.ts
index b6eb3f7d81..1b6cb3a58c 100644
--- a/tensorflow/tensorboard/components/tf-graph-common/lib/scene/scene.ts
+++ b/tensorflow/tensorboard/components/tf-graph-common/lib/scene/scene.ts
@@ -321,15 +321,18 @@ function position(sceneGroup, renderNode: render.RenderGroupNodeInfo) {
 
   // in-extract
   let hasInExtract = renderNode.isolatedInExtract.length > 0;
+  let hasOutExtract = renderNode.isolatedOutExtract.length > 0;
+
   if (hasInExtract) {
+    let offset = layout.PARAMS.subscene.meta.extractXOffset;
     let inExtractX = renderNode.coreBox.width -
-      renderNode.inExtractBox.width / 2 - renderNode.outExtractBox.width;
+      renderNode.inExtractBox.width / 2 - renderNode.outExtractBox.width -
+          (hasOutExtract ? offset : 0);
     translate(selectChild(sceneGroup, "g", Class.Scene.INEXTRACT),
                     inExtractX, yTranslate);
   }
 
   // out-extract
-  let hasOutExtract = renderNode.isolatedOutExtract.length > 0;
   if (hasOutExtract) {
     let outExtractX = renderNode.coreBox.width -
       renderNode.outExtractBox.width / 2;