Home
last modified time | relevance | path

Searched refs:_track_trackable (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/training/tracking/
Dbase_test.py34 root._track_trackable(leaf, name="leaf")
40 root._track_trackable(duplicate_name_dep, name="leaf")
41 root._track_trackable(duplicate_name_dep, name="leaf", overwrite=True)
Dtracking_test.py40 root._track_trackable(duplicate_name_dep, name="leaf")
44 root._track_trackable(duplicate_name_dep, name="leaf", overwrite=True)
Dbase.py817 return self._track_trackable(new_variable, name=name, overwrite=overwrite)
859 def _track_trackable(self, trackable, name, overwrite=False): member in Trackable
Ddata_structures.py135 trackable._track_trackable( # pylint: disable=protected-access
Dutil.py621 template._track_trackable( # pylint: disable=protected-access
Dutil_test.py426 root._track_trackable(leaf, name=".ATTRIBUTES")
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer.py77 self._track_trackable(self._optimizer, 'base_optimizer')
78 self._track_trackable(self._loss_scale, 'loss_scale')
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py155 …def _track_trackable(self, trackable, name, overwrite=False): # pylint: disable=redefined-outer-n… member in _DelegatingTrackableMixin
156 return self._trackable._track_trackable(trackable, name, overwrite)
547 self._track_trackable(self._loss_scale, 'loss_scale')
558 self._track_trackable(FakeOptimizerForRestoration(self._optimizer),
Dloss_scale_optimizer_test.py642 self._track_trackable(self.my_var, 'my_var')
/external/tensorflow/tensorflow/python/ops/
Dlookup_ops.py177 self._initializer = self._track_trackable(initializer, "_initializer")
727 self._filename = self._track_trackable(
1246 self._initializer = self._track_trackable(initializer, "_initializer")
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dload.py1025 layer._track_trackable(value, name=name)
1035 layer._track_trackable(value, name) # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/keras/
Doptimizer_v1.py753 self._track_trackable(optimizer, name='optimizer')
759 self._track_trackable(self.iterations, name='global_step')
/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/
Drnn_cell_impl.py1121 self._track_trackable(self.cell, name="cell")
1268 self._track_trackable(cell, name="cell-%d" % (cell_number,))
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py317 …self._layer._track_trackable(v, feature_column.name + '/' + part_name) # pylint: disable=protecte…
320 …self._layer._track_trackable(var, feature_column.name + '/' + name) # pylint: disable=protected-a…
376 …self._layer._track_trackable(var, feature_column.name + '/' + name) # pylint: disable=protected-a…
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py819 self._filename = self._track_trackable(
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py777 self._track_trackable(value, name, overwrite=True)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_v1.py381 self._track_trackable(
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py203 self._variant_tracker = self._track_trackable(