def broadcast_checked.repad

bendtorch-core@0.1.0.0 · src/shape/core.bend:872

import bendtorch-core@0.1.0.0/src/shape/core.bend as Core
import Base

872def broadcast_checked.repad(+left_padding: Nat, +right_padding: Nat, -left: List<&2, U32>, -right: List<&2, U32>, -output: List<&2, U32>, aligned: Aligned(naturals(pad_words(left_padding, left)), naturals(pad_words(right_padding, right)), naturals(output))) -> Aligned(pad(left_padding, naturals(left)), pad(right_padding, naturals(right)), naturals(output)):
873  %naturals_pad(left_padding, left) : Aligned(_, pad(right_padding, naturals(right)), naturals(output))
874  %naturals_pad(right_padding, right) : Aligned(naturals(pad_words(left_padding, left)), _, naturals(output))
875  aligned

Uses (8)

Used by (4)