| /external/tensorflow/tensorflow/contrib/rnn/python/tools/ |
| checkpoint_convert.py | 191 not_sharded: Names of the non-sharded variables.
192 sharded: Names of the sharded variables.
194 sharded = []
199 sharded.append(name)
204 return not_sharded, sharded
225 not_sharded, sharded = _split_sharded_vars(name_shape_map)
235 for var_name in sharded:
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| exporter_test.py | 57 sharded=True,
108 sharded=sharded,
196 if sharded:
223 self.doBasicsOneExportPath(export_path, sharded=False)
|
| /external/tensorflow/tensorflow/python/training/ |
| saver.py | 255 preferred_shard: Int. Shard to open first when loading a sharded file.
283 preferred_shard: Int. Shard to open first when loading a sharded file.
332 Note that the sharded save procedure for the V2 format is different from
344 "<user-fed prefix>" only and does not include the sharded spec suffix.
348 # Suffix for any well-formed "checkpoint_prefix", when sharded.
396 # sharded spec suffix.
421 # Return the sharded name for the save path.
442 preferred_shard: Shard to open first when loading a sharded file.
708 sharded=False,
724 sharded: If True, shard the checkpoints, one per device that ha
[all...] |
| saver_test.py | 553 save = saver_module.Saver({"v0": v0_1}, sharded=True, allow_empty=True)
560 save = saver_module.Saver({"v0": v0_2}, sharded=True, allow_empty=True)
777 sharded=True)
[all...] |
| /external/tensorflow/tensorflow/contrib/session_bundle/example/ |
| export_half_plus_two.py | 68 sharded=True,
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| builder_impl.py | 294 # Initialize a saver to generate a sharded output for all saveables in the
298 sharded=True,
382 # Initialize a saver to generate a sharded output for all saveables in the
386 sharded=True,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| embedding_ops_test.py | 238 # This test looks up [0, 0] in a parameter matrix sharded 2 ways. Since
538 # Run a few random sharded versions
544 sharded = embedding_ops.embedding_lookup(split_params, ids).eval()
545 self.assertAllEqual(simple, sharded)
565 # Run a few different sharded versions.
571 sharded = embedding_ops.embedding_lookup(
573 self.assertAllEqual(simple, sharded)
578 # - params sharded/unsharded
600 # Run a few different sharded versions.
606 sharded = embedding_ops._embedding_lookup_and_transform
[all...] |
| /external/tensorflow/tensorflow/python/estimator/ |
| estimator.py | 625 sharded=True)
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| graph_actions.py | 89 sharded=True,
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| evaluation_test.py | 94 saver = saver_lib.Saver(sharded=True)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| estimator.py | [all...] |
| /prebuilts/go/darwin-x86/src/cmd/dist/ |
| test.go | 470 // tests, but we should be tolerant of the non-sharded
[all...] |
| /prebuilts/go/linux-x86/src/cmd/dist/ |
| test.go | 470 // tests, but we should be tolerant of the non-sharded
[all...] |
