diff --git a/tensorflow/core/common_runtime/dml/dml_kernel_manager.h b/tensorflow/core/common_runtime/dml/dml_kernel_manager.h index 0b2a6406b7..c6eb49efd3 100644 --- a/tensorflow/core/common_runtime/dml/dml_kernel_manager.h +++ b/tensorflow/core/common_runtime/dml/dml_kernel_manager.h @@ -49,7 +49,7 @@ class NoOpInitializationHelper; class DmlKernelManager { public: // Can be overridden by the TF_DIRECTML_KERNEL_CACHE_SIZE environment variable - static constexpr size_t kDefaultMaxCacheSize = 1024; + static constexpr size_t kDefaultMaxCacheSize = 1536; DmlKernelManager();