| /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.h | 54 int preferred_shard, bool restore_slice, int restore_index);
|
| 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()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader_cache.cc | 35 int preferred_shard) const { in GetReader()
41 preferred_shard); in GetReader()
54 TensorSliceReader::OpenTableFunction open_function, int preferred_shard) { in GetReader() argument
88 new TensorSliceReader(filepattern, open_function, preferred_shard)); in GetReader()
|
| D | tensor_slice_reader_cache.h | 45 int preferred_shard) const;
63 TensorSliceReader::OpenTableFunction open_function, int preferred_shard);
|
| 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.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/core/ops/compat/ops_history_v1/ |
| D | Restore.pbtxt | 20 name: "preferred_shard"
46 name: "preferred_shard"
|
| D | RestoreSlice.pbtxt | 24 name: "preferred_shard"
54 name: "preferred_shard"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Restore.pbtxt | 20 name: "preferred_shard"
46 name: "preferred_shard"
|
| D | RestoreSlice.pbtxt | 24 name: "preferred_shard"
54 name: "preferred_shard"
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 125 def bulk_restore(self, filename_tensor, saveables, preferred_shard, argument
152 self.restore_op(filename_tensor, saveable, preferred_shard))
156 def restore_op(self, filename_tensor, saveable, preferred_shard): argument
318 preferred_shard=-1, argument
335 all_tensors = self.bulk_restore(filename_tensor, saveables, preferred_shard,
388 preferred_shard=shard,
570 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 | 3729 …argspec: "args=[\'file_pattern\', \'tensor_name\', \'dt\', \'preferred_shard\', \'name\'], varargs…
3733 …s=[\'file_pattern\', \'tensor_name\', \'shape_and_slice\', \'dt\', \'preferred_shard\', \'name\'],…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3729 …argspec: "args=[\'file_pattern\', \'tensor_name\', \'dt\', \'preferred_shard\', \'name\'], varargs…
3733 …s=[\'file_pattern\', \'tensor_name\', \'shape_and_slice\', \'dt\', \'preferred_shard\', \'name\'],…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 11835 If a `file_pattern` matches several files, `preferred_shard` can be used to hint
11837 open the file at index `preferred_shard` in the list of matching files and try
11840 `preferred_shard` to match the value passed as the `shard` input
11854 DefaultValuedAttr<I64Attr, "-1">:$preferred_shard
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 27805 name: "preferred_shard"
27835 name: "preferred_shard"
|
| D | ops_history.v2.pbtxt | 62191 name: "preferred_shard"
62217 name: "preferred_shard"
62248 name: "preferred_shard"
62278 name: "preferred_shard"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 40531 name: "preferred_shard"
40562 name: "preferred_shard"
|