Home
last modified time | relevance | path

Searched defs:preferred_shard (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drestore_op.cc30 int preferred_shard; in RestoreOp() local
56 int preferred_shard; in RestoreSliceOp() local
Dsave_restore_tensor.cc146 int preferred_shard, bool restore_slice, int restore_index) { in RestoreTensor()
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader_cache.cc54 TensorSliceReader::OpenTableFunction open_function, int preferred_shard) { in GetReader()
Dtensor_slice_reader.cc110 int preferred_shard) in TensorSliceReader()
/external/tensorflow/tensorflow/python/training/
Dsaver.py125 def bulk_restore(self, filename_tensor, saveables, preferred_shard, argument
156 def restore_op(self, filename_tensor, saveable, preferred_shard): argument
318 preferred_shard=-1, argument
570 def bulk_restore(self, filename_tensor, saveables, preferred_shard, argument
/external/tensorflow/tensorflow/python/ops/
Dio_ops.py76 name="restore_slice", preferred_shard=-1): argument