HomeSort by relevance Sort by last modified time
    Searched full:bundlereader (Results 1 - 16 of 16) sorted by null

  /external/tensorflow/tensorflow/core/util/tensor_bundle/
tensor_bundle_test.cc 54 void Expect(BundleReader* reader, const string& key,
71 void ExpectVariant(BundleReader* reader, const string& key,
95 void ExpectNext(BundleReader* reader, const Tensor& expected_val) {
104 std::vector<string> AllTensorKeys(BundleReader* reader) {
170 BundleReader reader(Env::Default(), Prefix("foo"));
181 BundleReader reader(Env::Default(), Prefix("foo"));
200 BundleReader reader(Env::Default(), Prefix("bar"));
211 BundleReader reader(Env::Default(), Prefix("bar"));
224 BundleReader reader(Env::Default(), Prefix("merged"));
240 BundleReader reader(Env::Default(), Prefix("merged"))
    [all...]
tensor_bundle.h 36 // A tensor bundle can be accessed randomly using a BundleReader. Usage:
38 // BundleReader reader(env, "/fs/model/train/ckpt-step/ckpt");
181 // All threads accessing the same BundleReader must synchronize.
182 class BundleReader {
184 BundleReader(Env* const env, StringPiece prefix);
185 ~BundleReader();
305 TF_DISALLOW_COPY_AND_ASSIGN(BundleReader);
tensor_bundle.cc 698 BundleReader::BundleReader(Env* env, StringPiece prefix)
743 BundleReader::~BundleReader() {
757 Status BundleReader::GetBundleEntryProto(StringPiece key,
778 Status BundleReader::GetValue(const BundleEntryProto& entry, Tensor* val) {
    [all...]
  /external/tensorflow/tensorflow/c/
checkpoint_reader.h 34 // A wrapper around BundleReader (for V2 checkpoints) and
71 std::unique_ptr<BundleReader> v2_reader_;
checkpoint_reader.cc 44 new BundleReader(Env::Default(), filename /* prefix to a V2 ckpt */));
  /external/tensorflow/tensorflow/tools/graph_transforms/
sparsify_gather.cc 157 const string& tensor_name, const std::unique_ptr<BundleReader>& ckpt_reader,
184 std::unique_ptr<BundleReader>* ckpt_reader) {
187 ckpt_reader->reset(new BundleReader(Env::Default(), input_checkpoint));
228 const std::unique_ptr<BundleReader>& ckpt_reader,
597 std::unique_ptr<BundleReader> ckpt_reader;
sparsify_gather_test.cc 36 const string& tensor_name, const std::unique_ptr<BundleReader>& ckpt_reader,
585 std::unique_ptr<BundleReader> reader(
586 new BundleReader(Env::Default(), checkpoint_path));
  /external/tensorflow/tensorflow/core/kernels/
merge_v2_checkpoints_op_test.cc 91 BundleReader reader(Env::Default(), kMergedPrefix);
load_and_remap_matrix_op.cc 134 BundleReader reader(context->env(), ckpt_path);
save_restore_tensor.cc 246 BundleReader reader(Env::Default(), prefix_string);
save_v2_op_test.cc 126 BundleReader reader(Env::Default(), prefix);
  /prebuilts/tools/common/m2/repository/org/eclipse/tycho/tycho-core/0.18.1/
tycho-core-0.18.1.jar 
  /external/tensorflow/tensorflow/core/kernels/data/
cache_dataset_ops.cc 250 BundleReader reader_ GUARDED_BY(mu_);
  /external/tensorflow/tensorflow/python/kernel_tests/
checkpoint_ops_test.py 295 TensorBundle's BundleReader and TensorSlice logic).
  /prebuilts/tools/common/m2/repository/org/eclipse/tycho/sisu-equinox-launching/0.18.1/
sisu-equinox-launching-0.18.1.jar 
  /prebuilts/tools/common/m2/repository/org/eclipse/tycho/tycho-p2-facade/0.18.1/
tycho-p2-facade-0.18.1.jar 

Completed in 598 milliseconds