Searched refs:slice_reader_cache (Results 1 – 8 of 8) sorted by relevance
166 if (context->slice_reader_cache()) { in RestoreTensor()167 reader = context->slice_reader_cache()->GetReader(file_pattern, open_func, in RestoreTensor()
179 params.slice_reader_cache = &slice_reader_cache_wrapper; in TEST_F()424 params.slice_reader_cache = &slice_reader_cache_wrapper; in TEST_F()
177 params_.get()->slice_reader_cache = &slice_reader_cache_wrapper; in RunOpKernel()
673 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache = nullptr; member1103 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache() const { in slice_reader_cache() function1104 return params_->slice_reader_cache; in slice_reader_cache()
220 params.slice_reader_cache = nullptr; // TODO(mrry): Too severe? in RunAsync()
195 params_->slice_reader_cache = slice_reader_cache_.get(); in CreateOpKernelContext()
240 params.slice_reader_cache = &slice_reader_cache_; in Run()
1638 params.slice_reader_cache = slice_reader_cache_; in Process()