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()137 return HandleElementwiseUnary(hlo); in HandleAbs()143 return HandleElementwiseUnary(hlo); in HandleRound()146 return HandleElementwiseUnary(hlo); in HandleSign()149 return HandleElementwiseUnary(hlo); in HandleNegate()[all …]
56 Status HandleElementwiseUnary(HloInstruction* hlo) override { in TEST_F() function in xla::__anon1ba6c03c0111::TEST_F::ElementwiseTestVisitor
28 Status DfsHloVisitorBase<HloInstructionPtr>::HandleElementwiseUnary( in HandleElementwiseUnary() function in xla::DfsHloVisitorBase
70 Status HandleElementwiseUnary(HloInstruction* hlo) override;316 Status DynamicDimensionInferenceVisitor::HandleElementwiseUnary( in HandleElementwiseUnary() function in xla::DynamicDimensionInferenceVisitor
52 Status HandleElementwiseUnary(const HloInstruction* hlo) override;
54 Status HandleElementwiseUnary(HloInstructionPtr hlo) override { in HandleElementwiseUnary() function
42 Status HandleElementwiseUnary(HloInstruction* hlo) override;
100 Status ShapeVerifier::HandleElementwiseUnary(HloInstruction* hlo) { in HandleElementwiseUnary() function in xla::ShapeVerifier1386 Status HandleElementwiseUnary(HloInstruction* instruction) override { in HandleElementwiseUnary() function in xla::__anond588ae4e0811::InstructionVerifier
132 Status HloCostAnalysis::HandleElementwiseUnary(const HloInstruction* hlo) { in HandleElementwiseUnary() function in xla::HloCostAnalysis