Tracking issue for missing HLO derivatives

[mofeing]

NOTE: Strikethrough ops are deliberately not annotated.

• StableHLO
  • AbsOp
  • AddOp
  • AfterAllOp
  • AllGatherOp
  • AllReduceOp
  • AllToAllOp
  • AndOp
  • Atan2Op Add a bunch of rules for scalar and non-differentiable functions #90
  • BatchNormGradOp
  • BatchNormInferenceOp
  • BatchNormTrainingOp
  • BitcastConvertOp
  • BroadcastInDimOp
  • CaseOp
  • CbrtOp Add a bunch of rules for scalar and non-differentiable functions #90
  • CeilOp Add a bunch of rules for scalar and non-differentiable functions #90
  • CholeskyOp
  • ClampOp Clamp derivatives #148
  • CollectiveBroadcastOp
  • CollectivePermuteOp
  • CompareOp
  • ComplexOp Add a bunch of rules for scalar and non-differentiable functions #90
  • CompositeOp
  • ConcatenateOp
  • ConstantOp
  • ConvertOp
  • ConvolutionOp Convolution backward #178
  • CosineOp
  • ClzOp
  • CustomCallOp
  • DivOp
  • DotGeneralOp
  • DynamicBroadcastInDimOp
  • DynamicConvOp
  • DynamicGatherOp
  • DynamicIotaOp
  • DynamicPadOp
  • DynamicReshapeOp
  • DynamicSliceOp feat: DynamicSliceOp adjoint #220
  • DynamicUpdateSliceOp DynamicUpdateSliceOp reverse derivative #159
  • ExpOp
  • Expm1Op Add a bunch of rules for scalar and non-differentiable functions #90
  • FftOp (current rule is wrong!) Add a bunch of rules for scalar and non-differentiable functions #90
  • FloorOp Add a bunch of rules for scalar and non-differentiable functions #90
  • GatherOp
  • GetDimensionSizeOp
  • GetTupleElementOp
  • IfOp IfOp #149
  • ImagOp implement derivative for real and imag #145
  • InfeedOp
  • IotaOp Add a bunch of rules for scalar and non-differentiable functions #90
  • IsFiniteOp Add a bunch of rules for scalar and non-differentiable functions #90
  • LogOp
  • Logp1Op Add a bunch of rules for scalar and non-differentiable functions #90
  • LogisticOp Add a bunch of rules for scalar and non-differentiable functions #90
  • MapOp
  • MaxOp
  • MinOp Add a bunch of rules for scalar and non-differentiable functions #90
  • MulOp
  • NegateOp
  • NotOp
  • OptimizationBarrierOp
  • OrOp
  • OutfeedOp
  • PadOp
  • PartitionIdOp
  • PopcntOp
  • PowOp
  • RealOp implement derivative for real and imag #145
  • RecvOp
  • ReduceOp
  • ReducePrecisionOp
  • ReduceScatterOp
  • ReduceWindowOp
  • RemainderOp
  • ReplicaIdOp
  • ReshapeOp
  • ReverseOp Add a bunch of rules for scalar and non-differentiable functions #90
  • RngOp Add a bunch of rules for scalar and non-differentiable functions #90
  • RngBitGeneratorOp Add a bunch of rules for scalar and non-differentiable functions #90
  • RoundOp Add a bunch of rules for scalar and non-differentiable functions #90
  • RoundNearestEvenOp Add a bunch of rules for scalar and non-differentiable functions #90
  • RsqrtOp
  • ScatterOp
  • SelectOp
  • SelectAndScatterOp
  • SendOp
  • ShiftLeftOp
  • ShiftRightArithmeticOp
  • ShiftRightLogicalOp
  • SignOp Add a bunch of rules for scalar and non-differentiable functions #90
  • SineOp
  • SliceOp
  • SortOp
  • SqrtOp
  • SubtractOp
  • TanhOp
  • TransposeOp
  • TriangularSolveOp
  • TupleOp #157
  • UniformDequantizeOp
  • UniformQuantizeOp
  • WhileOp WhileOp reverse derivative #160
  • XorOp
  • Deprecated operations in StableHLO
    • BroadcastOp
    • CreateTokenOp
    • CrossReplicaSumOp
    • DotOp
    • EinsumOp
    • TorchIndexSelectOp
    • UnaryEinsumOp
• CHLO
  • Binary Element-wise Operations
    • BroadcastAddOp
    • BroadcastAtan2Op
    • BroadcastDivOp
    • BroadcastMaxOp
    • BroadcastMinOp
    • BroadcastMulOp
    • BroadcastNextAfterOp
    • BroadcastPolygammaOp
    • BroadcastPowOp
    • BroadcastRemOp
    • BroadcastShiftLeftOp
    • BroadcastShiftRightArithmeticOp
    • BroadcastShiftRightLogicalOp
    • BroadcastSubOp
    • BroadcastZetaOp
  • Binary Logical Element-wise Operations
    • BroadcastAndOp
    • BroadcastOrOp
    • BroadcastXorOp
  • Non-broadcasting Binary Operations
    • NextAfterOp
    • PolygammaOp Add a bunch of rules for scalar and non-differentiable functions #90
    • ZetaOp
  • ComplexOp
  • Unary Element-wise Operations
    • AcosOp Add a bunch of rules for scalar and non-differentiable functions #90
    • AcoshOp Add a bunch of rules for scalar and non-differentiable functions #90
    • AsinOp Add a bunch of rules for scalar and non-differentiable functions #90
    • AsinhOp Add a bunch of rules for scalar and non-differentiable functions #90
    • AtanOp Add a bunch of rules for scalar and non-differentiable functions #90
    • AtanhOp Add a bunch of rules for scalar and non-differentiable functions #90
    • BesselI1eOp
    • ConjOp Add a bunch of rules for scalar and non-differentiable functions #90
    • CoshOp Add a bunch of rules for scalar and non-differentiable functions #90
    • SinhOp Add a bunch of rules for scalar and non-differentiable functions #90
    • TanOp Add a bunch of rules for scalar and non-differentiable functions #90
    • ConstantOp (shared with StableHLO_ConstantOp)
    • ConstantLikeOp
    • DigammaOp Add a bunch of rules for scalar and non-differentiable functions #90
    • ErfOp
    • ErfInvOp
    • ErfcOp
    • IsInfOp Add a bunch of rules for scalar and non-differentiable functions #90
    • IsNegInfOp Add a bunch of rules for scalar and non-differentiable functions #90
    • IsPosInfOp Add a bunch of rules for scalar and non-differentiable functions #90
    • LgammaOp
  • BroadcastCompareOp
  • BroadcastSelectOp
  • TopKOp