Chenkai Kuang 239fe406d3 Modify some v2 initializers to be able to return a value that corresponds to a partition of the entire value. This is useful for efficiently initializing sharded variables where only a shard of the initial value is necessary at a time. ...

PiperOrigin-RevId: 338371904
Change-Id: Ib4320d73cbaec30f5a61793debe7755026175781

2020-10-21 18:20:24 -07:00

..

__init__.py

Fork the tf_inspect to keras, which use tf_decorator.unwrap as __internal__ API.

2020-09-14 17:26:35 -07:00

initializers_v1.py

…

initializers_v2.py

Modify some v2 initializers to be able to return a value that corresponds to a partition of the entire value. This is useful for efficiently initializing sharded variables where only a shard of the initial value is necessary at a time.

2020-10-21 18:20:24 -07:00