Searched refs:DfsHloVisitorBase (Results 1 – 7 of 7) sorted by relevance
28 Status DfsHloVisitorBase<HloInstructionPtr>::HandleElementwiseUnary( in HandleElementwiseUnary()35 Status DfsHloVisitorBase<HloInstructionPtr>::HandleElementwiseBinary( in HandleElementwiseBinary()42 typename DfsHloVisitorBase<HloInstructionPtr>::VisitState43 DfsHloVisitorBase<HloInstructionPtr>::GetVisitState( in GetVisitState()49 void DfsHloVisitorBase<HloInstructionPtr>::SetVisiting( in SetVisiting()57 void DfsHloVisitorBase<HloInstructionPtr>::SetVisited( in SetVisited()65 Status DfsHloVisitorBase<HloInstructionPtr>::Preprocess(HloInstructionPtr) { in Preprocess()70 Status DfsHloVisitorBase<HloInstructionPtr>::Postprocess(HloInstructionPtr) { in Postprocess()75 template class DfsHloVisitorBase<HloInstruction*>; variable76 template class DfsHloVisitorBase<const HloInstruction*>; variable
60 class DfsHloVisitorBase {68 DfsHloVisitorBase() {} in DfsHloVisitorBase() function69 virtual ~DfsHloVisitorBase() {} in ~DfsHloVisitorBase()347 TF_DISALLOW_COPY_AND_ASSIGN(DfsHloVisitorBase);352 using DfsHloVisitor = DfsHloVisitorBase<HloInstruction*>;353 using ConstDfsHloVisitor = DfsHloVisitorBase<const HloInstruction*>;
281 Status Accept(DfsHloVisitorBase<HloInstructionPtr>* visitor) const;293 Status AcceptOrdered(DfsHloVisitorBase<HloInstructionPtr>* visitor,
40 : public DfsHloVisitorBase<HloInstructionPtr> {
649 DfsHloVisitorBase<HloInstructionPtr>* visitor) const { in Accept()684 DfsHloVisitorBase<HloInstructionPtr>* visitor, in AcceptOrdered()
647 Status Accept(DfsHloVisitorBase<HloInstructionPtr>* visitor,683 Status Visit(DfsHloVisitorBase<HloInstructionPtr>* visitor);
2517 Status HloInstruction::Visit(DfsHloVisitorBase<HloInstructionPtr>* visitor) { in Visit()2796 Status HloInstruction::Accept(DfsHloVisitorBase<HloInstructionPtr>* visitor, in Accept()