habitat#

Subpackages#

Submodules#

Package Contents#

Classes#

Metric

Generic enumeration.

Predictor

OperationTracker

Attributes#

Device

habitat.Device[source]#
class habitat.Metric[source]#

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

DRAMUtilization#
DRAMReadBytes#
DRAMWriteBytes#
SinglePrecisionFLOPEfficiency#
SinglePrecisionAddOps#
class habitat.Predictor(kernel_metadata_file=None, wave_scaling_strategy=unified_wave_scaling)[source]#
predict_operation(operation, dest_device)[source]#
_wave_scale(run_time, dest_device)[source]#
_special_scale(operation, dest_device, scaler)[source]#
_conv2d_scale(operation, dest_device)[source]#
_conv_transpose2d_scale(operation, dest_device)[source]#
_linear_scale(operation, dest_device)[source]#
_bmm_scale(operation, dest_device)[source]#
_lstm_scale(operation, dest_device)[source]#
class habitat.OperationTracker(device, metrics=None, metrics_threshold_ms=0)[source]#

Bases: habitat.tracking.base.TrackerBase

start_tracking()[source]#
stop_tracking()[source]#
get_tracked_trace()[source]#
_hook_creator(func)[source]#