chainer.function_hooks.CUDAProfileHook¶
- class chainer.function_hooks.CUDAProfileHook[source]¶
Methods
- 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
function (FunctionNode) – Function object to which the function hook is registered.
in_data (tuple of N-dimensional array) – Input of forward propagation.
out_grad (tuple of N-dimensional array) – Gradient data of backward propagation.
- backward_preprocess(function, in_data, out_grad)[source]¶
Callback function invoked before backward 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.
out_grad (tuple of N-dimensional array) – Gradient data of backward propagation.
- 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'¶