if (args.size() != 2) return false;
  if (!absl::StrContains(equation_attr.str(), "->")) return false;

  // Check the types and ranks of the input arguments.
  if (!IsRankedInt(args[0], 32)) return false;
  if (!IsRankedInt(args[1], 32)) return false;

  // Trace the graph to see if the conversion is applicable.
  Operation *op_input = args[0].getDefiningOp();
  Operation *op_weight = args[1].getDefiningOp();