aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-01-24 20:13:14 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-01-24 20:16:57 -0800
commit0d11ed25e49e9dfbb45c15c1af9a5892e9146768 (patch)
tree74ce4f673267dce7dcd30087c429454f2510ae66
parent1a6216e61e804019cd64732d6f95d4d9bbedb594 (diff)
Make the graph generation for TFBT deterministic.
PiperOrigin-RevId: 183183086
-rw-r--r--tensorflow/contrib/boosted_trees/python/ops/batch_ops_utils.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/boosted_trees/python/ops/batch_ops_utils.py b/tensorflow/contrib/boosted_trees/python/ops/batch_ops_utils.py
index 23168bf493..b281a4c6d1 100644
--- a/tensorflow/contrib/boosted_trees/python/ops/batch_ops_utils.py
+++ b/tensorflow/contrib/boosted_trees/python/ops/batch_ops_utils.py
@@ -104,7 +104,7 @@ def run_handler_scheduled_ops(per_handler_ops, stamp, worker_device):
batched_ops = collections.defaultdict(list)
# Group the ops by their batching_key. Ops that share the same batching key
# can be executed together.
- for handler in per_handler_ops.keys():
+ for handler in sorted(per_handler_ops.keys()):
for op in per_handler_ops[handler]:
batched_ops[(op.batching_key(), op.batch_runner_fn())].append(op)
op_results = {}