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
58 class DfsHloVisitorBase {66 DfsHloVisitorBase() {} in DfsHloVisitorBase() function67 virtual ~DfsHloVisitorBase() {} in ~DfsHloVisitorBase()389 DfsHloVisitorBase(const DfsHloVisitorBase&) = delete;390 DfsHloVisitorBase& operator=(const DfsHloVisitorBase&) = delete;394 extern template class DfsHloVisitorBase<HloInstruction*>;395 extern template class DfsHloVisitorBase<const HloInstruction*>;399 using DfsHloVisitor = DfsHloVisitorBase<HloInstruction*>;400 using ConstDfsHloVisitor = DfsHloVisitorBase<const HloInstruction*>;
469 Status Accept(DfsHloVisitorBase<HloInstructionPtr>* visitor) const;481 Status AcceptOrdered(DfsHloVisitorBase<HloInstructionPtr>* visitor,762 DfsHloVisitorBase<HloInstructionPtr>* visitor) const { in Accept()781 DfsHloVisitorBase<HloInstructionPtr>* visitor, in AcceptOrdered()
45 : public DfsHloVisitorBase<HloInstructionPtr> {
1406 Status Accept(DfsHloVisitorBase<HloInstructionPtr>* visitor,1426 Status Visit(DfsHloVisitorBase<HloInstructionPtr>* visitor);
3473 Status HloInstruction::Visit(DfsHloVisitorBase<HloInstructionPtr>* visitor) { in Visit()3822 Status HloInstruction::Accept(DfsHloVisitorBase<HloInstructionPtr>* visitor, in Accept()
394 xla::DfsHloVisitorBase::SetVisited395 xla::DfsHloVisitorBase<class xla::HloInstruction.*>::SetVisited