| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | tracking_test.py | 72 (_, dep_object), = root._checkpoint_dependencies
81 self.assertEqual(1, len(root._checkpoint_dependencies))
82 self.assertIs(root._checkpoint_dependencies[0].ref, root.hasdep)
100 self.assertEqual(1, len(root._checkpoint_dependencies))
102 self.assertIs(root.a, root._checkpoint_dependencies[0].ref)
105 self.assertEqual(0, len(root._checkpoint_dependencies))
108 self.assertEqual(1, len(root._checkpoint_dependencies))
110 self.assertIs(root.a, root._checkpoint_dependencies[0].ref)
121 direct_a_dep, = a._checkpoint_dependencies
200 [dep.name for dep in a.attribute._checkpoint_dependencies])
|
| D | data_structures_test.py | 102 self.assertEqual(2, len(model._checkpoint_dependencies))
103 self.assertIs(model.layer_list, model._checkpoint_dependencies[0].ref)
105 model._checkpoint_dependencies[1].ref)
107 10, len(model._checkpoint_dependencies[0].ref._checkpoint_dependencies))
478 self.assertEqual(len(l._checkpoint_dependencies), 0) # pylint: disable=protected-access
507 self.assertEqual([("4", v)], l._checkpoint_dependencies)
514 self.assertEqual([("4", v), ("9", v)], l._checkpoint_dependencies)
531 return l._checkpoint_dependencies # pylint: disable=protected-access
566 self.assertEqual(1, len(model._checkpoint_dependencies))
567 self.assertIs(model.layer_dict, model._checkpoint_dependencies[0].ref)
[all …]
|
| D | base_test.py | 35 (current_name, current_dependency), = root._checkpoint_dependencies
42 (current_name, current_dependency), = root._checkpoint_dependencies
|
| D | data_structures.py | 498 def _checkpoint_dependencies(self): member in ListWrapper
520 return super(ListWrapper, self)._checkpoint_dependencies
775 def _checkpoint_dependencies(self): member in _DictWrapper
799 return super(_DictWrapper, self)._checkpoint_dependencies
994 def _checkpoint_dependencies(self): member in _TupleWrapper
1002 return super(_TupleWrapper, self)._checkpoint_dependencies
|
| D | graph_view.py | 111 if slot_variable._checkpoint_dependencies: # pylint: disable=protected-access
159 return obj._checkpoint_dependencies
|
| D | util_test.py | 1448 manual_dep, = manual_scope._checkpoint_dependencies
1473 self.assertLen(load_template._checkpoint_dependencies, 3)
1474 self.assertEqual("v", load_template._checkpoint_dependencies[0].name)
1475 self.assertEqual("v2", load_template._checkpoint_dependencies[1].name)
1477 load_template._checkpoint_dependencies[2].name)
1513 outer_template_dependencies = load_root.my_template._checkpoint_dependencies
1519 self.assertLen(inner_template_one._checkpoint_dependencies, 1)
1520 self.assertEqual("v", inner_template_one._checkpoint_dependencies[0].name)
1521 self.assertLen(inner_template_two._checkpoint_dependencies, 1)
1522 self.assertEqual("v", inner_template_two._checkpoint_dependencies[0].name)
|
| D | util_with_v1_optimizers_test.py | 765 manual_dep, = manual_scope._checkpoint_dependencies
787 self.assertEqual(3, len(load_template._checkpoint_dependencies))
788 self.assertEqual("v", load_template._checkpoint_dependencies[0].name)
789 self.assertEqual("v2", load_template._checkpoint_dependencies[1].name)
791 load_template._checkpoint_dependencies[2].name)
|
| D | base.py | 643 def _checkpoint_dependencies(self): member in Trackable
|
| D | util.py | 770 not trackable_object._checkpoint_dependencies):
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale.py | 152 def _checkpoint_dependencies(self): member in LossScale
163 return super(LossScale, self)._checkpoint_dependencies + weights
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | rnn_cell_wrapper_v2_test.py | 50 (name, dep), = wrapper_object._checkpoint_dependencies
93 (name, dep), = wrapped_cell._checkpoint_dependencies
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | template_test.py | 390 self.assertEqual(2, len(tmpl1._checkpoint_dependencies))
391 self.assertEqual("nested", tmpl1._checkpoint_dependencies[0].name)
392 self.assertEqual("nested_1", tmpl1._checkpoint_dependencies[1].name)
|
| D | rnn_cell_test.py | 240 (name, dep), = full_dropout_cell._checkpoint_dependencies
2589 self.assertEqual("cell-0", cell._checkpoint_dependencies[0].name)
2590 self.assertEqual("cell-1", cell._checkpoint_dependencies[1].name)
2836 (name, dep), = wrapper_object._checkpoint_dependencies
2879 (name, dep), = wrapped_cell._checkpoint_dependencies
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 846 def _checkpoint_dependencies(self): member in Optimizer
857 return (super(Optimizer, self)._checkpoint_dependencies
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_subclassing_test.py | 369 self.assertEqual(1, len(m._checkpoint_dependencies))
370 self.assertIs(m.isdep, m._checkpoint_dependencies[0].ref)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | network.py | 397 def _checkpoint_dependencies(self): member in Network
401 dependencies.extend(super(Network, self)._checkpoint_dependencies)
|
| D | base_layer_test.py | 807 {id(v) for _, v in layer._checkpoint_dependencies})
|