Searched refs:NoRestoreSaveable (Results 1 – 5 of 5) sorted by relevance
24 class SaveableHook(base.NoRestoreSaveable):
420 if any(isinstance(saveable, trackable.NoRestoreSaveable)
131 class NoRestoreSaveable(saveable_object.SaveableObject): class137 super(NoRestoreSaveable, self).__init__(tensor, [spec], name)215 return NoRestoreSaveable(
429 base.NoRestoreSaveable(
1130 base.NoRestoreSaveable(