afe.ir.transform.requantization_hoisting.defines

Shared definitions for optimization of requantize nodes.

Attributes

A
Need
DataNeeds
NeedMapping

Functions

need_mapping_empty(→ NeedMapping[A])	Create an emtpy NeedMapping.
need_mapping_singleton(→ NeedMapping[A])	Create a NeedMapping containing a single (k, v) entry.
need_mapping_insert(→ None)	Insert (k, v) into m. Replaces any existing entry for k.
need_mapping_get(→ Optional[A])	Get the value associated with need k, or None if k is not found.
need_mapping_find(→ A)	Get the value associated with need k. Raise an exception if it is not found.

Module Contents

afe.ir.transform.requantization_hoisting.defines.A

afe.ir.transform.requantization_hoisting.defines.Need

afe.ir.transform.requantization_hoisting.defines.DataNeeds

afe.ir.transform.requantization_hoisting.defines.NeedMapping

afe.ir.transform.requantization_hoisting.defines.need_mapping_empty() → NeedMapping[A]

Create an emtpy NeedMapping.

afe.ir.transform.requantization_hoisting.defines.need_mapping_singleton(k: Need, v: A) → NeedMapping[A]

Create a NeedMapping containing a single (k, v) entry.

afe.ir.transform.requantization_hoisting.defines.need_mapping_insert(m: NeedMapping[A], k: Need, v: A) → None

Insert (k, v) into m. Replaces any existing entry for k.

afe.ir.transform.requantization_hoisting.defines.need_mapping_get(m: NeedMapping[A], k: Need) → A | None

Get the value associated with need k, or None if k is not found.

afe.ir.transform.requantization_hoisting.defines.need_mapping_find(m: NeedMapping[A], k: Need) → A

Get the value associated with need k. Raise an exception if it is not found.