Searched refs:HloTraceInstruction (Results 1 – 3 of 3) sorted by relevance
1106 HloTraceInstruction::HloTraceInstruction(const string& tag, in HloTraceInstruction() function in xla::HloTraceInstruction1114 HloInstructionProto HloTraceInstruction::ToProto() const { in ToProto()1120 bool HloTraceInstruction::IdenticalSlowPath( in IdenticalSlowPath()1127 std::unique_ptr<HloInstruction> HloTraceInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()
685 class HloTraceInstruction : public HloInstruction {687 explicit HloTraceInstruction(const string& tag, HloInstruction* operand);
645 return absl::make_unique<HloTraceInstruction>(tag, operand); in CreateTrace()3285 return Cast<HloTraceInstruction>(this)->TracingTag(); in TracingTag()