fastvideo.v1.layers.custom_op
#
Module Contents#
Classes#
Base class for custom ops. Dispatches the forward method to the appropriate backend. |
Data#
API#
- class fastvideo.v1.layers.custom_op.CustomOp[source]#
Bases:
torch.nn.Module
Base class for custom ops. Dispatches the forward method to the appropriate backend.
Initialization
Initialize internal Module state, shared by both nn.Module and ScriptModule.
- abstract static default_on() bool [source]#
On by default if level < CompilationLevel.PIECEWISE Specifying βallβ or βnoneβ in custom_op takes precedence.