SYMBOL INDEX (109 symbols across 15 files)

FILE: examples/efficientnet.py
  function swish (line 19) | def swish(x):
  function conv_pad_same (line 23) | def conv_pad_same(x, size, kernel=1, stride=1, **kw):
  function conv_bn_act (line 40) | def conv_bn_act(x, size, kernel=1, stride=1, groups=1, bias=False, eps=1...
  function mb_block (line 46) | def mb_block(x, size_out, expand=1, kernel=1, stride=1, se_ratio=0.25, d...
  function squeeze_excitation (line 61) | def squeeze_excitation(x, size_se, name='', **kw):
  function drop_connect (line 70) | def drop_connect(x, rate):
  class WarmEfficientNet (line 89) | class WarmEfficientNet(nn.Module):
    method __init__ (line 90) | def __init__(self):
    method forward (line 93) | def forward(self, x):

FILE: examples/lstm.py
  class WarmTagger (line 32) | class WarmTagger(nn.Module):
    method __init__ (line 33) | def __init__(self, embedding_dim, hidden_dim, vocab_size, tagset_size):
    method forward (line 37) | def forward(self, x): # D
  class TorchTagger (line 46) | class TorchTagger(nn.Module):
    method __init__ (line 47) | def __init__(self, embedding_dim, hidden_dim, vocab_size, tagset_size):
    method forward (line 53) | def forward(self, sentence):
  function prepare_sequence (line 61) | def prepare_sequence(seq, to_ix):
  function main (line 66) | def main():

FILE: examples/mnist.py
  class WarmNet (line 22) | class WarmNet(nn.Module):
    method __init__ (line 23) | def __init__(self):
    method forward (line 26) | def forward(self, x):
  class TorchNet (line 37) | class TorchNet(nn.Module):
    method __init__ (line 38) | def __init__(self):
    method forward (line 44) | def forward(self, x):
  function train (line 55) | def train(p, model, device, train_loader, optimizer, epoch):
  function test (line 70) | def test(p, model, device, test_loader):
  function main (line 86) | def main():

FILE: examples/mobilenet.py
  function conv_bn_relu (line 19) | def conv_bn_relu(x, size, stride=1, expand=1, kernel=3, groups=1, name=''):
  function bottleneck (line 25) | def bottleneck(x, size_out, stride, expand, name=''):
  function conv1x1 (line 37) | def conv1x1(x, *arg, **kw):
  function pool (line 41) | def pool(x, *arg, **kw):
  function classify (line 45) | def classify(x, size, *arg, **kw):
  class WarmMobileNetV2 (line 64) | class WarmMobileNetV2(nn.Module):
    method __init__ (line 65) | def __init__(self):
    method forward (line 68) | def forward(self, x):
  function test (line 78) | def test():

FILE: examples/resnet.py
  function basic (line 20) | def basic(x, size, stride, stack_index, block_index):
  function stack (line 33) | def stack(x, num_block, size, stride, stack_index, block=basic):
  class WarmResNet (line 40) | class WarmResNet(nn.Module):
    method __init__ (line 41) | def __init__(self, block=basic, stack_spec=((2, 64, 1), (2, 128, 2), (...
    method forward (line 46) | def forward(self, x):
  function test_time (line 58) | def test_time(fn, *arg, repeat=10, **kw):
  function test (line 67) | def test():

FILE: examples/transformer.py
  function multi_head_attention (line 17) | def multi_head_attention(x, y=None, num_head=8, dropout=0.1, mask=None, ...
  function feed_forward (line 44) | def feed_forward(x, size_ff=2048, dropout=0.1, **kw):
  function residual_add (line 50) | def residual_add(x, layer, dropout=0.1, **kw):
  function encoder (line 57) | def encoder(x, num_encoder=6, **kw):
  function decoder (line 64) | def decoder(x, y, num_decoder=6, mask_x=None, mask_y=None, **kw):
  function transformer (line 72) | def transformer(x, y, **kw):
  class Transformer (line 78) | class Transformer(nn.Module):
    method __init__ (line 79) | def __init__(self, *shape, **kw):
    method forward (line 83) | def forward(self, x, y):

FILE: tests/test_engine.py
  function test_set_get_default_parent (line 15) | def test_set_get_default_parent():
  function test_auto_name (line 24) | def test_auto_name():
  function test_initialize (line 32) | def test_initialize():
  function test_activate (line 47) | def test_activate():
  function test_permute (line 58) | def test_permute():
  function test_unused_kwargs (line 81) | def test_unused_kwargs():
  function test_prepare_model_is_ready (line 88) | def test_prepare_model_is_ready():
  function test_forward (line 108) | def test_forward():
  function test_namespace (line 130) | def test_namespace():

FILE: tests/test_functional.py
  function test_conv (line 14) | def test_conv():
  function test_linear (line 31) | def test_linear():
  function test_batch_norm (line 48) | def test_batch_norm():
  function test_lstm (line 65) | def test_lstm():
  function test_gru (line 79) | def test_gru():
  function test_identity (line 89) | def test_identity():
  function test_dropout (line 94) | def test_dropout():
  function test_transformer (line 109) | def test_transformer():
  function test_layer_norm (line 125) | def test_layer_norm():
  function test_embedding (line 141) | def test_embedding():

FILE: tests/test_module.py
  function test_lambda (line 15) | def test_lambda():
  function test_sequential (line 29) | def test_sequential():
  function test_shortcut (line 40) | def test_shortcut():

FILE: tests/test_util.py
  function test_camel_to_snake (line 14) | def test_camel_to_snake():
  function test_summary_str (line 24) | def test_summary_str():
  function test_summary (line 31) | def test_summary():

FILE: tests/test_warm.py
  function test_warm_up (line 12) | def test_warm_up():

FILE: warm/engine.py
  function set_default_parent (line 14) | def set_default_parent(parent):
  function get_default_parent (line 20) | def get_default_parent():
  function _auto_name (line 26) | def _auto_name(name, parent):
  function prepare_model_ (line 40) | def prepare_model_(model, *data, device='cpu'):
  function is_ready (line 68) | def is_ready(model):
  function activate (line 73) | def activate(x, spec, lookup=None):
  function initialize_ (line 101) | def initialize_(x, spec):
  function permute (line 110) | def permute(x, in_shape='BCD', out_shape='BCD', **kw):
  function unused_kwargs (line 134) | def unused_kwargs(kw):
  function forward (line 143) | def forward(x, base_class,
  function namespace (line 201) | def namespace(f):

FILE: warm/functional.py
  function conv (line 16) | def conv(x, size, kernel, init_weight=None, init_bias=None, bias=True, *...
  function linear (line 50) | def linear(x, size, init_weight=None, init_bias=None, bias=True, **kw):
  function batch_norm (line 73) | def batch_norm(x, **kw):
  function lstm (line 89) | def lstm(x, size,
  function gru (line 141) | def gru(*arg, **kw):
  function identity (line 170) | def identity(x, *arg, **kw):
  function dropout (line 175) | def dropout(x, rate=0.5, by_channel=False, **kw):
  function transformer (line 192) | def transformer(x, y=None, num_encoder=6, num_decoder=6, num_head=8,
  function layer_norm (line 244) | def layer_norm(x, dim=1, **kw):
  function embedding (line 269) | def embedding(x, size, vocabulary=None, **kw):

FILE: warm/module.py
  class Lambda (line 43) | class Lambda(nn.Module):
    method __init__ (line 48) | def __init__(self, fn, *arg, **kw):
    method forward (line 53) | def forward(self, x):
  class Sequential (line 58) | class Sequential(nn.Sequential):
    method forward (line 61) | def forward(self, x):
  class Shortcut (line 74) | class Shortcut(Sequential):
    method __init__ (line 79) | def __init__(self, *arg, projection=None):
    method forward (line 82) | def forward(self, x):

FILE: warm/util.py
  function camel_to_snake (line 16) | def camel_to_snake(name):
  function summary_str (line 22) | def summary_str(model):
  function summary (line 47) | def summary(model):