Searched refs:set_visiting_hlo (Results 1 – 2 of 2) sorted by relevance
83 void set_visiting_hlo(HloInstruction* hlo) { visiting_hlo_ = hlo; } in set_visiting_hlo() function
1409 b_.set_visiting_hlo(hlo); in Preprocess()1456 b_.set_visiting_hlo(nullptr); in Postprocess()