| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle_test.cc | 81 void Expect(BundleReader* reader, const string& key, in Expect()
98 void ExpectVariant(BundleReader* reader, const string& key, in ExpectVariant()
122 void ExpectNext(BundleReader* reader, const Tensor& expected_val) { in ExpectNext()
131 std::vector<string> AllTensorKeys(BundleReader* reader) { in AllTensorKeys()
203 BundleReader reader(Env::Default(), Prefix("foo")); in TestBasic()
214 BundleReader reader(Env::Default(), Prefix("foo")); in TestBasic()
233 BundleReader reader(Env::Default(), Prefix("bar")); in TestBasic()
244 BundleReader reader(Env::Default(), Prefix("bar")); in TestBasic()
257 BundleReader reader(Env::Default(), Prefix("merged")); in TestBasic()
273 BundleReader reader(Env::Default(), Prefix("merged")); in TestBasic()
[all …]
|
| D | tensor_bundle.h | 197 class BundleReader {
199 BundleReader(Env* const env, StringPiece prefix);
200 ~BundleReader();
337 TF_DISALLOW_COPY_AND_ASSIGN(BundleReader);
376 Status BundleReader::SortForSequentialAccess( in SortForSequentialAccess()
|
| D | tensor_bundle.cc | 757 BundleReader::BundleReader(Env* env, StringPiece prefix) in BundleReader() function in tensorflow::BundleReader
812 BundleReader::~BundleReader() { in ~BundleReader()
835 Status BundleReader::GetBundleEntryProto(StringPiece key, in GetBundleEntryProto()
856 Status BundleReader::GetValue(const BundleEntryProto& entry, Tensor* val) { in GetValue()
952 Status BundleReader::Lookup(StringPiece key, Tensor* val) { in Lookup()
966 Status BundleReader::ReadCurrent(Tensor* val) { in ReadCurrent()
984 Status BundleReader::LookupTensorSlices(StringPiece key, in LookupTensorSlices()
996 Status BundleReader::LookupSlice(StringPiece full_tensor_key, in LookupSlice()
1004 Status BundleReader::GetSliceValue(StringPiece full_tensor_key, in GetSliceValue()
1119 bool BundleReader::Contains(StringPiece key) { in Contains()
[all …]
|
| /external/tensorflow/tensorflow/cc/experimental/libexport/ |
| D | load.h | 86 tensorflow::BundleReader* GetVariableReader() { in GetVariableReader()
100 std::unique_ptr<tensorflow::BundleReader> variable_reader_;
|
| D | load.cc | 65 tf_package.variable_reader_ = std::make_unique<tensorflow::BundleReader>( in Load()
|
| /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 | 73 Status ReadCheckpointObjectGraph(BundleReader* bundle_reader, in ReadCheckpointObjectGraph()
136 new BundleReader(Env::Default(), variables_prefix)); in Load()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_tensor.cc | 274 bool should_run_in_pool(BundleReader* reader) const { in should_run_in_pool()
287 BundleReader reader(Env::Default(), reader_prefix); in run_with_new_reader()
296 Status run(BundleReader* reader) { in run()
380 BundleReader default_reader(Env::Default(), prefix_string); in RestoreTensorsV2()
|
| D | merge_v2_checkpoints_op_test.cc | 96 BundleReader reader(Env::Default(), kMergedPrefix); in RunMergeTest()
|
| D | load_and_remap_matrix_op.cc | 143 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 | 138 tensorflow::BundleReader::BundleReader
139 tensorflow::BundleReader::~BundleReader
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | cache_dataset_ops.cc | 637 BundleReader reader_ TF_GUARDED_BY(mu_);
|