aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/model_pruning
Commit message (Collapse)AuthorAge
* Add missing documentation for use_tpu hparamGravatar Suyog Gupta2018-10-02
| | | | PiperOrigin-RevId: 215462000
* Updating the V2 variables API.Gravatar Alexandre Passos2018-09-27
| | | | PiperOrigin-RevId: 214824023
* Move from deprecated self.test_session() to self.cached_session().Gravatar A. Unique TensorFlower2018-09-10
| | | | | | | | self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about: * the fact that the session may be reused. * the session is not closed even when doing a "with self.test_session()" statement. PiperOrigin-RevId: 212336321
* Bump test size for pruning_utils_test to avoid flaky timeouts.Gravatar Gunhan Gulsoy2018-08-29
| | | | PiperOrigin-RevId: 210857377
* Use nbins as given in hparams when pruning on TPUsGravatar Suyog Gupta2018-08-27
| | | | PiperOrigin-RevId: 210461150
* Move from deprecated self.test_session() to self.cached_session().Gravatar A. Unique TensorFlower2018-08-21
| | | | | | | | self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about: * the fact that the session may be reused. * the session is not closed even when doing a "with self.test_session()" statement. PiperOrigin-RevId: 209701635
* Move from deprecated self.test_session() to self.session() when a graph is set.Gravatar A. Unique TensorFlower2018-08-21
| | | | | | self.test_session() has been deprecated in cl/208545396 as its behavior confuses readers of the test. Moving to self.session() instead. PiperOrigin-RevId: 209696110
* Optimization: implementation of mask expansion for block sparsity that does ↵Gravatar Suyog Gupta2018-08-20
| | | | | | not depend on kronecker product PiperOrigin-RevId: 209432310
* 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
* Merge pull request #21086 from taehoonlee:fix_typosGravatar TensorFlower Gardener2018-08-08
|\ | | | | | | PiperOrigin-RevId: 207988541
* | Add helper function for validating the user-provided pruning hparamsGravatar Suyog Gupta2018-08-08
| | | | | | | | PiperOrigin-RevId: 207946581
* | Add utility function to pruning library to strip a trained graph of ↵Gravatar Suyog Gupta2018-08-08
| | | | | | | | | | | | pruning-related variables PiperOrigin-RevId: 207902316
* | Add support for layer-dependent sparsity. Accept layer_name:target_sparsity ↵Gravatar Suyog Gupta2018-07-31
| | | | | | | | | | | | | | | | mapping as hyperparameter Deprecate do_not_prune hyperparameter PiperOrigin-RevId: 206851318
| * Fix typosGravatar Taehoon Lee2018-07-24
|/
* Merge pull request #19779 from Huizerd:masterGravatar TensorFlower Gardener2018-07-19
|\ | | | | | | PiperOrigin-RevId: 205266716
* | Update documentation for using pruning and contrib/slim training utilityGravatar A. Unique TensorFlower2018-07-17
| | | | | | | | PiperOrigin-RevId: 205027982
| * Fix lint error.Gravatar Mark Daoust2018-07-02
| |
| * Removed redundant use of enumerationGravatar Jesse2018-06-06
| | | | | | Since every mask has an accompanying threshold, zip(masks, thresholds) can be used instead of enumerate(masks) and calling thresholds by index.
| * Prevent redundant ":0" in summary namesGravatar Jesse2018-06-05
| | | | | | Take identical approach as is done with thresholds: using tf.Variable.op.name instead of tf.Variable.name, to prevent TensorFlow saying summary names are illegal (due to ":")
| * Put some emphasis on incrementing global stepGravatar Jesse2018-06-05
| | | | | | Pruning will not work if the global step is not incremented
| * Updated line for creating global step + grammarGravatar Jesse2018-06-05
|/ | | tf.train.get_global_step() returns None if there is no global step, preventing the pruning from working. Therefore, tf.train.get_or_create_global_step() is a safer option.
* Extend block sparsity support for TPUsGravatar A. Unique TensorFlower2018-05-07
| | | | PiperOrigin-RevId: 195685740
* -- Add a new histogram/cdf computation method compatible with the TPU.Gravatar A. Unique TensorFlower2018-04-13
| | | | | | -- Refactor utility functions into pruning_utils.py and add tests PiperOrigin-RevId: 192727737
* Remove all_opensource_files. It's not needed any more.Gravatar Martin Wicke2018-03-28
| | | | PiperOrigin-RevId: 190878279
* Merge changes from github.Gravatar Jacques Pienaar2018-03-21
| | | | PiperOrigin-RevId: 189945839
* Make block-based pruning more general, allowing it to operate on ↵Gravatar A. Unique TensorFlower2018-02-27
| | | | | | higher-dimensional arrays that can be squeezed to 2-dimensional. PiperOrigin-RevId: 187195105
* Add pylint check for W0611 unused-import in ci_sanity.sh and fix existing ↵Gravatar Yifei Feng2018-02-06
| | | | | | pylint errors. PiperOrigin-RevId: 184790548
* Add C0326 bad-whitespace error to pylint sanity check.Gravatar Yifei Feng2018-01-29
| | | | PiperOrigin-RevId: 183689499
* Add C0301 line-too-long error to pylint sanity check.Gravatar Yifei Feng2018-01-26
| | | | PiperOrigin-RevId: 183467186
* Add C0330 bad-continuation check to pylint.Gravatar Yifei Feng2018-01-25
| | | | PiperOrigin-RevId: 183270896
* Make compute_output_shape public in masked core layersGravatar A. Unique TensorFlower2018-01-05
| | | | PiperOrigin-RevId: 180988293
* Add block sparsity support for 2D weight tensors only.Gravatar A. Unique TensorFlower2017-12-14
| | | | PiperOrigin-RevId: 179130257
* Merge changes from github.Gravatar Shanqing Cai2017-12-06
| | | | PiperOrigin-RevId: 178185697
* Update the pruning library to handle graphs which has both partitioned and ↵Gravatar A. Unique TensorFlower2017-12-03
| | | | | | non-partitioned variables PiperOrigin-RevId: 177761638
* Further BUILD cleanup in contrib/...Gravatar A. Unique TensorFlower2017-11-10
| | | | PiperOrigin-RevId: 175370768
* Fix documentation for contrib/model_pruningGravatar A. Unique TensorFlower2017-11-07
| | | | PiperOrigin-RevId: 174907982
* Convert BasicRNNCell and GRUCell to proper layers.Gravatar Eugene Brevdo2017-11-01
| | | | PiperOrigin-RevId: 174272860
* Bug fix: Expose get_pruning_hparams functionGravatar A. Unique TensorFlower2017-11-01
| | | | PiperOrigin-RevId: 174260120
* Moving model_pruning library to tf.contribGravatar A. Unique TensorFlower2017-11-01
PiperOrigin-RevId: 174214419