HomeSort by relevance Sort by last modified time
    Searched refs:kLoadAllShards (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
restore_op.cc 34 preferred_shard_ = checkpoint::TensorSliceReader::kLoadAllShards;
60 preferred_shard_ = checkpoint::TensorSliceReader::kLoadAllShards;
  /external/tensorflow/tensorflow/core/util/
tensor_slice_reader.cc 101 kLoadAllShards) {}
105 : TensorSliceReader(filepattern, std::move(open_function), kLoadAllShards) {
132 if (preferred_shard == kLoadAllShards || fnames_.size() == 1 ||
tensor_slice_reader_test.cc 374 filepattern, open_function, TensorSliceReader::kLoadAllShards);
390 filepattern, open_function, TensorSliceReader::kLoadAllShards);
394 TensorSliceReader::kLoadAllShards);
tensor_slice_reader.h 64 static const int kLoadAllShards = -1;

Completed in 138 milliseconds