Index

A

add_operator() (OperatorGraph method)

B

best() (Sweep method) build_estimator() (HardwareInfo method) built-in function estimate_memory() get_hardware_info() load_hardware_yaml() load_model_yaml() simulate() sweep()

C

CustomModel (built-in class)

E

estimate_memory() built-in function

ExecutionMode (built-in class)

G

get_hardware_info() built-in function

get_peak_tflops() (HardwareInfo method) get_peak_tflops_mm_for_dtype() (HardwareInfo method)

H

HardwareConfig (built-in class)

HardwareInfo (built-in class) HFModel (built-in class)

L

load_hardware_yaml() built-in function

load_model_yaml() built-in function

M

ModelConfig (built-in class)

N

NetworkParams (built-in class)

O

OperatorGraph (built-in class)

OperatorNode (built-in class) OperatorType (built-in class)

P

ParallelismConfig (built-in class)

S

simulate() built-in function simulate_on() (Trace method) SimulationReport (built-in class) SimulatorConfig (built-in class)

summary() (OperatorGraph method) Sweep (built-in class) sweep() built-in function SweepRow (built-in class)

T

TensorMeta (built-in class) to_dataframe() (SimulationReport method) (Sweep method) to_dict() (OperatorNode method) (SimulationReport method) (TensorMeta method)

to_dot() (OperatorGraph method) to_json() (OperatorGraph method) (SimulationReport method) topological_sort() (OperatorGraph method) Trace (built-in class) TrainingConfig (built-in class)

V

validate() (OperatorGraph method)

W

world_size (ParallelismConfig property)