Home
last modified time | relevance | path

Searched refs:slice_reader_cache_wrapper (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drestore_op_test.cc178 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in TEST_F() local
179 params.slice_reader_cache = &slice_reader_cache_wrapper; in TEST_F()
423 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in TEST_F() local
424 params.slice_reader_cache = &slice_reader_cache_wrapper; in TEST_F()
Dops_testutil.h176 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in RunOpKernel() local
177 params_.get()->slice_reader_cache = &slice_reader_cache_wrapper; in RunOpKernel()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc192 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in CreateOpKernelContext() local