pytorch_quantization.utils.amp_wrapper Namespace Reference

Functions
def	half_function (fn)
def	float_function (fn)
def	promote_function (fn)

Function Documentation

half_function()

def pytorch_quantization.utils.amp_wrapper.half_function

(

fn

)

Here is the caller graph for this function:

float_function()

def pytorch_quantization.utils.amp_wrapper.float_function

(

fn

)

Here is the caller graph for this function:

promote_function()

def pytorch_quantization.utils.amp_wrapper.promote_function

(

fn

)

Here is the call graph for this function: