Searched defs:_gather_saveables_for_checkpoint (Results 1 – 16 of 16) sorted by relevance
85 def _gather_saveables_for_checkpoint(self): member in PythonState
53 def _gather_saveables_for_checkpoint(self): member in _TrivialRestore
197 def _gather_saveables_for_checkpoint(self): member in _OwnsMirroredVariables
1010 def _gather_saveables_for_checkpoint(self): member in Trackable
81 def _gather_saveables_for_checkpoint(self): member in ParallelVariable
149 def _gather_saveables_for_checkpoint(self): member in QuantileAccumulator253 def _gather_saveables_for_checkpoint(self): member in TreeEnsemble
1948 def _gather_saveables_for_checkpoint(self): member in MutableHashTable2249 def _gather_saveables_for_checkpoint(self): member in DenseHashTable
1182 def _gather_saveables_for_checkpoint(self): member in Variable
496 def _gather_saveables_for_checkpoint(self): member in Iterator813 def _gather_saveables_for_checkpoint(self): member in OwnedIterator
536 def _gather_saveables_for_checkpoint(self): member in ShardedVariableMixin
197 def _gather_saveables_for_checkpoint(self): member in AggregatingVariable
832 def _gather_saveables_for_checkpoint(self): member in DistributedVariable1074 def _gather_saveables_for_checkpoint(self): member in MirroredVariable1267 def _gather_saveables_for_checkpoint(self): member in SyncOnReadVariable
2984 def _gather_saveables_for_checkpoint(self): member in _OwnsAVariableSimple3019 def _gather_saveables_for_checkpoint(self): member in _OwnsMirroredVariables
879 def _gather_saveables_for_checkpoint( member in TPUEmbedding
350 def _gather_saveables_for_checkpoint(self): member in AutoCastVariable
170 def _gather_saveables_for_checkpoint(self): member in _DelegatingTrackableMixin