TrainingExtensions/torch/test/python/test_model_preparer.py
def custom_function_not_to_be_traced(x, y):
    """ Function which we do not want to be traced, when traced using torch FX API, call to this function will
    be inserted as call_function, and won't be traced through """
    for i in range(2):
        x += x
        y += y
    return x * x + y * y
 Function which we do not want to be traced, when traced using torch FX API, call to this function will
    be inserted as call_function, and won't be traced through 
Function which we do not want to be traced, when traced using torch FX API, call to this function will
be inserted as call_function, and won't be traced through
Function which we do not want to be traced when traced using torch FX API call to this function will be inserted as call_function and won ' t be traced through
def custom_function_not_to_be_traced(x, y):
    for i in range(2):
        x += x
        y += y
    return x * x + y * y
def custom_function_not_to_be_traced ( x , y ) : for i in range ( 2 ) : x += x y += y return x * x + y * y
Function which we do not want to be traced, when traced using torch FX API, call to this function will
be inserted as call_function, and won't be traced through
Function which we do not want to be traced when traced using torch FX API call to this function will be inserted as call_function and won ' t be traced through
""" Function which we do not want to be traced, when traced using torch FX API, call to this function will
    be inserted as call_function, and won't be traced through """
x
y