Ejemplo n.º 1
0
def generate_c2_gradient_test(configs, c2_bench_op):
    """ This function creates Caffe2 op test based on the given operator
    """
    _generate_test(configs,
                   c2_bench_op,
                   register_caffe2_op_test_case,
                   run_backward=True)
Ejemplo n.º 2
0
def generate_c2_test(configs, c2_bench_op):
    """ This function creates Caffe2 op test based on the given operator
    """
    return _generate_test(configs,
                          c2_bench_op,
                          create_caffe2_op_test_case,
                          run_backward=False)