Home
last modified time | relevance | path

Searched refs:ShardInfo (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy_v2.py658 shard_info=trackable.ShardInfo(
/external/tensorflow/tensorflow/python/training/tracking/
Dbase.py59 ShardInfo = collections.namedtuple( variable
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2.py1745 shard_info = base.ShardInfo(kwargs["shape"], (offset, 0))