Searched refs:max_batch_size_ (Results 1 – 1 of 1) sorted by relevance
703 OP_REQUIRES_OK(c, c->GetAttr("max_batch_size", &max_batch_size_)); in BatchFunctionKernel()726 BatchResource::Create(num_batch_threads_, max_batch_size_, in ComputeAsync()756 if (i == allowed_batch_sizes_.size() - 1 && size != max_batch_size_) { in ValidateAllowedBatchSizes()770 int32 max_batch_size_; member in tensorflow::BatchFunctionKernel792 OP_REQUIRES_OK(c, c->GetAttr("max_batch_size", &max_batch_size_)); in BatchKernel()806 num_batch_threads_, max_batch_size_, batch_timeout_micros_, in ComputeAsync()836 if (i == allowed_batch_sizes_.size() - 1 && size != max_batch_size_) { in ValidateAllowedBatchSizes()850 int32 max_batch_size_; member in tensorflow::BatchKernel