Home
last modified time | relevance | path

Searched refs:kLoadAllShards (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drestore_op.cc34 preferred_shard_ = checkpoint::TensorSliceReader::kLoadAllShards; in RestoreOp()
60 preferred_shard_ = checkpoint::TensorSliceReader::kLoadAllShards; in RestoreSliceOp()
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader.cc101 kLoadAllShards) {} in TensorSliceReader()
105 : TensorSliceReader(filepattern, std::move(open_function), kLoadAllShards) { in TensorSliceReader()
132 if (preferred_shard == kLoadAllShards || fnames_.size() == 1 || in TensorSliceReader()
Dtensor_slice_reader_test.cc375 filepattern, open_function, TensorSliceReader::kLoadAllShards); in TEST_SIMPLE_INT()
391 filepattern, open_function, TensorSliceReader::kLoadAllShards); in TEST_SIMPLE_INT()
395 TensorSliceReader::kLoadAllShards); in TEST_SIMPLE_INT()
Dtensor_slice_reader.h64 static const int kLoadAllShards = -1; variable