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

Update tensorflow/core/profiler/internal/tfprof_code.cc

Browse Source 
Co-authored-by: Mihai Maruseac <mihai.maruseac@gmail.com>
This commit is contained in:
tg-at-google 2020-07-28 20:59:05 -04:00 committed by GitHub
parent 5bbedc7ec0
commit 3fb90eb802
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

4
tensorflow/core/profiler/internal/tfprof_code.cc
View File

@ -452,8 +452,8 @@ void TFCode::Build() {
GetTraceString(fn->call_stack()->traces().at(i)) + kGradientSuffix;
pre_code_node = pre_code_node->AddChildren(
trace, &fn->call_stack()->traces().at(i), kGradientSuffix);
const int64 max_stack_trace_allowed_size = fn->call_stack()->traces().size() - 1;
if (i == max_stack_trace_allowed_size) {
const int64 last_trace = fn->call_stack()->traces().size() - 1;
if (i == last_trace) {
leaf = pre_code_node;
}
}