Searched refs:hlo_instruction_ (Results 1 – 2 of 2) sorted by relevance
91 : profiler_(profiler), hlo_instruction_(hlo_instruction) { in ScopedInstructionProfiler()97 if (hlo_instruction_ != nullptr) { in ~ScopedInstructionProfiler()98 profiler_->FinishHloInstruction(hlo_instruction_); in ~ScopedInstructionProfiler()104 const HloInstruction* hlo_instruction_; variable
70 : kind_(kind), hlo_instruction_(hlo_instruction) {} in Thunk()76 const HloInstruction* hlo_instruction() const { return hlo_instruction_; } in hlo_instruction()103 const HloInstruction* hlo_instruction_; variable