Searched refs:enable_large_batch_splitting_ (Results 1 – 3 of 3) sorted by relevance
253 bool enable_large_batch_splitting_; member in tensorflow::tfd::__anon6e2be26b0111::BatchFunctionFallbackKernel303 &enable_large_batch_splitting_)); in BatchFunctionFallbackKernel()305 enable_large_batch_splitting_ = false; in BatchFunctionFallbackKernel()380 enable_large_batch_splitting_, *exec_ctx, &new_resource)); in ComputeAsync()416 if ((!enable_large_batch_splitting_) && in ValidateAllowedBatchSizes()
95 bool enable_large_batch_splitting_; variable
253 &enable_large_batch_splitting_)); in BatchFunctionKernel()256 enable_large_batch_splitting_ = false; in BatchFunctionKernel()290 ? absl::make_optional(enable_large_batch_splitting_) in ComputeAsync()345 enable_large_batch_splitting_, &new_resource)); in ComputeAsync()451 if ((!enable_large_batch_splitting_) && in ValidateAllowedBatchSizes()