Lower level of python TraceMe to 1
PiperOrigin-RevId: 313802412
Change-Id: Ia7d3e0866f13676b6562f3290ca8385a1a37c312
diff --git a/tensorflow/python/profiler/internal/traceme_wrapper.h b/tensorflow/python/profiler/internal/traceme_wrapper.h
index c074e90..6e61d9c 100644
--- a/tensorflow/python/profiler/internal/traceme_wrapper.h
+++ b/tensorflow/python/profiler/internal/traceme_wrapper.h
@@ -34,13 +34,15 @@
// pybind11::str and pybind11::kwargs are taken by const reference to avoid
// python reference-counting overhead.
TraceMeWrapper(const pybind11::str& name, const pybind11::kwargs& kwargs)
- : traceme_([&]() {
- std::string name_and_metadata(name);
- if (!kwargs.empty()) {
- AppendMetadata(&name_and_metadata, kwargs);
- }
- return name_and_metadata;
- }) {}
+ : traceme_(
+ [&]() {
+ std::string name_and_metadata(name);
+ if (!kwargs.empty()) {
+ AppendMetadata(&name_and_metadata, kwargs);
+ }
+ return name_and_metadata;
+ },
+ /*level=*/1) {}
// pybind11::kwargs is taken by const reference to avoid python
// reference-counting overhead.
