Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernels.cc242 OP_REQUIRES_OK(c, c->GetAttr("num_batch_threads", &num_batch_threads_)); in BatchFunctionKernel()
263 SetAdaptiveBatchSchedulerOptions(c, num_batch_threads_); in BatchFunctionKernel()
294 RecordBatchParamNumBatchThreads(num_batch_threads_, GetModelName(c)); in ComputeAsync()
343 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()
550 OP_REQUIRES_OK(c, c->GetAttr("num_batch_threads", &num_batch_threads_)); in BatchKernel()
565 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()
608 int32 num_batch_threads_; member in tensorflow::BatchKernel
H A Dbatch_kernels.h87 int32 num_batch_threads_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc248 int32 num_batch_threads_; member in tensorflow::tfd::__anonb95ad7eb0111::BatchFunctionFallbackKernel
281 OP_REQUIRES_OK(c, c->GetAttr("num_batch_threads", &num_batch_threads_)); in BatchFunctionFallbackKernel()
311 SetAdaptiveBatchSchedulerOptions(c, num_batch_threads_); in BatchFunctionFallbackKernel()
378 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()