chainer.function_hooks.CUDAProfileHook

class chainer.function_hooks.CUDAProfileHook[source]

Methods

__enter__()[source]
__exit__(*_)[source]
added(function)[source]

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)[source]

Callback function invoked after backward propagation.

Parameters
backward_preprocess(function, in_data, out_grad)[source]

Callback function invoked before backward propagation.

Parameters
deleted(function)[source]

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)[source]

Callback function invoked after forward propagation.

Parameters
  • function (FunctionNode) – Function object to which the function hook is registered.

  • in_data (tuple of N-dimensional array) – Input data of forward propagation.

forward_preprocess(function, in_data)[source]

Callback function invoked before forward propagation.

Parameters
  • function (FunctionNode) – Function object to which the function hook is registered.

  • in_data (tuple of N-dimensional array) – Input data of forward propagation.

__eq__(value, /)

Return self==value.

__ne__(value, /)

Return self!=value.

__lt__(value, /)

Return self<value.

__le__(value, /)

Return self<=value.

__gt__(value, /)

Return self>value.

__ge__(value, /)

Return self>=value.

Attributes

name = 'CUDAProfileHook'