diff --git a/tensorflow/compiler/tf2xla/xla_compiler.cc b/tensorflow/compiler/tf2xla/xla_compiler.cc index 86a25177d27..a39656ed5d3 100644 --- a/tensorflow/compiler/tf2xla/xla_compiler.cc +++ b/tensorflow/compiler/tf2xla/xla_compiler.cc @@ -436,11 +436,10 @@ XlaCompiler::XlaCompiler(XlaCompiler::Options options) FunctionDefLibrary{})); local_pflr_.reset(new ProcessFunctionLibraryRuntime( &device_mgr_, Env::Default(), options.graph_def_version, - local_flib_def_.get(), OptimizerOptions(), - nullptr /* custom_kernel_creator */)); + local_flib_def_.get(), OptimizerOptions())); pflr_.reset(new ProcessFunctionLibraryRuntime( &device_mgr_, Env::Default(), options.graph_def_version, options.flib_def, - OptimizerOptions(), nullptr /* custom_kernel_creator */)); + OptimizerOptions())); local_flib_runtime_ = local_pflr_->GetFLR(device_->name()); flib_runtime_ = pflr_->GetFLR(device_->name());