Searched refs:HandleElementwiseUnary (Results 1 – 9 of 9) sorted by relevance
74 virtual Status HandleElementwiseUnary(HloInstructionPtr hlo);88 return HandleElementwiseUnary(hlo); in HandleConvert()91 return HandleElementwiseUnary(hlo); in HandleBitcastConvert()94 return HandleElementwiseUnary(hlo); in HandleCopy()107 return HandleElementwiseUnary(hlo); in HandleSqrt()110 return HandleElementwiseUnary(hlo); in HandleRsqrt()113 return HandleElementwiseUnary(hlo); in HandleCbrt()145 return HandleElementwiseUnary(hlo); in HandleAbs()151 return HandleElementwiseUnary(hlo); in HandleRound()154 return HandleElementwiseUnary(hlo); in HandleLogistic()[all …]
56 Status HandleElementwiseUnary(HloInstruction* hlo) override { in TEST_F() function in xla::__anon291407da0111::TEST_F::ElementwiseTestVisitor
28 Status DfsHloVisitorBase<HloInstructionPtr>::HandleElementwiseUnary( in HandleElementwiseUnary() function in xla::DfsHloVisitorBase
53 Status HandleElementwiseUnary(const HloInstruction* hlo) override;
55 Status HandleElementwiseUnary(HloInstructionPtr hlo) override { in HandleElementwiseUnary() function
44 Status HandleElementwiseUnary(HloInstruction* hlo) override;
131 Status HandleElementwiseUnary(HloInstruction* hlo) override;786 Status DynamicDimensionInferenceVisitor::HandleElementwiseUnary( in HandleElementwiseUnary() function in xla::DynamicDimensionInferenceVisitor
94 Status ShapeVerifier::HandleElementwiseUnary(HloInstruction* hlo) { in HandleElementwiseUnary() function in xla::ShapeVerifier1848 Status HandleElementwiseUnary(HloInstruction* instruction) override { in HandleElementwiseUnary() function in xla::__anon45533f2c0811::InstructionVerifier
188 Status HloCostAnalysis::HandleElementwiseUnary(const HloInstruction* hlo) { in HandleElementwiseUnary() function in xla::HloCostAnalysis