OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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