Modified code generators to support ops_arg_idx
ops_arg_idx treatment