Searched refs:TPUMirroredVariable (Results 1 – 5 of 5) sorted by relevance
104 result = values.TPUMirroredVariable(433 assert not isinstance(v, values.TPUMirroredVariable)480 assert isinstance(var, values.TPUMirroredVariable)500 assert isinstance(var, values.TPUMirroredVariable)512 elif isinstance(val, values.TPUMirroredVariable):
608 if not isinstance(v, TPUMirroredVariable):770 class TPUMirroredVariable(trackable.Trackable): class1189 ops.register_tensor_conversion_function(TPUMirroredVariable,1191 ops.register_dense_tensor_like_type(TPUMirroredVariable)
61 value_lib.TPUMirroredVariable,
665 distributed_values.TPUMirroredVariable,
526 self.assertIsInstance(mirrored, values.TPUMirroredVariable)