Searched refs:open_function (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader_cache.cc | 35 TensorSliceReader::OpenTableFunction open_function, in GetReader() argument
41 return cache_->GetReader(filepattern, std::move(open_function), in GetReader()
55 TensorSliceReader::OpenTableFunction open_function, int preferred_shard) { in GetReader() argument
61 open_function.target<TensorSliceReaderCache::OpenFuncType>(); in GetReader()
89 new TensorSliceReader(filepattern, open_function, preferred_shard)); in GetReader()
|
| D | tensor_slice_reader_test.cc | 57 TensorSliceReader::OpenTableFunction open_function) { in SimpleFloatHelper() argument
116 TensorSliceReader reader(filepattern, std::move(open_function)); in SimpleFloatHelper()
181 TensorSliceReader::OpenTableFunction open_function, in SimpleIntXHelper() argument
241 TensorSliceReader reader(filepattern, std::move(open_function)); in SimpleIntXHelper()
314 const TensorSliceReader::OpenTableFunction& open_function) { in TEST_SIMPLE_INT()
375 filepattern, open_function, TensorSliceReader::kLoadAllShards); in TEST_SIMPLE_INT()
391 filepattern, open_function, TensorSliceReader::kLoadAllShards); in TEST_SIMPLE_INT()
394 reader = cache.GetReader("file_does_not_exist", open_function, in TEST_SIMPLE_INT()
|
| D | tensor_slice_reader_cache.h | 44 TensorSliceReader::OpenTableFunction open_function,
63 TensorSliceReader::OpenTableFunction open_function, int preferred_shard);
|
| D | tensor_slice_reader.cc | 104 OpenTableFunction open_function) in TensorSliceReader() argument
105 : TensorSliceReader(filepattern, std::move(open_function), kLoadAllShards) { in TensorSliceReader()
109 OpenTableFunction open_function, in TensorSliceReader() argument
111 : filepattern_(filepattern), open_function_(std::move(open_function)) { in TensorSliceReader()
|
| D | tensor_slice_reader.h | 66 TensorSliceReader(const string& filepattern, OpenTableFunction open_function);
67 TensorSliceReader(const string& filepattern, OpenTableFunction open_function,
|