Searched refs:TensorSliceReaderCacheWrapper (Results 1 – 9 of 9) sorted by relevance
26 TensorSliceReaderCacheWrapper::TensorSliceReaderCacheWrapper() {} in TensorSliceReaderCacheWrapper() function in tensorflow::checkpoint::TensorSliceReaderCacheWrapper27 TensorSliceReaderCacheWrapper::~TensorSliceReaderCacheWrapper() { in ~TensorSliceReaderCacheWrapper()32 const TensorSliceReader* TensorSliceReaderCacheWrapper::GetReader( in GetReader()
36 class TensorSliceReaderCacheWrapper {38 TensorSliceReaderCacheWrapper();39 ~TensorSliceReaderCacheWrapper();
181 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in TEST_F()426 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in TEST_F()
175 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in RunOpKernel()
248 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_;
67 class TensorSliceReaderCacheWrapper; variable693 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache = nullptr;1145 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache() const { in slice_reader_cache()
362 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache_;408 slice_reader_cache_(new checkpoint::TensorSliceReaderCacheWrapper), in ExecutorState()
769 std::unique_ptr<checkpoint::TensorSliceReaderCacheWrapper>
504 absl::make_unique<checkpoint::TensorSliceReaderCacheWrapper>(); in CreateOpKernelContext()