Searched refs:BundleReader (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle_test.cc | 60 void Expect(BundleReader* reader, const string& key, in Expect()
77 void ExpectVariant(BundleReader* reader, const string& key, in ExpectVariant()
101 void ExpectNext(BundleReader* reader, const Tensor& expected_val) { in ExpectNext()
110 std::vector<string> AllTensorKeys(BundleReader* reader) { in AllTensorKeys()
176 BundleReader reader(Env::Default(), Prefix("foo")); in TestBasic()
187 BundleReader reader(Env::Default(), Prefix("foo")); in TestBasic()
206 BundleReader reader(Env::Default(), Prefix("bar")); in TestBasic()
217 BundleReader reader(Env::Default(), Prefix("bar")); in TestBasic()
230 BundleReader reader(Env::Default(), Prefix("merged")); in TestBasic()
246 BundleReader reader(Env::Default(), Prefix("merged")); in TestBasic()
[all …]
|
| D | tensor_bundle.h | 182 class BundleReader {
184 BundleReader(Env* const env, StringPiece prefix);
185 ~BundleReader();
305 TF_DISALLOW_COPY_AND_ASSIGN(BundleReader);
|
| D | tensor_bundle.cc | 716 BundleReader::BundleReader(Env* env, StringPiece prefix) in BundleReader() function in tensorflow::BundleReader
761 BundleReader::~BundleReader() { in ~BundleReader()
775 Status BundleReader::GetBundleEntryProto(StringPiece key, in GetBundleEntryProto()
796 Status BundleReader::GetValue(const BundleEntryProto& entry, Tensor* val) { in GetValue()
879 Status BundleReader::Lookup(StringPiece key, Tensor* val) { in Lookup()
893 Status BundleReader::ReadCurrent(Tensor* val) { in ReadCurrent()
911 Status BundleReader::LookupTensorSlices(StringPiece key, in LookupTensorSlices()
923 Status BundleReader::LookupSlice(StringPiece full_tensor_key, in LookupSlice()
931 Status BundleReader::GetSliceValue(StringPiece full_tensor_key, in GetSliceValue()
1045 bool BundleReader::Contains(StringPiece key) { in Contains()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_tensor.cc | 246 bool should_run_in_pool(BundleReader* reader) const { in should_run_in_pool()
259 BundleReader reader(Env::Default(), reader_prefix); in run_with_new_reader()
268 Status run(BundleReader* reader) { in run()
337 BundleReader default_reader(Env::Default(), prefix_string); in RestoreTensorsV2()
|
| D | merge_v2_checkpoints_op_test.cc | 91 BundleReader reader(Env::Default(), kMergedPrefix); in RunMergeTest()
|
| D | load_and_remap_matrix_op.cc | 134 BundleReader reader(context->env(), ckpt_path); in Compute()
|
| D | save_v2_op_test.cc | 125 BundleReader reader(Env::Default(), prefix); in TEST_F()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 157 const string& tensor_name, const std::unique_ptr<BundleReader>& ckpt_reader, in ReadTensorFromCheckpoint()
184 std::unique_ptr<BundleReader>* ckpt_reader) { in InitializeCheckpointReader()
187 ckpt_reader->reset(new BundleReader(Env::Default(), input_checkpoint)); in InitializeCheckpointReader()
228 const std::unique_ptr<BundleReader>& ckpt_reader, in SparsifyGatherInternal()
597 std::unique_ptr<BundleReader> ckpt_reader; in SparsifyGather()
|
| D | sparsify_gather_test.cc | 36 const string& tensor_name, const std::unique_ptr<BundleReader>& ckpt_reader,
585 std::unique_ptr<BundleReader> reader( in TestReadTensorSlice()
586 new BundleReader(Env::Default(), checkpoint_path)); in TestReadTensorSlice()
|
| /external/tensorflow/tensorflow/c/ |
| D | checkpoint_reader.h | 71 std::unique_ptr<BundleReader> v2_reader_;
|
| D | checkpoint_reader.cc | 44 new BundleReader(Env::Default(), filename /* prefix to a V2 ckpt */)); in CheckpointReader()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | cache_dataset_ops.cc | 526 BundleReader reader_ GUARDED_BY(mu_);
|