experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
cherrypick_339370490
STT-tensorflow/tensorflow/python/module
History
Chenkai Kuang bb41d04dd0 Add expand_composite option to tf.Module._flatten to support expanding composite tensors. 
Use the said `expand_composite` option in `tf.Module.variables` like APIs to support expanding composite tensors that are a collection of variables.

PiperOrigin-RevId: 338300859
Change-Id: Iddcb1f34a87557e9de15d1887fed6d3b61319301
2020-10-21 11:47:00 -07:00
..
BUILD Add expand_composite option to tf.Module._flatten to support expanding composite tensors. 2020-10-21 11:47:00 -07:00
module_test.py Add expand_composite option to tf.Module._flatten to support expanding composite tensors. 2020-10-21 11:47:00 -07:00
module.py Add expand_composite option to tf.Module._flatten to support expanding composite tensors. 2020-10-21 11:47:00 -07:00