Source code for habitat.analysis
from habitat.analysis.device import _Device
[docs]SPECIAL_OPERATIONS = {
# Convolution
'conv2d',
'conv_transpose2d',
# Matrix multiply operations
'linear',
'bmm',
# Recurrent operations
'lstm',
'gru',
'rnn_tanh',
'rnn_relu',
}