Searched refs:num_batch_threads_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 193 OP_REQUIRES_OK(c, c->GetAttr("num_batch_threads", &num_batch_threads_)); in BatchFunctionKernel()
203 if (num_batch_threads_ <= 0) { in BatchFunctionKernel()
253 RecordBatchParamNumBatchThreads(num_batch_threads_, GetModelName(c)); in ComputeAsync()
300 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()
424 int32 num_batch_threads_; member in tensorflow::BatchFunctionKernel
470 OP_REQUIRES_OK(c, c->GetAttr("num_batch_threads", &num_batch_threads_)); in BatchKernel()
485 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()
528 int32 num_batch_threads_; member in tensorflow::BatchKernel
|