aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tensor_forest/python/tensor_forest.py
diff options
context:
space:
mode:
authorGravatar Andrew Selle <aselle@google.com>2016-12-06 11:17:07 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-12-06 12:59:17 -0800
commit6de612e7767c4114d65cb06fc6606436e3081dc3 (patch)
tree9b54f3a4179af50365f47fa63804f6c083632ba6 /tensorflow/contrib/tensor_forest/python/tensor_forest.py
parentff0ef451a1c66b87441f996623c0d541665330c1 (diff)
Force fix documentation causing open source blocks.
Change: 141201358
Diffstat (limited to 'tensorflow/contrib/tensor_forest/python/tensor_forest.py')
-rw-r--r--tensorflow/contrib/tensor_forest/python/tensor_forest.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/tensor_forest/python/tensor_forest.py b/tensorflow/contrib/tensor_forest/python/tensor_forest.py
index 18f64b9ab5..504462c8c6 100644
--- a/tensorflow/contrib/tensor_forest/python/tensor_forest.py
+++ b/tensorflow/contrib/tensor_forest/python/tensor_forest.py
@@ -633,7 +633,7 @@ class RandomTreeGraphs(object):
if isinstance(input_data, sparse_tensor.SparseTensor):
sparse_indices = input_data.indices
sparse_values = input_data.values
- sparse_shape = input_data.shape
+ sparse_shape = input_data.dense_shape
input_data = []
# Count extremely random stats.
@@ -890,7 +890,7 @@ class RandomTreeGraphs(object):
if isinstance(input_data, sparse_tensor.SparseTensor):
sparse_indices = input_data.indices
sparse_values = input_data.values
- sparse_shape = input_data.shape
+ sparse_shape = input_data.dense_shape
input_data = []
return self.inference_ops.tree_predictions(
input_data, sparse_indices, sparse_values, sparse_shape, data_spec,