Searched refs:TPUShardedVariable (Results 1 – 3 of 3) sorted by relevance
71 class TPUShardedVariable(sharded_variable.ShardedVariableMixin): class1700 hosts: List[Text]) -> Callable[..., TPUShardedVariable]:1753 return TPUShardedVariable(variables, name=name)
644 if isinstance(variable, tpu_embedding_v2.TPUShardedVariable):
1315 if isinstance(variable, tpu_embedding_v2.TPUShardedVariable):