experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
587ac71f68
STT-tensorflow/tensorflow/tools
History
Peng Wang 587ac71f68 Allows creating tf.random.Generator under distribution-strategy scopes. Different replicas will get different random-number streams. 
All strategies are supported except for CentralStorageStrategy and ParameterServerStrategy.

This CL also removes the CompositeTensor superclass from Generator. Generator is a wrapper around tf.Variable, and because tf.Variable is not a CompositeTensor, Generator can't be a CompositeTensor in theory. Previously we made it a CompositeTensor by returning Variable.handle, but that breaks down when the variable is a DistributedVariable (in cross-replica context).

PiperOrigin-RevId: 350851648
Change-Id: I5f4d77ddb990557fcc9c7336987203ecdaec5b9a
2021-01-08 15:58:22 -08:00
..
android Move deprecated example. 2020-09-29 07:19:05 -07:00
api Allows creating tf.random.Generator under distribution-strategy scopes. Different replicas will get different random-number streams. 2021-01-08 15:58:22 -08:00
benchmark Update references from android_tensorflow_lib to portable_tensorflow_lib. 2020-10-06 08:26:04 -07:00
build_info [TF] Add TensorRT version to tf.sysconfig.get_build_info. 2020-11-10 11:45:26 -08:00
ci_build Create a new ROCM RBE Docker image with ROCM 4.0. 2021-01-08 00:09:30 -08:00
common Add python/lib BUILD and refactor python/BUILD 2020-12-10 12:42:17 -08:00
compatibility Removing no_rocm tag from unit-tests that are now passing on the ROCm platform 2020-11-25 02:06:27 +00:00
def_file_filter add BUILD file for python/util and refactor python/BUILD 2020-12-15 11:43:18 -08:00
dockerfiles [Dockerfile] move stubs to the end of LD_LIBRARY_PATH 2021-01-05 16:00:53 -08:00
docs add BUILD file for python/util and refactor python/BUILD 2020-12-15 11:43:18 -08:00
gcs_test
git
graph_transforms add BUILD file for python/util and refactor python/BUILD 2020-12-15 11:43:18 -08:00
lib_package Delete SYCL support 2020-09-15 11:21:47 -07:00
mlpbtxt BUILD file cleanup 2020-09-30 16:08:07 -07:00
optimization BUILD file cleanup 2020-09-30 16:08:07 -07:00
pip_package Upgrade to gast 0.4. 2021-01-06 10:17:47 -08:00
proto_text Optimize calls to std::string::find() and friends for a single char. 2020-12-17 18:11:13 -08:00
swig
tensorflow_builder
test Remove do_check_futures_test from sanity CI 2020-11-05 10:53:12 -08:00
__init__.py