HomeSort by relevance Sort by last modified time
    Searched refs:preferred_shard (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
restore_op.cc 30 int preferred_shard; local
32 context->GetAttr("preferred_shard", &preferred_shard));
33 if (preferred_shard == -1) {
36 OP_REQUIRES(context, preferred_shard >= 0,
37 errors::InvalidArgument("Attribute 'preferred_shard' must be "
39 preferred_shard_ = preferred_shard;
56 int preferred_shard; local
58 context->GetAttr("preferred_shard", &preferred_shard));
    [all...]
save_restore_tensor.h 41 // it as context->output(restore_index). "preferred_shard" is the same the
42 // TensorSliceReader preferred_shard parameter.
54 int preferred_shard, bool restore_slice, int restore_index);
save_restore_tensor.cc 144 int preferred_shard, bool restore_slice, int restore_index) {
164 preferred_shard);
167 file_pattern, open_func, preferred_shard));
  /external/tensorflow/tensorflow/core/util/
tensor_slice_reader_cache.h 45 int preferred_shard) const;
63 TensorSliceReader::OpenTableFunction open_function, int preferred_shard);
tensor_slice_reader_cache.cc 36 int preferred_shard) const {
42 preferred_shard);
55 TensorSliceReader::OpenTableFunction open_function, int preferred_shard) {
89 new TensorSliceReader(filepattern, open_function, preferred_shard));
tensor_slice_reader.cc 110 int preferred_shard)
132 if (preferred_shard == kLoadAllShards || fnames_.size() == 1 ||
133 static_cast<size_t>(preferred_shard) >= fnames_.size()) {
136 VLOG(1) << "Loading shard " << preferred_shard << " for " << filepattern_;
137 LoadShard(preferred_shard);
tensor_slice_reader.h 68 int preferred_shard);
  /external/tensorflow/tensorflow/python/ops/
io_ops.py 121 name="restore_slice", preferred_shard=-1):
133 preferred_shard: Int. Optional shard to open first in the checkpoint file.
141 preferred_shard, name=name)
  /external/tensorflow/tensorflow/python/training/
saver.py 245 def bulk_restore(self, filename_tensor, saveables, preferred_shard,
255 preferred_shard: Int. Shard to open first when loading a sharded file.
270 self.restore_op(filename_tensor, saveable, preferred_shard))
274 def restore_op(self, filename_tensor, saveable, preferred_shard):
283 preferred_shard: Int. Shard to open first when loading a sharded file.
431 preferred_shard=-1,
442 preferred_shard: Shard to open first when loading a sharded file.
448 all_tensors = self.bulk_restore(filename_tensor, saveables, preferred_shard,
501 preferred_shard=shard,
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 361 milliseconds