Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Drestore_op_test.cc181 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in TEST_F() local
182 params.slice_reader_cache = &slice_reader_cache_wrapper; in TEST_F()
426 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in TEST_F() local
427 params.slice_reader_cache = &slice_reader_cache_wrapper; in TEST_F()
H A Dops_testutil.cc175 checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper; in RunOpKernel() local
176 params_->slice_reader_cache = &slice_reader_cache_wrapper; in RunOpKernel()