Add keyword "_partitioned_" into the partition graph dump filename

PiperOrigin-RevId: 631470588
tensorflow · May 7, 2024 · 2db5012 · 2db5012
1 parent 5f1eb4d
commit 2db5012
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow/core/common_runtime/process_function_library_runtime.cc b/tensorflow/core/common_runtime/process_function_library_runtime.cc
@@ -634,7 +634,8 @@ Status ProcessFunctionLibraryRuntime::InstantiateMultiDevice(
   // Generate a random function_name to avoid one function reuse the partition
   // function instantiated by another function.
   FunctionNameGenerator name_generator(
-      &data_lib_def, absl::StrCat(function_name, "_", random::New64()));
+      &data_lib_def,
+      absl::StrCat(function_name, "_partitioned_", random::New64()));
   const int num_subgraphs = subgraphs->size();
   gtl::InlinedVector<Status, 4> instantiate_status(num_subgraphs);