For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo struct
ROIAlign
struct ROIAlign
Registers the mo.roi_align graph op with the graph compiler.
Implemented traits
Methods
execute
static def execute[aligned: Bool, mode: StringSlice[ImmStaticOrigin], dtype: DType](output: ManagedTensorSlice[IOSpec[_, _].Output, static_spec=output.static_spec], input: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input.static_spec], rois: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=rois.static_spec], output_height: Int64, output_width: Int64, spatial_scale: Scalar, sampling_ratio: Scalar)