Creating UI options for trace mode when capturing profiles using sampling mode

PiperOrigin-RevId: 812867750

Author: tensorflower-gardener

tsl/profiler/protobuf/BUILD

@@ -2,6 +2,7 @@
 # Placeholder: load py_proto_library
 load("@xla//xla/tsl:tsl.bzl", "internal_visibility")
 load("@xla//xla/tsl/platform:build_config.bzl", "tf_proto_library")
+load("//tools/build_defs/js/jspb_proto_library:rules.bzl", "jspb_proto_library")
 
 # copybara:uncomment package(default_applicable_licenses = ["//tensorflow:license"])
 
@@ -26,6 +27,12 @@ tf_proto_library(
     visibility = ["//visibility:public"],
 )
 
+jspb_proto_library(
+    name = "profiler_options_jspb_proto",
+    visibility = ["//visibility:public"],
+    deps = [":profiler_options_proto"],
+)
+
 tf_proto_library(
     name = "profiler_service_monitor_result_proto",
     srcs = ["profiler_service_monitor_result.proto"],

tsl/profiler/protobuf/profiler_options.proto

@@ -84,6 +84,13 @@ message ProfileOptions {
     }
   }
 
+  enum TraceMode {
+    TRACE_ONLY_HOST = 0;
+    TRACE_ONLY_XLA = 1;
+    TRACE_COMPUTE = 2;
+    TRACE_COMPUTE_AND_SYNC = 3;
+  }
+
   // Advanced configuration for the profiler contains a map of config name to
   // config value. It gives the flexibility to pass any configuration to the
   // profiler. eg: