Searched refs:variable_specs (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/federated-compute/fcp/artifact_building/ |
| H A D | checkpoint_utils_test.py | 344 variable_specs = tf.train.list_variables(test_filepath)
345 self.assertEqual([('x', []), ('y', []), ('z', [])], variable_specs)
359 variable_specs = tf.train.list_variables(test_filepath)
360 self.assertEqual([('bar', []), ('baz', []), ('foo', [])], variable_specs)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | sharded_variable.py | 237 def __init__(self, *variable_specs): argument
238 self._variable_specs = tuple(variable_specs)
|