Home
last modified time | relevance | path

Searched refs:sharded_filename (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dsaver.py186 def sharded_filename(self, filename_tensor, shard, num_shards): member in BaseSaverBuilder
197 return gen_io_ops.sharded_filename(filename_tensor, shard, num_shards)
265 sharded_filename = self.sharded_filename(tmp_checkpoint_prefix, shard,
267 sharded_prefixes.append(sharded_filename)
268 sharded_saves.append(self._AddSaveOps(sharded_filename, saveables))
301 sharded_filename = self.sharded_filename(filename_tensor, shard,
303 sharded_saves.append(self._AddSaveOps(sharded_filename, saveables))
/external/tensorflow/tensorflow/python/kernel_tests/
Dsave_restore_ops_test.py50 gen_io_ops.sharded_filename("foo", 4, 100).eval(),
Dscalar_test.py94 self.check(gen_io_ops.sharded_filename, ('foo', 4, [100]),