(FloatValueEquals<"0"> $zero),
    (FloatValueEquals<"1"> $one),
    (HasOneUse $select_out),
    (HasOneUse $exp_out),
    (HasOneUse $sub_out),
  ]>;


def isF32Splat : Constraint<
  CPred<"IsF32Splat($0)">>;

def ExtractF32AtIndex0: NativeCodeCall<
    "$_builder.getF32FloatAttr($_self.cast<DenseElementsAttr>().getValues<float>()[0])">;

def FuseLeakyReluConst : Pat<
  (TFL_SelectOp

  DenseElementsAttr vals;
  if (!matchPattern(value, m_Constant(&vals))) return false;
  for (auto elem : vals.getValues<float>())
    if (elem != 0.0f) return false;
  return true;
}

bool IsF32Splat(Attribute input_splat) {
  if (!input_splat) return false;
  auto val = dyn_cast_or_null<DenseElementsAttr>(input_splat);
  if (val) {
    return !val.empty() && val.getElementType().isF32() && val.isSplat();
  }