@switch_to_static_graph
def _clone_for_test(self, main_program):
    return main_program.clone(for_test=True)
@switch_to_static_graph
def _clone_for_test(self, main_program):
    return main_program.clone(for_test=True)