afe.apis.compilation_job

The API for controlling a compilation job. This API is intended for use in a user-defined module that is passed to the command-line interface.

The recommended usage is to pass a CompilationJob to set_compilation_job. Parameters that direct compilation are incorporated into the CompilationJob. The job will be read by the command-line driver.

Classes

CompilationJob

A specification of how to calibrate, quantize, evaluate,

Functions

set_compilation_job(→ None)

Use the given CompilationJob to control compilation. If called multiple

Module Contents

class afe.apis.compilation_job.CompilationJob

A specification of how to calibrate, quantize, evaluate, and compile a model.

preprocess_transforms: List[afe.apis.transform.Transform]

postprocess_transforms: List[afe.apis.transform.Transform]

calibration_input: Iterable[afe.apis.compilation_job_base.Tensors]

evaluation_input: Iterable[Tuple[afe.apis.compilation_job_base.Tensors, afe.apis.compilation_job_base.GroundTruth]]

evaluate_result: afe.apis.statistic.Statistic[Tuple[afe.apis.compilation_job_base.Tensors, afe.apis.compilation_job_base.GroundTruth], str]

afe.apis.compilation_job.set_compilation_job(job: CompilationJob[Any]) → None

Use the given CompilationJob to control compilation. If called multiple times, the job that is passed to the final call will be used.