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()
176 params_->slice_reader_cache = &slice_reader_cache_wrapper; in RunOpKernel()
693 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache = nullptr; member1145 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache() const { in slice_reader_cache() function1146 return params_->slice_reader_cache; in slice_reader_cache()
257 params.slice_reader_cache = &slice_reader_cache_; in Run()
298 params.slice_reader_cache = nullptr; // TODO(mrry): Too severe? in Run()
505 params->slice_reader_cache = slice_reader_cache_.get(); in CreateOpKernelContext()
297 params->slice_reader_cache = nullptr; in PrepOpKernel()
709 params.slice_reader_cache = slice_reader_cache_; in Process()