Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbatch_resource_base.h207 virtual Status CreateBatchTask(
H A Dbatch_resource_base.cc239 TF_RETURN_IF_ERROR(CreateBatchTask(context, &batch_components)); in RegisterInput()
896 Status BatchResourceBase::CreateBatchTask( in CreateBatchTask() function in tensorflow::serving::BatchResourceBase
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc208 Status CreateBatchTask(OpKernelContext* c, in CreateBatchTask() function in tensorflow::tfd::__anonb95ad7eb0111::FallbackBatchResource