Searched refs:HloInstructionPtr (Results 1 – 6 of 6) sorted by relevance
59 template <typename HloInstructionPtr>62 std::is_same<HloInstruction*, HloInstructionPtr>::value ||63 std::is_same<const HloInstruction*, HloInstructionPtr>::value,74 virtual Status HandleElementwiseUnary(HloInstructionPtr hlo);75 virtual Status HandleElementwiseBinary(HloInstructionPtr hlo);77 virtual Status HandleClamp(HloInstructionPtr hlo) = 0;78 virtual Status HandleSelect(HloInstructionPtr hlo) = 0;79 virtual Status HandleTupleSelect(HloInstructionPtr hlo) = 0;80 virtual Status HandleMaximum(HloInstructionPtr hlo) { in HandleMaximum()83 virtual Status HandleMinimum(HloInstructionPtr hlo) { in HandleMinimum()[all …]
45 template <typename HloInstructionPtr>47 : public DfsHloVisitorBase<HloInstructionPtr> {53 virtual Status DefaultAction(HloInstructionPtr hlo_instruction) = 0;55 Status HandleElementwiseUnary(HloInstructionPtr hlo) override { in HandleElementwiseUnary()58 Status HandleElementwiseBinary(HloInstructionPtr hlo) override { in HandleElementwiseBinary()62 Status HandleBatchNormTraining(HloInstructionPtr hlo) override { in HandleBatchNormTraining()66 Status HandleBatchNormInference(HloInstructionPtr hlo) override { in HandleBatchNormInference()70 Status HandleBatchNormGrad(HloInstructionPtr hlo) override { in HandleBatchNormGrad()74 Status HandleClamp(HloInstructionPtr clamp) override { in HandleClamp()77 Status HandleConcatenate(HloInstructionPtr concatenate) override { in HandleConcatenate()[all …]
27 template <typename HloInstructionPtr>28 Status DfsHloVisitorBase<HloInstructionPtr>::HandleElementwiseUnary( in HandleElementwiseUnary()29 HloInstructionPtr hlo) { in HandleElementwiseUnary()34 template <typename HloInstructionPtr>35 Status DfsHloVisitorBase<HloInstructionPtr>::HandleElementwiseBinary( in HandleElementwiseBinary()36 HloInstructionPtr hlo) { in HandleElementwiseBinary()41 template <typename HloInstructionPtr>42 typename DfsHloVisitorBase<HloInstructionPtr>::VisitState43 DfsHloVisitorBase<HloInstructionPtr>::GetVisitState( in GetVisitState()48 template <typename HloInstructionPtr>[all …]
365 template <typename HloInstructionPtr>366 Status Accept(DfsHloVisitorBase<HloInstructionPtr>* visitor) const;377 template <typename HloInstructionPtr>378 Status AcceptOrdered(DfsHloVisitorBase<HloInstructionPtr>* visitor,566 template <typename HloInstructionPtr>568 DfsHloVisitorBase<HloInstructionPtr>* visitor) const { in Accept()585 template <typename HloInstructionPtr>587 DfsHloVisitorBase<HloInstructionPtr>* visitor, in AcceptOrdered()
1232 template <typename HloInstructionPtr>1233 Status Accept(DfsHloVisitorBase<HloInstructionPtr>* visitor,1252 template <typename HloInstructionPtr>1253 Status Visit(DfsHloVisitorBase<HloInstructionPtr>* visitor);
3055 template <typename HloInstructionPtr>3056 Status HloInstruction::Visit(DfsHloVisitorBase<HloInstructionPtr>* visitor) { in Visit()3395 template <typename HloInstructionPtr>3396 Status HloInstruction::Accept(DfsHloVisitorBase<HloInstructionPtr>* visitor, in Accept()