Home
last modified time | relevance | path

Searched refs:sharded (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/python/data/experimental/ops/
Diterator_ops.py240 saveables, self._latest_filename, sharded=True)
304 def __init__(self, var_list, latest_filename, sharded=False): argument
305 super(_CustomSaver, self).__init__(var_list, sharded=sharded)
/external/tensorflow/tensorflow/python/training/
Dsaver.py422 sharded=False, argument
465 sharded=sharded,
475 sharded=False, argument
504 if sharded:
537 sharded=sharded,
562 sharded=sharded,
607 saver = Saver(sharded=True, allow_empty=True)
692 sharded=False, argument
817 self._sharded = sharded
879 sharded=self._sharded,
Dcheckpoint_management_test.py82 saver = saver_module.Saver(sharded=False)
92 saver = saver_module.Saver(sharded=True)
98 saver = saver_module.Saver(sharded=True)
277 for sharded in (False, True):
282 saver = saver_module.Saver(sharded=sharded, write_version=version)
284 path = os.path.join(self._base_dir, "%s-%s" % (sharded, version))
311 for sharded in (False, True):
316 saver = saver_module.Saver(sharded=sharded, write_version=version)
318 path = os.path.join(self._base_dir, "%s-%s" % (sharded, version))
Dsaver_test.py600 save = saver_module.Saver({"v0": v0_1}, sharded=True, allow_empty=True)
607 save = saver_module.Saver({"v0": v0_2}, sharded=True, allow_empty=True)
882 sharded=True)
908 sharded=True)
932 sharded=True)
961 sharded=True)
996 save = saver_module.Saver({"v0": v0}, sharded=True)
998 self.assertTrue(sd.sharded)
1135 sharded=True)
1168 sharded=True)
[all …]
Dcheckpoint_utils_test.py444 saver = saver_lib.Saver(sharded=True)
Dmonitored_session.py241 sharded=True)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_MergeV2Checkpoints.pbtxt22 summary: "V2 format specific: merges the metadata files of sharded checkpoints. The"
27 Intended for "grouping" multiple checkpoints in a sharded checkpoint setup.
Dapi_def_ShardedFilespec.pbtxt3 summary: "Generate a glob pattern matching all sharded file names."
Dapi_def_ShardedFilename.pbtxt3 summary: "Generate a sharded filename. The filename is printf formatted as"
Dapi_def_TPUPartitionedOutput.pbtxt21 summary: "An op that demultiplexes a tensor to be sharded by XLA to a list of partitioned"
Dapi_def_ExperimentalAutoShardDataset.pbtxt25 sharded dataset for the index-th worker. This attempts to automatically shard
Dapi_def_AutoShardDataset.pbtxt25 sharded dataset for the index-th worker. This attempts to automatically shard
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-saver-def.pbtxt30 name: "sharded"
Dtensorflow.train.-saver.pbtxt11 …argspec: "args=[\'self\', \'var_list\', \'reshape\', \'sharded\', \'max_to_keep\', \'keep_checkpoi…
/external/tensorflow/tensorflow/core/protobuf/
Dsaver.proto27 bool sharded = 5; field
Dmeta_graph.proto333 // An asset file def for a single file or a set of sharded files with the same
/external/tensorflow/tensorflow/python/kernel_tests/
Dembedding_ops_test.py567 sharded = embedding_ops.embedding_lookup(split_params, ids)
568 self.assertAllEqual(simple, sharded)
600 sharded = embedding_ops.embedding_lookup(
602 self.assertAllEqual(simple, sharded)
636 sharded = embedding_ops._embedding_lookup_and_transform(
644 self.assertAllClose(simple, sharded)
/external/rust/crates/crossbeam-utils/
DREADME.md25 * [`ShardedLock`], a sharded reader-writer lock with fast concurrent reads.
/external/tensorflow/tensorflow/tools/ci_build/
DREADME.md66 If you’re looking at target that is sharded or a test that is flaky, then
/external/tensorflow/tensorflow/python/saved_model/
Dbuilder_impl.py225 sharded=True,
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto640 // This op is manually sharded: the shapes are already partitioned and the
645 // The shape of the sharded tile.
662 // Only used for OTHER type. If true, data is sharded according to other
/external/libaom/libaom/
DREADME.md27 - [Sharded testing](#sharded-testing)
504 # supports building targets concurrently tests will be sharded by default.
/external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/
Dsaved_model.pbtxt2457 sharded: true
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_rewrite.mlir1214 // Test `tf_device.cluster_func` on TPU with pre-split replicate sharded
1252 // Test `tf_device.cluster_func` on TPU with pre-split tile sharded input/
2247 // Tests device assignment is well preserved for tile sharded outputs.
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td2092 An op that demultiplexes a tensor to be sharded by XLA to a list of partitioned

12