Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc264 bool enable_adaptive_batch_threads_ = false; member in tensorflow::tfd::__anonb95ad7eb0111::BatchFunctionFallbackKernel
316 if (enable_adaptive_batch_threads_) { in BatchFunctionFallbackKernel()
432 &enable_adaptive_batch_threads_)); in SetAdaptiveBatchSchedulerOptions()
436 enable_adaptive_batch_threads_ = true; in SetAdaptiveBatchSchedulerOptions()
439 if (!enable_adaptive_batch_threads_) { in SetAdaptiveBatchSchedulerOptions()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernels.cc268 if (enable_adaptive_batch_threads_) { in BatchFunctionKernel()
474 &enable_adaptive_batch_threads_)); in SetAdaptiveBatchSchedulerOptions()
478 enable_adaptive_batch_threads_ = true; in SetAdaptiveBatchSchedulerOptions()
481 if (!enable_adaptive_batch_threads_) { in SetAdaptiveBatchSchedulerOptions()
H A Dbatch_kernel_test_util.cc26 return kernel_->enable_adaptive_batch_threads_; in enable_adaptive_batch_threads()
H A Dbatch_kernels.h97 bool enable_adaptive_batch_threads_ = false; variable