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[source]

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

preprocess_transforms: List[afe.apis.transform.Transform][source]
postprocess_transforms: List[afe.apis.transform.Transform][source]
calibration_input: Iterable[afe.apis.compilation_job_base.Tensors][source]
evaluation_input: Iterable[Tuple[afe.apis.compilation_job_base.Tensors, afe.apis.compilation_job_base.GroundTruth]][source]
evaluate_result: afe.apis.statistic.Statistic[Tuple[afe.apis.compilation_job_base.Tensors, afe.apis.compilation_job_base.GroundTruth], str][source]
afe.apis.compilation_job.set_compilation_job(job: CompilationJob[Any]) → None[source]

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