def forward(ctx, features, filters, indice_pairs, indice_pair_num, num_activate_out, algo): ctx.save_for_backward(indice_pairs, indice_pair_num, features, filters) ctx.algo = algo return ops.indice_conv(features, filters, indice_pairs, indice_pair_num, num_activate_out, False, algo=algo)
def forward(ctx, features, filters, indice_pairs, indice_pair_num, num_activate_out): ctx.save_for_backward(indice_pairs, indice_pair_num, features, filters) return ops.indice_conv(features, filters, indice_pairs, indice_pair_num, num_activate_out, True, False)