Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc242 void SetAdaptiveBatchSchedulerOptions(OpKernelConstruction* c,
311 SetAdaptiveBatchSchedulerOptions(c, num_batch_threads_); in BatchFunctionFallbackKernel()
428 void BatchFunctionFallbackKernel::SetAdaptiveBatchSchedulerOptions( in SetAdaptiveBatchSchedulerOptions() function in tensorflow::tfd::__anonb95ad7eb0111::BatchFunctionFallbackKernel
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernels.h82 void SetAdaptiveBatchSchedulerOptions(OpKernelConstruction* c,
H A Dbatch_kernels.cc263 SetAdaptiveBatchSchedulerOptions(c, num_batch_threads_); in BatchFunctionKernel()
470 void BatchFunctionKernel::SetAdaptiveBatchSchedulerOptions( in SetAdaptiveBatchSchedulerOptions() function in tensorflow::BatchFunctionKernel