Home
last modified time | relevance | path

Searched refs:ShardedVariableMixin (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py251 class ShardedVariableMixin(trackable.Trackable): class
279 super(ShardedVariableMixin, self).__init__()
571 class ShardedVariable(ShardedVariableMixin, composite_tensor.CompositeTensor):
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2.py71 class TPUShardedVariable(sharded_variable.ShardedVariableMixin):