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
|