Searched refs:allowed_batch_sizes_ (Results 1 – 1 of 1) sorted by relevance
228 new_resource->allowed_batch_sizes_ = allowed_batch_sizes; in Create()317 if (allowed_batch_sizes_.empty()) { in RoundToLowestAllowedBatchSize()320 for (int allowed_size : allowed_batch_sizes_) { in RoundToLowestAllowedBatchSize()687 std::vector<int32> allowed_batch_sizes_; member in tensorflow::BatchResource708 OP_REQUIRES_OK(c, c->GetAttr("allowed_batch_sizes", &allowed_batch_sizes_)); in BatchFunctionKernel()728 allowed_batch_sizes_, fhandle_, &new_resource)); in ComputeAsync()746 if (allowed_batch_sizes_.empty()) { in ValidateAllowedBatchSizes()750 for (size_t i = 0; i < allowed_batch_sizes_.size(); ++i) { in ValidateAllowedBatchSizes()751 const int32 size = allowed_batch_sizes_.at(i); in ValidateAllowedBatchSizes()756 if (i == allowed_batch_sizes_.size() - 1 && size != max_batch_size_) { in ValidateAllowedBatchSizes()[all …]