Searched refs:BundleWriter (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle_test.cc | 168 BundleWriter writer(Env::Default(), Prefix("foo")); in TestBasic()
198 BundleWriter writer(Env::Default(), Prefix("bar")); in TestBasic()
265 BundleWriter writer(Env::Default(), Prefix("nonstandard")); in TestNonStandardShapes()
331 BundleWriter writer(Env::Default(), Prefix("foo")); in TEST()
404 BundleWriter writer(Env::Default(), Prefix("foo")); in TEST()
487 BundleWriter writer(Env::Default(), Prefix("foo")); in TEST()
581 BundleWriter writer(Env::Default(), Prefix("foo")); in TEST()
604 BundleWriter writer(env, kBundlePrefixes[i]); in TEST()
646 BundleWriter writer(Env::Default(), Prefix("dup")); in TEST()
654 BundleWriter writer(Env::Default(), Prefix("bad")); in TEST()
[all …]
|
| D | tensor_bundle.h | 108 class BundleWriter {
116 BundleWriter(Env* env, StringPiece prefix,
159 TF_DISALLOW_COPY_AND_ASSIGN(BundleWriter);
|
| D | tensor_bundle.cc | 388 BundleWriter::BundleWriter(Env* env, StringPiece prefix, const Options& options) in BundleWriter() function in tensorflow::BundleWriter
412 Status BundleWriter::Add(StringPiece key, const Tensor& val) { in Add()
449 Status BundleWriter::AddSlice(StringPiece full_tensor_key, in AddSlice()
492 Status BundleWriter::Finish() { in Finish()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | merge_v2_checkpoints_op_test.cc | 38 BundleWriter writer(Env::Default(), prefix); in WriteCheckpoint()
|
| D | save_restore_v2_ops.cc | 108 BundleWriter writer(Env::Default(), prefix_string); in Compute()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 129 BundleWriter writer(Env::Default(), checkpoint_path); in TestSinglePartition()
366 BundleWriter writer(Env::Default(), checkpoint_path); in TestMultiPartition()
580 BundleWriter writer(Env::Default(), checkpoint_path); in TestReadTensorSlice()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | cache_dataset_ops.cc | 328 writer_ = absl::make_unique<BundleWriter>(dataset()->env_, filename_); in RestoreInternal()
389 absl::make_unique<BundleWriter>(dataset()->env_, filename_); in EnsureLockFileExists()
433 std::unique_ptr<BundleWriter> writer_ GUARDED_BY(mu_);
|