aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tpu
Commit message (Collapse)AuthorAge
...
* [TPU] Deprecate the computation_shape attribute to the TpuReplicate op in ↵Gravatar Peter Hawkins2018-09-15
| | | | | | lieu of a new num_cores_per_replica. PiperOrigin-RevId: 213111326
* global_step/sec renamed to global_steps/secGravatar A. Unique TensorFlower2018-09-14
| | | | PiperOrigin-RevId: 212986442
* Add root of profile broken down by program to Profile proto.Gravatar A. Unique TensorFlower2018-09-13
| | | | PiperOrigin-RevId: 212857508
* Give a warning about partitioned variable on TPU and set it to None, instead ↵Gravatar A. Unique TensorFlower2018-09-10
| | | | | | of erring out. PiperOrigin-RevId: 212385555
* Fixed the bug that the pipeline version fit does not return history.Gravatar Jianwei Xie2018-09-07
| | | | PiperOrigin-RevId: 212050240
* Improve weights check for TPU vs CPU.Gravatar Jianwei Xie2018-09-07
| | | | PiperOrigin-RevId: 212039241
* Fixed the KerasOnTPU support for non-sgd optimizer with using_single_core=True.Gravatar Jianwei Xie2018-09-07
| | | | PiperOrigin-RevId: 211984222
* Enable outside_compilation in contrib.tpu module.Gravatar Ruoxin Sang2018-09-06
| | | | PiperOrigin-RevId: 211843340
* Ignore partitioned variable in TPU computation.Gravatar A. Unique TensorFlower2018-09-06
| | | | PiperOrigin-RevId: 211833891
* [Keras / Cloud TPU]: Correct indexing for software pipelining.Gravatar Brennan Saeta2018-09-05
| | | | PiperOrigin-RevId: 211724843
* Re-added proto field for dynamic learning rate support (not usable yet).Gravatar A. Unique TensorFlower2018-09-05
| | | | PiperOrigin-RevId: 211719009
* Implements TPU alltoall op.Gravatar Youlong Cheng2018-09-05
| | | | | | RELNOTES: n/a PiperOrigin-RevId: 211718248
* Remove logging which generates tons of logs for large model.Gravatar Jianwei Xie2018-09-05
| | | | PiperOrigin-RevId: 211707155
* Automatically use single core for stateful RNN in Keras TPU.Gravatar Jianwei Xie2018-09-04
| | | | PiperOrigin-RevId: 211532963
* Removed old dynamic learning rate support code.Gravatar A. Unique TensorFlower2018-09-04
| | | | PiperOrigin-RevId: 211514287
* Add more data fields to step proto.Gravatar A. Unique TensorFlower2018-09-04
| | | | PiperOrigin-RevId: 211501909
* [Cloud TPU / Keras]: Pipeline inputs for performance.Gravatar Brennan Saeta2018-08-31
| | | | | | With this pipelining change (and a little bit of re-tuning of input pipelines to have _less_ parallelism to avoid thread starvation), we are able to significantly reduce the overheads of supporting dynamic shapes with Keras. PiperOrigin-RevId: 211157531
* Add experimental replicated variable support for Keras/TPU.Gravatar Russell Power2018-08-31
| | | | PiperOrigin-RevId: 211129794
* [tf.data / Keras / Cloud TPU]: Support evaluating with a tf.data.DatasetGravatar Brennan Saeta2018-08-31
| | | | PiperOrigin-RevId: 211099558
* Fix typo and formatting issues in tpu.py commentsGravatar Yanan Cao2018-08-29
| | | | PiperOrigin-RevId: 210855008
* Add verbose error message for capture_tpu_profileGravatar A. Unique TensorFlower2018-08-29
| | | | PiperOrigin-RevId: 210827158
* [TF:XLA] Change group_assignment from 1d array attribute to 2d array input ↵Gravatar Youlong Cheng2018-08-28
| | | | | | tensor with shape [num_groups, num_replica_per_group]. PiperOrigin-RevId: 210656091
* Step 1 toward automatically assigning # of TPU cores.Gravatar Jianwei Xie2018-08-28
| | | | | | | | Also fix the API as num_cores is not a good API. Changed to a debugging oriented argument using_single_core. Ideally we should not need that in future. The updated lstm example cannot work with TF 1.10 anymore. PiperOrigin-RevId: 210632592
* Fix error message in TF-keras dataset shape checkGravatar A. Unique TensorFlower2018-08-28
| | | | | | (Dimension and tensor # were transposed in the error message) PiperOrigin-RevId: 210598445
* Misc clean up tpu.pyGravatar Yanan Cao2018-08-27
| | | | PiperOrigin-RevId: 210423935
* Added note that gradient accumulation is experimental.Gravatar A. Unique TensorFlower2018-08-24
| | | | PiperOrigin-RevId: 210187980
* Support Input partition in Predict mode.Gravatar Youlong Cheng2018-08-24
| | | | PiperOrigin-RevId: 210185805
* Add optional num_cores parameter to TPU Distribution StrategyGravatar Frank Chen2018-08-23
| | | | PiperOrigin-RevId: 209969854
* Update initialization of variables in Keras.Gravatar Pavithra Vijay2018-08-21
| | | | | | - Removes the use of global variable collections for initialization. This means that variables created externally (such as input feed variables) will have to be initialized explicitly. PiperOrigin-RevId: 209692235
* Support export eval graph on CPU in TPUEstimator.export_savedmodel().Gravatar A. Unique TensorFlower2018-08-21
| | | | PiperOrigin-RevId: 209682829
* Remove `computation_shape` from tpu_config.py comments.Gravatar Ruoxin Sang2018-08-21
| | | | PiperOrigin-RevId: 209673552
* Add per program op profile.Gravatar A. Unique TensorFlower2018-08-21
| | | | PiperOrigin-RevId: 209605597
* Support Predict mode in PER_HOST_V2 input pipeline.Gravatar Youlong Cheng2018-08-20
| | | | PiperOrigin-RevId: 209507825
* Ensure models can be re-run with Keras (useful for Colab).Gravatar Russell Power2018-08-16
| | | | | | Adjust TPUDistributionStrategy API to remove duplicate master address. PiperOrigin-RevId: 209058959
* [Keras / Cloud TPU] Fix ClusterSpec propagation bug.Gravatar Brennan Saeta2018-08-16
| | | | PiperOrigin-RevId: 209010536
* Allows feature inputs to be arbitrary nested tuples in TPUEstimator which is ↵Gravatar Olivia Nordquist2018-08-15
| | | | | | | allowed behavior in tf.Estimator NB: Calls data/util/nest.flatten() which is slightly different than core/util/nest (already a dependency in tpu_estimator.py) as it maintains structure by treating tuples and lists differently PiperOrigin-RevId: 208848267
* Re-enable some tests on WindowsGravatar A. Unique TensorFlower2018-08-14
| | | | PiperOrigin-RevId: 208695032
* Upgrade cloud tpu profiler to 1.10.Gravatar A. Unique TensorFlower2018-08-13
| | | | PiperOrigin-RevId: 208491690
* Fix estimator dependencies in contrib/.Gravatar Michael Case2018-08-09
| | | | PiperOrigin-RevId: 208126204
* Remove usage of magic-api-link syntax from source files.Gravatar Mark Daoust2018-08-09
| | | | | | | | | | | | | | | | | | | | Back-ticks are now converted to links in the api_docs generator. With the new docs repo we're moving to simplify the docs pipeline, and make everything more readable. By doing this we no longer get test failures for symbols that don't exist (`tf.does_not_exist` will not get a link). There is also no way, not to set custom link text. That's okay. This is the result of the following regex replacement (+ a couple of manual edits.): re: @\{([^$].*?)(\$.+?)?} sub: `\1` Which does the following replacements: "@{tf.symbol}" --> "`tf.symbol`" "@{tf.symbol$link_text}" --> "`tf.symbol`" PiperOrigin-RevId: 208042358
* Add support to tell if we are currently under tpu.rewrite_for_inferece() ↵Gravatar A. Unique TensorFlower2018-08-08
| | | | | | context. PiperOrigin-RevId: 207996119
* Preliminary work for auto-detecting cores and hosts for TPU Distribution ↵Gravatar Frank Chen2018-08-07
| | | | | | Strategy PiperOrigin-RevId: 207802033
* Enable infeed enqueue, outfeed dequeue operations.Gravatar Russell Power2018-08-07
| | | | PiperOrigin-RevId: 207780309
* Refactor dependencies so keras_support can be imported directly.Gravatar Russell Power2018-08-06
| | | | PiperOrigin-RevId: 207569055
* PUBLIC: PREDICT mode should respect ctx.device_assignment.Gravatar Youlong Cheng2018-08-03
| | | | PiperOrigin-RevId: 207326276
* Fix syntax errorGravatar A. Unique TensorFlower2018-08-01
| | | | PiperOrigin-RevId: 206997688
* Add monitoring capability to capture_tpu_profile.Gravatar A. Unique TensorFlower2018-08-01
| | | | PiperOrigin-RevId: 206972475
* Adjust `tpu.rewrite` comment to make it clear that returned operations will ↵Gravatar Russell Power2018-08-01
| | | | | | be executed as part of the program. PiperOrigin-RevId: 206956131
* PUBLIC: Support Input Partition in TPUEstimator.Gravatar Youlong Cheng2018-07-31
| | | | | | RELNOTES: n/a PiperOrigin-RevId: 206865154
* PUBLIC: Remove old comment.Gravatar Youlong Cheng2018-07-31
| | | | PiperOrigin-RevId: 206844376
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-25 23:44:23 +0000