Searched refs:slice_reader_cache (Results 1 – 9 of 9) sorted by relevance
166 if (context->slice_reader_cache()) { in RestoreTensor()167 reader = context->slice_reader_cache()->GetReader(file_pattern, open_func, in RestoreTensor()
182 params.slice_reader_cache = &slice_reader_cache_wrapper; in TEST_F()427 params.slice_reader_cache = &slice_reader_cache_wrapper; in TEST_F()
165 params_->slice_reader_cache = &slice_reader_cache_wrapper; in RunOpKernel()
724 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache = nullptr; member1155 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache() const { in slice_reader_cache() function1156 return params_->slice_reader_cache; in slice_reader_cache()
251 params.slice_reader_cache = &slice_reader_cache_; in Run()
255 params.slice_reader_cache = nullptr; // TODO(mrry): Too severe? in Run()
496 params->slice_reader_cache = slice_reader_cache_.get(); in CreateOpKernelContext()
443 params->slice_reader_cache = nullptr; in PrepOpKernel()
1725 params.slice_reader_cache = slice_reader_cache_; in Process()