diff options
author | 2017-09-29 09:59:41 -0700 | |
---|---|---|
committer | 2017-09-29 10:03:55 -0700 | |
commit | 9b2912e745b1e6e20867ae3e7e58c7c7df5ded52 (patch) | |
tree | 292a5fbec5be4bf6ffc0ee65b8146c48a2156d19 /tensorflow/contrib/summary | |
parent | 83ba92a2d232c6379c24ab6883c01f1e466d3c08 (diff) |
TFE: Add tfe.SummaryWriter and usage examples
* Revised contrib/summary/summary_ops.py so that multiple summary writers can be created in the same process, each of them writing to a separate logdir.
* Based on the above, the SummaryWriter class supports multiple instances, each holding an independent global_step counter and an independent logdir.
* As the examples in linear_regerssion.py and cart_pole.py show, the SummaryWriter class simplifies user code by
1) taking care of the registration of unique writer keys,
2) moving Tensors from GPU to CPU if necessary,
3) creating an independent global_step tensor,
4) wrapping around the details of incrementing global_step.
PiperOrigin-RevId: 170495375
Diffstat (limited to 'tensorflow/contrib/summary')
-rw-r--r-- | tensorflow/contrib/summary/BUILD | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/contrib/summary/BUILD b/tensorflow/contrib/summary/BUILD index 527deab86a..d09ad48e10 100644 --- a/tensorflow/contrib/summary/BUILD +++ b/tensorflow/contrib/summary/BUILD @@ -13,7 +13,10 @@ load( tf_gen_op_wrapper_py( name = "gen_summary_ops", out = "gen_summary_ops.py", - deps = ["//tensorflow/core:summary_ops_op_lib"], + visibility = ["//tensorflow:internal"], + deps = [ + "//tensorflow/core:summary_ops_op_lib", + ], ) py_test( |