Searched refs:TensorSliceReaderCacheWrapper (Results 1 – 9 of 9) sorted by relevance
27 TensorSliceReaderCacheWrapper::TensorSliceReaderCacheWrapper() {} in TensorSliceReaderCacheWrapper() function in tensorflow::checkpoint::TensorSliceReaderCacheWrapper28 TensorSliceReaderCacheWrapper::~TensorSliceReaderCacheWrapper() { in ~TensorSliceReaderCacheWrapper()33 const TensorSliceReader* TensorSliceReaderCacheWrapper::GetReader( in GetReader()
36 class TensorSliceReaderCacheWrapper {38 TensorSliceReaderCacheWrapper();39 ~TensorSliceReaderCacheWrapper();
192 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in CreateOpKernelContext()194 absl::make_unique<checkpoint::TensorSliceReaderCacheWrapper>(); in CreateOpKernelContext()
194 std::unique_ptr<checkpoint::TensorSliceReaderCacheWrapper>
178 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in TEST_F()423 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in TEST_F()
176 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in RunOpKernel()
172 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_;
64 class TensorSliceReaderCacheWrapper; variable673 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache = nullptr;1103 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache() const { in slice_reader_cache()
1259 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache_;1392 slice_reader_cache_(new checkpoint::TensorSliceReaderCacheWrapper), in ExecutorState()