Searched refs:batch_timeout_micros_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_batch_tf_opkernels.cc | 250 int32 batch_timeout_micros_; member in tensorflow::tfd::__anon6e2be26b0111::BatchFunctionFallbackKernel
283 OP_REQUIRES_OK(c, c->GetAttr("batch_timeout_micros", &batch_timeout_micros_)); in BatchFunctionFallbackKernel()
367 batch_timeout_micros_, max_enqueued_batches_, allowed_batch_sizes_, in ComputeAsync()
378 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 244 OP_REQUIRES_OK(c, c->GetAttr("batch_timeout_micros", &batch_timeout_micros_)); in BatchFunctionKernel()
334 batch_timeout_micros_, max_enqueued_batches_, allowed_batch_sizes_, in ComputeAsync()
343 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()
553 c->GetAttr("batch_timeout_micros", &batch_timeout_micros_)); in BatchKernel()
565 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()
610 int32 batch_timeout_micros_; member in tensorflow::BatchKernel
|
| D | batch_kernels.h | 89 int32 batch_timeout_micros_; variable
|