Home
last modified time | relevance | path

Searched refs:allowed_batch_sizes_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc199 OP_REQUIRES_OK(c, c->GetAttr("allowed_batch_sizes", &allowed_batch_sizes_)); in BatchFunctionKernel()
291 batch_timeout_micros_, max_enqueued_batches_, allowed_batch_sizes_, in ComputeAsync()
301 max_enqueued_batches_, allowed_batch_sizes_, handle, flib_, in ComputeAsync()
397 if (allowed_batch_sizes_.empty()) { in ValidateAllowedBatchSizes()
401 for (size_t i = 0; i < allowed_batch_sizes_.size(); ++i) { in ValidateAllowedBatchSizes()
402 const int32 size = allowed_batch_sizes_.at(i); in ValidateAllowedBatchSizes()
409 (i == allowed_batch_sizes_.size() - 1) && (size != max_batch_size_)) { in ValidateAllowedBatchSizes()
428 std::vector<int32> allowed_batch_sizes_; member in tensorflow::BatchFunctionKernel
476 OP_REQUIRES_OK(c, c->GetAttr("allowed_batch_sizes", &allowed_batch_sizes_)); in BatchKernel()
486 max_enqueued_batches_, allowed_batch_sizes_, kInvalidHandle, in ComputeAsync()
[all …]
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.h123 allowed_batch_sizes_(std::move(allowed_batch_sizes)) { in BatchResourceBase()
124 allowed_batch_sizes_str_ = absl::StrJoin(allowed_batch_sizes_, ","); in BatchResourceBase()
134 allowed_batch_sizes_(std::move(allowed_batch_sizes)) {} in BatchResourceBase()
230 std::vector<int32> allowed_batch_sizes_; variable
Dbatch_resource_base.cc303 if (allowed_batch_sizes_.empty()) { in RoundToLowestAllowedBatchSize()
306 for (int allowed_size : allowed_batch_sizes_) { in RoundToLowestAllowedBatchSize()