Searched refs:enable_large_batch_splitting_ (Results 1 – 1 of 1) sorted by relevance
234 &enable_large_batch_splitting_)); in BatchFunctionKernel()237 enable_large_batch_splitting_ = false; in BatchFunctionKernel()249 ? absl::make_optional(enable_large_batch_splitting_) in ComputeAsync()302 enable_large_batch_splitting_, &new_resource)); in ComputeAsync()408 if ((!enable_large_batch_splitting_) && in ValidateAllowedBatchSizes()432 bool enable_large_batch_splitting_; member in tensorflow::BatchFunctionKernel