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

Use string_view to get frame names from GetRootFrame().

Browse Source
This commit is contained in:
Trent Lo 2019-05-21 10:20:12 -07:00
parent 83909d218f
commit c58c4aa634
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/jit/deadness_analysis.cc
View File

@ -958,7 +958,7 @@ Status GetFullFrame(const Node* n, absl::Span<const ControlFlowInfo> cfi_infos,
// If the node is inside some frames, get the name of the outermost non-empty // If the node is inside some frames, get the name of the outermost non-empty
// frame. Otherwise, get an empty frame name. // frame. Otherwise, get an empty frame name.
Status GetRootFrame(const Node* n, absl::Span<const ControlFlowInfo> cfi_infos, Status GetRootFrame(const Node* n, absl::Span<const ControlFlowInfo> cfi_infos,
string* frame) { absl::string_view* frame) {
int depth = 0; int depth = 0;
const ControlFlowInfo* cfi_iter = &cfi_infos[n->id()]; const ControlFlowInfo* cfi_iter = &cfi_infos[n->id()];
while (!cfi_iter->parent_frame->IsSource()) { while (!cfi_iter->parent_frame->IsSource()) {
@ -1257,12 +1257,12 @@ Status DeadnessAnalysisImpl::Populate(bool force_pessimistic) {
size_t frame_start = 0; size_t frame_start = 0;
while (frame_start < topo.size()) { while (frame_start < topo.size()) {
// Batching nodes who have the same root frame. // Batching nodes who have the same root frame.
string cur_frame_name; absl::string_view cur_frame_name;
TF_RETURN_IF_ERROR( TF_RETURN_IF_ERROR(
GetRootFrame(topo[frame_start], control_flow_info_, &cur_frame_name)); GetRootFrame(topo[frame_start], control_flow_info_, &cur_frame_name));
size_t frame_end = frame_start; size_t frame_end = frame_start;
for (size_t i = frame_start + 1; i < topo.size(); ++i) { for (size_t i = frame_start + 1; i < topo.size(); ++i) {
string i_frame_name; absl::string_view i_frame_name;
TF_RETURN_IF_ERROR( TF_RETURN_IF_ERROR(
GetRootFrame(topo[i], control_flow_info_, &i_frame_name)); GetRootFrame(topo[i], control_flow_info_, &i_frame_name));
if (i_frame_name == cur_frame_name) { if (i_frame_name == cur_frame_name) {