Searched refs:HandleElementwiseBinary (Results 1 – 10 of 10) sorted by relevance
73 virtual Status HandleElementwiseBinary(HloInstructionPtr hlo);78 return HandleElementwiseBinary(hlo); in HandleMaximum()81 return HandleElementwiseBinary(hlo); in HandleMinimum()94 return HandleElementwiseBinary(hlo); in HandleComplex()97 return HandleElementwiseBinary(hlo); in HandleMultiply()101 return HandleElementwiseBinary(hlo); in HandlePower()133 return HandleElementwiseBinary(hlo); in HandleCompare()136 return HandleElementwiseBinary(hlo); in HandleAdd()139 return HandleElementwiseBinary(hlo); in HandleDivide()142 return HandleElementwiseBinary(hlo); in HandleRemainder()[all …]
50 Status HandleElementwiseBinary(HloInstruction* hlo) override { in TEST_F() function in xla::__anon97ab43290111::TEST_F::ElementwiseTestVisitor
35 Status DfsHloVisitorBase<HloInstructionPtr>::HandleElementwiseBinary( in HandleElementwiseBinary() function in xla::DfsHloVisitorBase
90 Status HandleElementwiseBinary(const HloInstruction* hlo) override;
56 Status HandleElementwiseBinary(HloInstructionPtr hlo) override { in HandleElementwiseBinary() function
142 Status HandleElementwiseBinary(HloInstruction* hlo) override;
229 Status HandleElementwiseBinary(HloInstruction* hlo) override { in HandleElementwiseBinary() function in xla::__anon52bdec210111::LayoutNormalizationVisitor
149 Status HandleElementwiseBinary(HloInstruction* hlo) override;904 Status DynamicDimensionInferenceVisitor::HandleElementwiseBinary( in HandleElementwiseBinary() function in xla::DynamicDimensionInferenceVisitor
162 Status ShapeVerifier::HandleElementwiseBinary(HloInstruction* hlo) { in HandleElementwiseBinary() function in xla::ShapeVerifier2445 Status HandleElementwiseBinary(HloInstruction* instruction) override { in HandleElementwiseBinary() function in xla::__anon411d6f1b0e11::InstructionVerifier
197 Status HloCostAnalysis::HandleElementwiseBinary(const HloInstruction* hlo) { in HandleElementwiseBinary() function in xla::HloCostAnalysis