Example #1
0
 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)