Util

Utility functions for AutoTVM dialects

lorien.dialect.tvm_dial.autotvm_dial.util.infer_task_layout(record: Tuple[tvm.autotvm.measure.measure.MeasureInput, tvm.autotvm.measure.measure.MeasureResult]) → Optional[Tuple]

Infer the layout of the given task. Return None if the layout cannot be inferred.

Parameters

record (Tuple[MeasureInput, MeasureResult]) -- The AutoTVM record pair.

Returns

layout -- A tuple of input and output layout, or None if not inferrable.

Return type

Optional[Tuple]