Searched defs:preferred_shard (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | restore_op.cc | 30 int preferred_shard; in RestoreOp() local 56 int preferred_shard; in RestoreSliceOp() local
|
D | save_restore_tensor.cc | 146 int preferred_shard, bool restore_slice, int restore_index) { in RestoreTensor()
|
/external/tensorflow/tensorflow/core/util/ |
D | tensor_slice_reader_cache.cc | 54 TensorSliceReader::OpenTableFunction open_function, int preferred_shard) { in GetReader()
|
D | tensor_slice_reader.cc | 110 int preferred_shard) in TensorSliceReader()
|
/external/tensorflow/tensorflow/python/training/ |
D | saver.py | 125 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/ |
D | io_ops.py | 76 name="restore_slice", preferred_shard=-1): argument
|