Home
last modified time | relevance | path

Searched refs:BatchFunctionKernelTestAccess (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernel_test_util.cc21 BatchFunctionKernelTestAccess::BatchFunctionKernelTestAccess( in BatchFunctionKernelTestAccess() function in tensorflow::internal::BatchFunctionKernelTestAccess
25 bool BatchFunctionKernelTestAccess::enable_adaptive_batch_threads() const { in enable_adaptive_batch_threads()
H A Dbatch_kernel_test_util.h29 class BatchFunctionKernelTestAccess {
31 explicit BatchFunctionKernelTestAccess(BatchFunctionKernel* kernel);
H A Dbatch_kernels.h34 class BatchFunctionKernelTestAccess; variable
58 friend class internal::BatchFunctionKernelTestAccess;
H A Dbatch_kernels_test.cc29 EXPECT_EQ(internal::BatchFunctionKernelTestAccess(batch_kernel) in TEST_P()