experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Rolling back the rollback with a fix for Windows

Browse Source 
PiperOrigin-RevId: 352109875
Change-Id: Ia154e7bc1b120acea59f1e241f9c1aa596d18cd8
This commit is contained in:
George Karpenkov 2021-01-15 16:57:43 -08:00 committed by TensorFlower Gardener
parent 2c4e6e357a
commit 569a996c09
3 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/util/BUILD
View File

@ -481,6 +481,7 @@ cc_library(
],
deps = [
"//tensorflow/core/platform:status",
"@com_google_absl//absl/strings",
],
)

11
tensorflow/core/util/managed_stack_trace.h
View File

@ -18,6 +18,7 @@ limitations under the License.
#include <string>
#include "absl/strings/match.h"
#include "tensorflow/core/platform/status.h"
namespace tensorflow {
@ -33,6 +34,16 @@ using ToStackFramesFunctor = std::vector<StackFrame>(int, const StackTraceMap&,
const StackTraceFilter&,
bool, int);
// Returns whether the given frame is internal to TF.
inline bool IsInternalFrameForFilename(absl::string_view file_name) {
// Use a simple heuristic for now.
// TODO(cheshire): Build a more sophisticated mechanism, rely on @tf.export.
return (absl::StrContains(file_name, "tensorflow/python") ||
absl::StrContains(file_name, "tensorflow\\python")) &&
!absl::StrContains(file_name, "keras") &&
!absl::StrContains(file_name, "test.py");
}
// Language agnostic stack trace class. It only saves an id, and language
// clients are responsible for managing the actual stack trace objects.
class ManagedStackTrace {

9
tensorflow/python/util/tf_stack.cc
View File

@ -226,15 +226,6 @@ class StackTraceWrapper : public AbstractStackTrace {
});
}
static bool IsInternalFrameForFilename(absl::string_view file_name) {
// Use a simple heuristic for now.
// TODO(cheshire): Build a more sophisticated mechanism, rely on @tf.export.
return (absl::StrContains(file_name, "tensorflow/python") ||
absl::StrContains(file_name, "tensorflow\\python")) &&
!absl::StrContains(file_name, "keras") &&
!absl::StrContains(file_name, "test.py");
}
absl::optional<StackFrame> StackTraceMapping(SourceLoc loc) const {
if (source_map_->contains(loc)) {
return source_map_->at(loc);