Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/training/tracking/
Dtracking_test.py49 (_, dep_object), = root._checkpoint_dependencies
58 self.assertEqual(1, len(root._checkpoint_dependencies))
59 self.assertIs(root._checkpoint_dependencies[0].ref, root.hasdep)
77 self.assertEqual(1, len(root._checkpoint_dependencies))
79 self.assertIs(root.a, root._checkpoint_dependencies[0].ref)
82 self.assertEqual(0, len(root._checkpoint_dependencies))
85 self.assertEqual(1, len(root._checkpoint_dependencies))
87 self.assertIs(root.a, root._checkpoint_dependencies[0].ref)
98 direct_a_dep, = a._checkpoint_dependencies
177 [dep.name for dep in a.attribute._checkpoint_dependencies])
Ddata_structures_test.py89 self.assertEqual(2, len(model._checkpoint_dependencies))
90 self.assertIs(model.layer_list, model._checkpoint_dependencies[0].ref)
92 model._checkpoint_dependencies[1].ref)
94 10, len(model._checkpoint_dependencies[0].ref._checkpoint_dependencies))
368 self.assertEqual(len(l._checkpoint_dependencies), 0) # pylint: disable=protected-access
401 return l._checkpoint_dependencies # pylint: disable=protected-access
437 self.assertEqual(1, len(model._checkpoint_dependencies))
438 self.assertIs(model.layer_dict, model._checkpoint_dependencies[0].ref)
Dbase_test.py35 (current_name, current_dependency), = root._checkpoint_dependencies
42 (current_name, current_dependency), = root._checkpoint_dependencies
Ddata_structures.py422 def _checkpoint_dependencies(self): member in _ListWrapper
444 return super(_ListWrapper, self)._checkpoint_dependencies
681 def _checkpoint_dependencies(self): member in _DictWrapper
714 return super(_DictWrapper, self)._checkpoint_dependencies
Dutil_test.py1331 manual_dep, = manual_scope._checkpoint_dependencies
1354 self.assertLen(load_template._checkpoint_dependencies, 3)
1355 self.assertEqual("v", load_template._checkpoint_dependencies[0].name)
1356 self.assertEqual("v2", load_template._checkpoint_dependencies[1].name)
1358 load_template._checkpoint_dependencies[2].name)
1394 outer_template_dependencies = load_root.my_template._checkpoint_dependencies
1400 self.assertLen(inner_template_one._checkpoint_dependencies, 1)
1401 self.assertEqual("v", inner_template_one._checkpoint_dependencies[0].name)
1402 self.assertLen(inner_template_two._checkpoint_dependencies, 1)
1403 self.assertEqual("v", inner_template_two._checkpoint_dependencies[0].name)
Dgraph_view.py111 if slot_variable._checkpoint_dependencies: # pylint: disable=protected-access
159 return obj._checkpoint_dependencies
Dutil_with_v1_optimizers_test.py769 manual_dep, = manual_scope._checkpoint_dependencies
791 self.assertEqual(3, len(load_template._checkpoint_dependencies))
792 self.assertEqual("v", load_template._checkpoint_dependencies[0].name)
793 self.assertEqual("v2", load_template._checkpoint_dependencies[1].name)
795 load_template._checkpoint_dependencies[2].name)
Dbase.py528 def _checkpoint_dependencies(self): member in Trackable
Dutil.py593 not trackable_object._checkpoint_dependencies):
/external/tensorflow/tensorflow/contrib/checkpoint/python/
Dcontainers.py51 return [dep.ref for dep in self._checkpoint_dependencies]
/external/tensorflow/tensorflow/python/kernel_tests/
Dtemplate_test.py379 self.assertEqual(2, len(tmpl1._checkpoint_dependencies))
380 self.assertEqual("nested", tmpl1._checkpoint_dependencies[0].name)
381 self.assertEqual("nested_1", tmpl1._checkpoint_dependencies[1].name)
Drnn_cell_test.py241 (name, dep), = full_dropout_cell._checkpoint_dependencies
2578 self.assertEqual("cell-0", cell._checkpoint_dependencies[0].name)
2579 self.assertEqual("cell-1", cell._checkpoint_dependencies[1].name)
2826 (name, dep), = wrapper_object._checkpoint_dependencies
2871 (name, dep), = wrapped_cell._checkpoint_dependencies
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Dcheckpointable_utils_test.py628 self.assertEqual(2, len(load_template._checkpoint_dependencies))
629 self.assertEqual("v", load_template._checkpoint_dependencies[0].name)
630 self.assertEqual("v2", load_template._checkpoint_dependencies[1].name)
Doptimizer_v2.py1272 def _checkpoint_dependencies(self): member in OptimizerV2
/external/tensorflow/tensorflow/python/training/
Doptimizer.py823 def _checkpoint_dependencies(self): member in Optimizer
834 return (super(Optimizer, self)._checkpoint_dependencies
/external/tensorflow/tensorflow/python/keras/
Dmodel_subclassing_test.py469 self.assertEqual(1, len(m._checkpoint_dependencies))
470 self.assertIs(m.isdep, m._checkpoint_dependencies[0].ref)
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_test.py506 set([obj for unused_name, obj in layer._checkpoint_dependencies]))
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Drnn_cell_test.py139 self.assertEqual("cell-0", cell._checkpoint_dependencies[0].name)
140 self.assertEqual("cell-1", cell._checkpoint_dependencies[1].name)