message: [tf.data] Fixing a subtle use-after-free bug in the optimization logic.
diff: @@ -91,7 +91,7 @@ class OptimizeDatasetOp : public UnaryDatasetOpKernel {
  Graph graph(OpRegistry::Global());
  TF_RETURN_IF_ERROR(ImportGraphDef({}, graph_def, &graph, nullptr));
  std::vector<Tensor> outputs;
  - GraphRunner graph_runner(ctx->env());
  + GraphRunner graph_runner(ctx->function_library()->device());
  // Once rewrites that add/modify functions are introduced, we will need
  // persist the results in a function library runtime.
  TF_RETURN_IF_ERROR(graph_runner.Run(&graph, ctx->function_library(), {},
  