experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
nightly
STT-tensorflow/tensorflow/compiler/xla/experimental/xla_sharding
History
A. Unique TensorFlower 128b50a574 Adds get_sharding_tile_shape() function. 
Adds many unit tests for xla_sharding.Sharding and helper functions, making use of the new get_sharding_tile_shape() function.

PiperOrigin-RevId: 361245107
Change-Id: I85ee889c9f2f59803cf59a5c2972fc3e75d3f037
2021-03-05 16:20:35 -08:00
..
BUILD Include TF directly in xla_sharding_test BUILD 2021-03-05 14:20:53 -08:00
xla_sharding_test.py Adds get_sharding_tile_shape() function. 2021-03-05 16:20:35 -08:00
xla_sharding.py Adds get_sharding_tile_shape() function. 2021-03-05 16:20:35 -08:00