Home
last modified time | relevance | path

Searched refs:_trackable_children (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/python/trackable/
Dautotrackable_test.py41 self.assertIs(duplicate_name_dep, root._trackable_children()["leaf"])
46 self.assertEqual(1, len(root._trackable_children()))
48 self.assertIs(root.a, root._trackable_children()["a"])
51 self.assertEqual(0, len(root._trackable_children()))
54 self.assertEqual(1, len(root._trackable_children()))
56 self.assertIs(root.a, root._trackable_children()["a"])
67 self.assertIn("l", a._trackable_children())
68 direct_a_dep = a._trackable_children()["l"]
Ddata_structures_test.py184 self.assertEqual(m.l._trackable_children(), {})
274 self.assertEqual(len(l._trackable_children()), 0) # pylint: disable=protected-access
303 self.assertDictEqual({"4": v}, l._trackable_children())
310 self.assertDictEqual({"4": v, "9": v}, l._trackable_children())
327 return l._trackable_children() # pylint: disable=protected-access
609 v, m._trackable_children()["nt"]._trackable_children()["x"])
640 m._trackable_children()["nt"]._trackable_children().values())
641 self.assertIn("x", m.nt._trackable_children())
642 self.assertIn("0", m.nt._trackable_children())
657 self.assertIn("0", m.unt._trackable_children())
[all …]
Dbase_test.py29 (current_name, current_dependency), = root._trackable_children().items()
36 (current_name, current_dependency), = root._trackable_children().items()
Dbase_delegate.py126 def _trackable_children(self, *args, **kwargs): member in DelegatingTrackableMixin
127 return self._trackable._trackable_children(*args, **kwargs)
Dautotrackable.py80 def _trackable_children(self, save_type=base.SaveType.CHECKPOINT, **kwargs): member in AutoTrackable
83 return super(AutoTrackable, self)._trackable_children(
Dresource.py187 def _trackable_children(self, save_type, **kwargs): member in CapturableResource
188 children = super()._trackable_children(save_type, **kwargs)
Ddata_structures.py555 def _trackable_children(self, save_type=base.SaveType.CHECKPOINT, **kwargs): member in ListWrapper
576 children = super()._trackable_children(save_type, **kwargs)
853 def _trackable_children(self, save_type=base.SaveType.CHECKPOINT, **kwargs): member in _DictWrapper
875 children = super()._trackable_children(save_type, **kwargs)
1070 def _trackable_children(self, save_type=base.SaveType.CHECKPOINT, **kwargs): member in _TupleWrapper
1078 return super()._trackable_children(save_type, **kwargs)
Dbase.py940 def _trackable_children(self, member in Trackable
/external/tensorflow/tensorflow/python/checkpoint/
Dcheckpoint_with_v1_optimizers_test.py225 manual_scope._trackable_children())
245 self.assertEqual(3, len(load_template._trackable_children()))
247 load_template._trackable_children().keys())
Dtrackable_view.py84 for name, ref in obj._trackable_children(save_type, **kwargs).items():
Dutil.py46 if slot_variable._trackable_children(): # pylint: disable=protected-access
Dcheckpoint_test.py1215 outer_template_dependencies = load_root.my_template._trackable_children()
1219 self.assertLen(inner_template_one._trackable_children(), 1)
1220 self.assertIn("v", inner_template_one._trackable_children())
1221 self.assertLen(inner_template_two._trackable_children(), 1)
1222 self.assertIn("v", inner_template_two._trackable_children())
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale.py158 def _trackable_children(self, member in LossScale
172 super(LossScale, self)._trackable_children(save_type, **kwargs))
/external/tensorflow/tensorflow/python/saved_model/
Dsignature_serialization.py258 def _trackable_children(self, save_type=base.SaveType.CHECKPOINT, **kwargs): member in _SignatureMap
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py156 def _trackable_children(self, member in _DynamicLossScaleState
171 self)._trackable_children(save_type, **kwargs))
/external/tensorflow/tensorflow/python/training/
Doptimizer.py936 def _trackable_children(self, member in Optimizer
948 super(Optimizer, self)._trackable_children(save_type, **kwargs))
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable_test.py519 self.assertLen(model._trackable_children(), 1)
520 self.assertIs(model._trackable_children().popitem()[1], model.w)
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Drnn_cell_test.py241 self.assertIn("cell", full_dropout_cell._trackable_children())
242 self.assertIs(full_dropout_cell._trackable_children()["cell"], cell)
2669 self.assertIn("cell-0", cell._trackable_children())
2670 self.assertIn("cell-1", cell._trackable_children())
2917 self.assertIn("cell", wrapper_object._trackable_children())
2918 self.assertIs(wrapper_object._trackable_children()["cell"], base_cell)
2959 self.assertEqual(wrapped_cell._trackable_children()["cell"], cell)
/external/tensorflow/tensorflow/python/keras/engine/
Dfunctional.py365 def _trackable_children(self, member in Functional
370 super(Functional, self)._trackable_children(save_type, **kwargs))
Dbase_layer_v1.py2354 def _trackable_children(self, save_type='checkpoint', **kwargs): member in Layer
2363 children.update(super()._trackable_children(save_type, **kwargs))
Dtraining.py2734 def _trackable_children(self, save_type='checkpoint', **kwargs): member in Model
2746 children = super(Model, self)._trackable_children(save_type, **kwargs)
Dbase_layer.py3041 def _trackable_children(self, save_type='checkpoint', **kwargs): member in Layer
3050 children.update(super()._trackable_children(save_type, **kwargs))
/external/tensorflow/tensorflow/python/eager/
Ddef_function.py1168 def _trackable_children(self, save_type="checkpoint", **kwargs): member in Function
/external/tensorflow/tensorflow/python/kernel_tests/
Dtemplate_test.py445 self.assertEqual(["nested", "nested_1"], list(tmpl1._trackable_children()))
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py338 def _trackable_children(self, member in DatasetV2
352 children = super(DatasetV2, self)._trackable_children(save_type, **kwargs)
4677 def _trackable_children(self, member in _VariantTracker
4684 self)._trackable_children(save_type, **kwargs)

12