Searched refs:num_batch_threads_ (Results 1 – 1 of 1) sorted by relevance
702 OP_REQUIRES_OK(c, c->GetAttr("num_batch_threads", &num_batch_threads_)); in BatchFunctionKernel()726 BatchResource::Create(num_batch_threads_, max_batch_size_, in ComputeAsync()769 int32 num_batch_threads_; member in tensorflow::BatchFunctionKernel791 OP_REQUIRES_OK(c, c->GetAttr("num_batch_threads", &num_batch_threads_)); in BatchKernel()806 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()849 int32 num_batch_threads_; member in tensorflow::BatchKernel