Skip to main content

tile_patch_size

Function tile_patch_size 

Source
pub fn tile_patch_size(cfg: &TileConfig, divisor: usize) -> usize
Expand description

Padded input-patch size per axis for a config and the net’s size_divisor: core + 2·halo rounded up to a multiple of divisor. Isotropic, so one value serves all axes. Callers use this to build a reusable OnnxModel::plan_for plan of shape [1, 1, p, p, p].