Add overload of ParseTfNameScopes that takes op_name.
PiperOrigin-RevId: 387852979
Change-Id: I5bdd31b0b88d755dfdd1f874dd183a0e3d8a8db3
diff --git a/tensorflow/core/profiler/utils/tf_op_utils.cc b/tensorflow/core/profiler/utils/tf_op_utils.cc
index 2c8c869..7d6d7d6 100644
--- a/tensorflow/core/profiler/utils/tf_op_utils.cc
+++ b/tensorflow/core/profiler/utils/tf_op_utils.cc
@@ -122,14 +122,18 @@
return tf_op;
}
-std::vector<absl::string_view> ParseTfNameScopes(const TfOp& tf_op) {
+std::vector<absl::string_view> ParseTfNameScopes(absl::string_view tf_op_name) {
std::vector<absl::string_view> name_scopes =
- absl::StrSplit(tf_op.name, kNameScopeSeparator);
+ absl::StrSplit(tf_op_name, kNameScopeSeparator);
// The last element is an op name not TF name scope.
if (!name_scopes.empty()) name_scopes.pop_back();
return name_scopes;
}
+std::vector<absl::string_view> ParseTfNameScopes(const TfOp& tf_op) {
+ return ParseTfNameScopes(tf_op.name);
+}
+
std::string TfOpEventName(const TfOp& tf_op) {
std::string event_name;
if (tf_op.category == Category::kUnknown) {
diff --git a/tensorflow/core/profiler/utils/tf_op_utils.h b/tensorflow/core/profiler/utils/tf_op_utils.h
index 4ad6b45..4ad717b 100644
--- a/tensorflow/core/profiler/utils/tf_op_utils.h
+++ b/tensorflow/core/profiler/utils/tf_op_utils.h
@@ -49,7 +49,8 @@
};
TfOp ParseTfOpFullname(absl::string_view tf_op_fullname);
-// Returns a vector of TF name scopes extracted from tf_op_full_name.
+// Returns a vector of TF name scopes extracted from a TF op name.
+std::vector<absl::string_view> ParseTfNameScopes(absl::string_view tf_op_name);
std::vector<absl::string_view> ParseTfNameScopes(const TfOp& tf_op);
// Trace event name for TF ops is the op type so they have the same color in
