| /external/tensorflow/tensorflow/core/kernels/ |
| D | restore_op.cc | 30 int preferred_shard; in RestoreOp() local
32 context->GetAttr("preferred_shard", &preferred_shard)); in RestoreOp()
33 if (preferred_shard == -1) { in RestoreOp()
36 OP_REQUIRES(context, preferred_shard >= 0, in RestoreOp()
39 preferred_shard_ = preferred_shard; in RestoreOp()
56 int preferred_shard; in RestoreSliceOp() local
58 context->GetAttr("preferred_shard", &preferred_shard)); in RestoreSliceOp()
59 if (preferred_shard == -1) { in RestoreSliceOp()
62 OP_REQUIRES(context, preferred_shard >= 0, in RestoreSliceOp()
65 preferred_shard_ = preferred_shard; in RestoreSliceOp()
|
| D | save_restore_tensor.cc | 146 int preferred_shard, bool restore_slice, int restore_index) { in RestoreTensor() argument
168 preferred_shard); in RestoreTensor()
172 file_pattern, open_func, preferred_shard)); in RestoreTensor()
|
| D | save_restore_tensor.h | 54 int preferred_shard, bool restore_slice, int restore_index);
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader_cache.cc | 36 int preferred_shard) const { in GetReader()
42 preferred_shard); in GetReader()
55 TensorSliceReader::OpenTableFunction open_function, int preferred_shard) { in GetReader() argument
89 new TensorSliceReader(filepattern, open_function, preferred_shard)); in GetReader()
|
| D | tensor_slice_reader.cc | 110 int preferred_shard) in TensorSliceReader() argument
132 if (preferred_shard == kLoadAllShards || fnames_.size() == 1 || in TensorSliceReader()
133 static_cast<size_t>(preferred_shard) >= fnames_.size()) { in TensorSliceReader()
136 VLOG(1) << "Loading shard " << preferred_shard << " for " << filepattern_; in TensorSliceReader()
137 LoadShard(preferred_shard); in TensorSliceReader()
|
| D | tensor_slice_reader_cache.h | 45 int preferred_shard) const;
63 TensorSliceReader::OpenTableFunction open_function, int preferred_shard);
|
| D | tensor_slice_reader.h | 68 int preferred_shard);
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Restore.pbtxt | 30 name: "preferred_shard"
43 If a `file_pattern` matches several files, `preferred_shard` can be used to hint
45 open the file at index `preferred_shard` in the list of matching files and try
48 `preferred_shard` to match the value passed as the `shard` input
|
| D | api_def_RestoreSlice.pbtxt | 37 name: "preferred_shard"
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 127 def bulk_restore(self, filename_tensor, saveables, preferred_shard, argument
154 self.restore_op(filename_tensor, saveable, preferred_shard))
158 def restore_op(self, filename_tensor, saveable, preferred_shard): argument
313 preferred_shard=-1, argument
330 all_tensors = self.bulk_restore(filename_tensor, saveables, preferred_shard,
383 preferred_shard=shard,
566 def bulk_restore(self, filename_tensor, saveables, preferred_shard, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | io_ops.py | 76 name="restore_slice", preferred_shard=-1): argument
96 preferred_shard, name=name)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3029 …argspec: "args=[\'file_pattern\', \'tensor_name\', \'dt\', \'preferred_shard\', \'name\'], varargs…
3033 …s=[\'file_pattern\', \'tensor_name\', \'shape_and_slice\', \'dt\', \'preferred_shard\', \'name\'],…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3029 …argspec: "args=[\'file_pattern\', \'tensor_name\', \'dt\', \'preferred_shard\', \'name\'], varargs…
3033 …s=[\'file_pattern\', \'tensor_name\', \'shape_and_slice\', \'dt\', \'preferred_shard\', \'name\'],…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 27805 name: "preferred_shard"
27835 name: "preferred_shard"
|
| D | ops_history.v2.pbtxt | 62854 name: "preferred_shard"
62880 name: "preferred_shard"
62911 name: "preferred_shard"
62941 name: "preferred_shard"
|
| D | ops_history.v1.pbtxt | 63481 name: "preferred_shard"
63507 name: "preferred_shard"
63538 name: "preferred_shard"
63568 name: "preferred_shard"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 31317 name: "preferred_shard"
31348 name: "preferred_shard"
|