Searched refs:TPUDistributedVariable (Results 1 – 3 of 3) sorted by relevance
462 "VariableClass": tpu_values_lib.TPUDistributedVariable,
153 class TPUDistributedVariable(TPUVariableMixin, values.DistributedVariable): class
620 self.assertIsInstance(variable, tpu_values.TPUDistributedVariable)