vta.top.bitpack

Bit packing operators

Functions

bitpack(data, bits[, pack_type, name])	Packs lowest dimension into format needed by VTA
compute_bitpack(attrs, inputs)

Module Contents

vta.top.bitpack.bitpack(data, bits, pack_type='int8', name='bitpack')

Packs lowest dimension into format needed by VTA

Parameters

pack_axisint

index of the axis to pack in data

bit_axisint

index of axis to place bit axis in resulting packed data

Returns

packedTensor

The packed tensor.

vta.top.bitpack.compute_bitpack(attrs, inputs)