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__
()¶ Return self==value.
-
__ne__
()¶ Return self!=value.
-
__lt__
()¶ Return self<value.
-
__le__
()¶ Return self<=value.
-
__gt__
()¶ Return self>value.
-
__ge__
()¶ Return self>=value.
Attributes
-
name
= 'CUDAProfileHook'¶
-