diff --git a/tensorflow/tensorboard/components/tf-graph/demo/demo_datasets.json b/tensorflow/tensorboard/components/tf-graph/demo/demo_datasets.json index 187780757bf..f5ca9aada79 100644 --- a/tensorflow/tensorboard/components/tf-graph/demo/demo_datasets.json +++ b/tensorflow/tensorboard/components/tf-graph/demo/demo_datasets.json @@ -110,6 +110,12 @@ "name": "AlexNet", "path": "alexnet.pbtxt" }, + { + "id": "alexprivate", + "name": "AlexNet Private", + "path": "alexnet.pbtxt", + "private": true + }, { "name": "TestError404", "path": "nofile" diff --git a/tensorflow/tensorboard/components/tf-graph/demo/tf-graph-demo.html b/tensorflow/tensorboard/components/tf-graph/demo/tf-graph-demo.html index 723dcbd7e31..934115a9063 100644 --- a/tensorflow/tensorboard/components/tf-graph/demo/tf-graph-demo.html +++ b/tensorflow/tensorboard/components/tf-graph/demo/tf-graph-demo.html @@ -123,6 +123,8 @@ Polymer({ } d3.json(DEMO_DATASETS, function (error, datasets) { + let publicDatasets = []; + if (error) { console.log('Error loading demo datasets:'); console.log(error); @@ -135,6 +137,8 @@ Polymer({ _.each(datasets, function(dataset, index) { if (queryParams['graphid'] && dataset.id == queryParams['graphid']) { selectedDataset = index; + } else if (dataset['private']) { + return; } dataset.path = this._normalizePath(dataset.path); @@ -143,8 +147,9 @@ Polymer({ metadata.path = this._normalizePath(metadata.path); }, this); } + publicDatasets.push(dataset); }, this); - this.set('datasets', datasets); + this.set('datasets', publicDatasets); if (selectedDataset != 0) { this.set('selectedDataset', selectedDataset); }