コード例 #1
0
ファイル: benchmark_caffe2.py プロジェクト: z1b1y/pytorch 
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)
コード例 #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)