trw.layers.ops_conversion

Module Contents

Classes

OpsConversion

Helper to create standard N-d operations

Functions

upsample_mode(mode: typing_extensions.Literal[nearest, linear], dim: int) → trw.basic_typing.ModuleCreator

trw.layers.ops_conversion.upsample_mode(mode: typing_extensions.Literal[nearest, linear], dim: int) → trw.basic_typing.ModuleCreator

class trw.layers.ops_conversion.OpsConversion(upsample_mode: typing_extensions.Literal[nearest, linear] = 'linear')

Helper to create standard N-d operations

set_dim(self, dim: int)