Mojo module
conv_transpose
Structs
Functions
-
accumulate_wo_tile
: -
conv_transpose_naive
: Implements the ConvTranspose operator from the MO spec. -
conv_transpose_shape
: Compute the output shape of aconv-transpose
operation, and assert the inputs are compatible. -
conv_transposed
: -
get_num_partitions
: Partition the worload in (batch&group, C, F, H) dimensions. HOWO is the combination of HO and WO dimensions. The actual number of tasks are the product of return num_partitions. -
get_partition
: -
pack_filter
: This packs the filter form RSFC to FRSCf. -
pack_filter_shape
: Compute the output shape of transposed convolution filter packing. -
update_w_tile_2d
: -
update_w_tile_3d
:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!