Searched refs:HandleElementwiseBinary (Results 1 – 9 of 9) sorted by relevance
75 virtual Status HandleElementwiseBinary(HloInstructionPtr hlo);81 return HandleElementwiseBinary(hlo); in HandleMaximum()84 return HandleElementwiseBinary(hlo); in HandleMinimum()97 return HandleElementwiseBinary(hlo); in HandleComplex()100 return HandleElementwiseBinary(hlo); in HandleMultiply()104 return HandleElementwiseBinary(hlo); in HandlePower()135 return HandleElementwiseBinary(hlo); in HandleCompare()138 return HandleElementwiseBinary(hlo); in HandleAdd()141 return HandleElementwiseBinary(hlo); in HandleDivide()144 return HandleElementwiseBinary(hlo); in HandleRemainder()[all …]
50 Status HandleElementwiseBinary(HloInstruction* hlo) override { in TEST_F() function in xla::__anonbb86b51f0111::TEST_F::ElementwiseTestVisitor
35 Status DfsHloVisitorBase<HloInstructionPtr>::HandleElementwiseBinary( in HandleElementwiseBinary() function in xla::DfsHloVisitorBase
54 Status HandleElementwiseBinary(const HloInstruction* hlo) override;
58 Status HandleElementwiseBinary(HloInstructionPtr hlo) override { in HandleElementwiseBinary() function
45 Status HandleElementwiseBinary(HloInstruction* hlo) override;
133 Status HandleElementwiseBinary(HloInstruction* hlo) override;814 Status DynamicDimensionInferenceVisitor::HandleElementwiseBinary( in HandleElementwiseBinary() function in xla::DynamicDimensionInferenceVisitor
134 Status ShapeVerifier::HandleElementwiseBinary(HloInstruction* hlo) { in HandleElementwiseBinary() function in xla::ShapeVerifier2204 Status HandleElementwiseBinary(HloInstruction* instruction) override { in HandleElementwiseBinary() function in xla::__anonba98a3510d11::InstructionVerifier
192 Status HloCostAnalysis::HandleElementwiseBinary(const HloInstruction* hlo) { in HandleElementwiseBinary() function in xla::HloCostAnalysis