chainer.function_hooks.CUDAProfileHook

class chainer.function_hooks.CUDAProfileHook

Methods

__enter__()

__exit__(*_)

added(function)

Callback function invoked when the function hook is registered

Parameters:function (FunctionNode) – Function object to which the function hook is added. None if the function hook is registered globally.

backward_postprocess(function, in_data, out_grad)

Callback function invoked after backward propagation.

Parameters:

• function (FunctionNode) – Function object to which the function hook is registered.
• in_data (tuple of numpy.ndarray or tuple of cupy.ndarray) – Input of forward propagation.
• out_grad (tuple of numpy.ndarray or tuple of cupy.ndarray) – Gradient data of backward propagation.

backward_preprocess(function, in_data, out_grad)

Callback function invoked before backward propagation.

Parameters:

• function (FunctionNode) – Function object to which the function hook is registered.
• in_data (tuple of numpy.ndarray or tuple of cupy.ndarray) – Input data of forward propagation.
• out_grad (tuple of numpy.ndarray or tuple of cupy.ndarray) – Gradient data of backward propagation.

deleted(function)

Callback function invoked when the function hook is unregistered

Parameters:function (FunctionNode) – Function object from which the function hook is deleted. None if the function hook was registered globally.

forward_postprocess(function, in_data)

Callback function invoked after forward propagation.

Parameters:

• function (FunctionNode) – Function object to which the function hook is registered.
• in_data (tuple of numpy.ndarray or tuple of cupy.ndarray) – Input data of forward propagation.

forward_preprocess(function, in_data)

Callback function invoked before forward propagation.

Parameters:

• function (FunctionNode) – Function object to which the function hook is registered.
• in_data (tuple of numpy.ndarray or tuple of cupy.ndarray) – Input data of forward propagation.

Attributes

name = 'CUDAProfileHook'