Searched refs:TensorSliceReaderCacheWrapper (Results 1 – 10 of 10) 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()
70 class TensorSliceReaderCacheWrapper; variable665 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache = nullptr;1101 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache() const { in slice_reader_cache()
258 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_;
382 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache_;432 slice_reader_cache_(new checkpoint::TensorSliceReaderCacheWrapper), in ExecutorState()
822 std::unique_ptr<checkpoint::TensorSliceReaderCacheWrapper>
509 std::make_unique<checkpoint::TensorSliceReaderCacheWrapper>(); in CreateOpKernelContext()
1624 std::make_unique<checkpoint::TensorSliceReaderCacheWrapper>(); in CreateContext()1686 std::unique_ptr<checkpoint::TensorSliceReaderCacheWrapper>