diff options
author | 2018-01-24 20:13:14 -0800 | |
---|---|---|
committer | 2018-01-24 20:16:57 -0800 | |
commit | 0d11ed25e49e9dfbb45c15c1af9a5892e9146768 (patch) | |
tree | 74ce4f673267dce7dcd30087c429454f2510ae66 | |
parent | 1a6216e61e804019cd64732d6f95d4d9bbedb594 (diff) |
Make the graph generation for TFBT deterministic.
PiperOrigin-RevId: 183183086
-rw-r--r-- | tensorflow/contrib/boosted_trees/python/ops/batch_ops_utils.py | 2 |
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 = {} |