Searched refs:BundleWriter (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle_test.cc | 187 BundleWriter writer(Env::Default(), Prefix("foo")); in TestBasic()
217 BundleWriter writer(Env::Default(), Prefix("bar")); in TestBasic()
386 BundleWriter writer(Env::Default(), Prefix("foo")); in TestEndianness()
417 BundleWriter writer(Env::Default(), Prefix("bar")); in TestEndianness()
485 BundleWriter writer(Env::Default(), Prefix("nonstandard")); in TestNonStandardShapes()
569 BundleWriter writer(Env::Default(), Prefix("foo")); in TEST()
642 BundleWriter writer(Env::Default(), Prefix("foo")); in TEST()
727 BundleWriter writer(Env::Default(), Prefix("foo")); in TEST()
825 BundleWriter writer(Env::Default(), Prefix("foo")); in TEST()
848 BundleWriter writer(env, kBundlePrefixes[i]); in TEST()
[all …]
|
| D | tensor_bundle.h | 108 class BundleWriter {
116 BundleWriter(Env* env, StringPiece prefix,
160 TF_DISALLOW_COPY_AND_ASSIGN(BundleWriter);
|
| D | tensor_bundle.cc | 408 BundleWriter::BundleWriter(Env* env, StringPiece prefix, const Options& options) in BundleWriter() function in tensorflow::BundleWriter
439 Status BundleWriter::Add(StringPiece key, const Tensor& val) { in Add()
476 Status BundleWriter::AddSlice(StringPiece full_tensor_key, in AddSlice()
519 Status BundleWriter::Finish() { in Finish()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | merge_v2_checkpoints_op_test.cc | 39 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 | 378 writer_ = absl::make_unique<BundleWriter>(dataset()->env_, filename_); in RestoreInternal()
440 writer_ = absl::make_unique<BundleWriter>(dataset()->env_, filename_); in EnsureLockFileExists()
483 std::unique_ptr<BundleWriter> writer_ TF_GUARDED_BY(mu_);
|