Home
last modified time | relevance | path

Searched refs:TensorSliceReaderCacheWrapper (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader_cache.cc27 TensorSliceReaderCacheWrapper::TensorSliceReaderCacheWrapper() {} in TensorSliceReaderCacheWrapper() function in tensorflow::checkpoint::TensorSliceReaderCacheWrapper
28 TensorSliceReaderCacheWrapper::~TensorSliceReaderCacheWrapper() { in ~TensorSliceReaderCacheWrapper()
33 const TensorSliceReader* TensorSliceReaderCacheWrapper::GetReader( in GetReader()
Dtensor_slice_reader_cache.h36 class TensorSliceReaderCacheWrapper {
38 TensorSliceReaderCacheWrapper();
39 ~TensorSliceReaderCacheWrapper();
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc192 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in CreateOpKernelContext()
194 absl::make_unique<checkpoint::TensorSliceReaderCacheWrapper>(); in CreateOpKernelContext()
Ddataset_test_base.h194 std::unique_ptr<checkpoint::TensorSliceReaderCacheWrapper>
/external/tensorflow/tensorflow/core/kernels/
Drestore_op_test.cc178 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in TEST_F()
423 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in TEST_F()
Dops_testutil.h176 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in RunOpKernel()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h172 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_;
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h64 class TensorSliceReaderCacheWrapper; variable
673 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache = nullptr;
1103 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache() const { in slice_reader_cache()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc1259 checkpoint::TensorSliceReaderCacheWrapper* slice_reader_cache_;
1392 slice_reader_cache_(new checkpoint::TensorSliceReaderCacheWrapper), in ExecutorState()