habitat.analysis.predictor

`habitat.analysis.predictor`#

Module Contents#

Classes#

Attributes#

`logger`
`CONV2D_PARAMS`
`CONVTRANSPOSE2D_PARAMS`
`LINEAR_PARAMS`
`BMM_PARAMS`
`LSTM_PARAMS_NO_BATCH_SIZES`
`LSTM_PARAMS`
`MATMUL_PARAMS`

habitat.analysis.predictor.logger[source]#

habitat.analysis.predictor.CONV2D_PARAMS = ['input', 'weight', 'bias', 'stride', 'padding', 'dilation', 'groups'][source]#

habitat.analysis.predictor.CONVTRANSPOSE2D_PARAMS = ['input', 'weight', 'bias', 'stride', 'padding', 'dilation', 'groups'][source]#

habitat.analysis.predictor.LINEAR_PARAMS = ['input', 'weight', 'bias'][source]#

habitat.analysis.predictor.BMM_PARAMS = ['input', 'mat2', 'out'][source]#

habitat.analysis.predictor.LSTM_PARAMS_NO_BATCH_SIZES = ['input', 'hx', 'flat_weights', 'bias', 'num_layers', 'dropout', 'training', 'bidirectional',...[source]#

habitat.analysis.predictor.LSTM_PARAMS = ['input', 'batch_sizes', 'hx', 'flat_weights', 'bias', 'num_layers', 'dropout', 'training',...[source]#

habitat.analysis.predictor.MATMUL_PARAMS = ['input', 'other', 'out'][source]#

class habitat.analysis.predictor.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]#