Searched refs:BundleReader (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle_test.cc | 73 void Expect(BundleReader* reader, const string& key, in Expect()
90 void ExpectVariant(BundleReader* reader, const string& key, in ExpectVariant()
114 void ExpectNext(BundleReader* reader, const Tensor& expected_val) { in ExpectNext()
123 std::vector<string> AllTensorKeys(BundleReader* reader) { in AllTensorKeys()
195 BundleReader reader(Env::Default(), Prefix("foo")); in TestBasic()
206 BundleReader reader(Env::Default(), Prefix("foo")); in TestBasic()
225 BundleReader reader(Env::Default(), Prefix("bar")); in TestBasic()
236 BundleReader reader(Env::Default(), Prefix("bar")); in TestBasic()
249 BundleReader reader(Env::Default(), Prefix("merged")); in TestBasic()
265 BundleReader reader(Env::Default(), Prefix("merged")); in TestBasic()
[all …]
|
| D | tensor_bundle.h | 183 class BundleReader {
185 BundleReader(Env* const env, StringPiece prefix);
186 ~BundleReader();
311 TF_DISALLOW_COPY_AND_ASSIGN(BundleReader);
|
| D | tensor_bundle.cc | 744 BundleReader::BundleReader(Env* env, StringPiece prefix) in BundleReader() function in tensorflow::BundleReader
799 BundleReader::~BundleReader() { in ~BundleReader()
822 Status BundleReader::GetBundleEntryProto(StringPiece key, in GetBundleEntryProto()
843 Status BundleReader::GetValue(const BundleEntryProto& entry, Tensor* val) { in GetValue()
939 Status BundleReader::Lookup(StringPiece key, Tensor* val) { in Lookup()
953 Status BundleReader::ReadCurrent(Tensor* val) { in ReadCurrent()
971 Status BundleReader::LookupTensorSlices(StringPiece key, in LookupTensorSlices()
983 Status BundleReader::LookupSlice(StringPiece full_tensor_key, in LookupSlice()
991 Status BundleReader::GetSliceValue(StringPiece full_tensor_key, in GetSliceValue()
1106 bool BundleReader::Contains(StringPiece key) { in Contains()
[all …]
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | bundle_v2.h | 60 BundleReader* variable_reader() { return variable_reader_.get(); } in variable_reader()
81 std::unique_ptr<BundleReader> variable_reader_;
|
| D | bundle_v2.cc | 70 Status ReadCheckpointObjectGraph(BundleReader* bundle_reader, in ReadCheckpointObjectGraph()
127 new BundleReader(Env::Default(), variables_prefix)); in Load()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_tensor.cc | 247 bool should_run_in_pool(BundleReader* reader) const { in should_run_in_pool()
260 BundleReader reader(Env::Default(), reader_prefix); in run_with_new_reader()
269 Status run(BundleReader* reader) { in run()
356 BundleReader default_reader(Env::Default(), prefix_string); in RestoreTensorsV2()
|
| D | merge_v2_checkpoints_op_test.cc | 92 BundleReader reader(Env::Default(), kMergedPrefix); in RunMergeTest()
|
| D | load_and_remap_matrix_op.cc | 133 BundleReader reader(context->env(), ckpt_path); in Compute()
|
| D | save_v2_op_test.cc | 127 BundleReader reader(Env::Default(), prefix); in TEST_F()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 156 const string& tensor_name, const std::unique_ptr<BundleReader>& ckpt_reader, in ReadTensorFromCheckpoint()
183 std::unique_ptr<BundleReader>* ckpt_reader) { in InitializeCheckpointReader()
186 ckpt_reader->reset(new BundleReader(Env::Default(), input_checkpoint)); in InitializeCheckpointReader()
227 const std::unique_ptr<BundleReader>& ckpt_reader, in SparsifyGatherInternal()
596 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 | 43 new BundleReader(Env::Default(), filename /* prefix to a V2 ckpt */)); in CheckpointReader()
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 129 tensorflow::BundleReader::BundleReader
130 tensorflow::BundleReader::~BundleReader
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | cache_dataset_ops.cc | 574 BundleReader reader_ TF_GUARDED_BY(mu_);
|