Deeploy.TilingExtension.TilingCodegen

Functions

calculateFlatOffset(offsets, strides)

Calculate the flat memory offset from multi-dimensional coordinates.

calculateFlatOffsetInBytes(tile, referenceBuffer)

Calculate the flat memory offset in bytes for a hyperrectangle tile.

computeTileHyperRectangles(memoryTransfer)

Compute hyperrectangle tiles for a memory transfer operation.

minimizeRectangle(rect, referenceShape)

Minimize a hyperrectangle by collapsing dimensions where possible.

minimizeVariableReplacement(scheme, ...)

Optimize a variable replacement scheme by eliminating constant replacements.

padOffset(offset, rank)

Pad an offset tuple to a target rank by prepending zeros.

padShape(shape, rank)

Pad a shape tuple to a target rank by prepending ones.

padStride(stride, rank, paddingStride)

Pad a stride tuple to a target rank by prepending a specified stride value.

stridesFromShape(shape)

Calculate memory strides from a tensor shape.

Classes

AbsoluteHyperRectangle(rectangle, absoluteOffset)

Represents a HyperRectangle with an absolute offset in memory space.

HyperRectangle(offset, dims)

Represents a multi-dimensional rectangular region in tensor space.

MemoryTransfer(source, destination)

Represents a memory transfer operation between two memory levels.

TilingSchedule(inputBaseOffsets, ...)

Represents a complete schedule for tiled execution of neural network operations.

VariableReplacementScheme(...)

Defines how variables should be replaced with tile-specific values.