MCPcopy Create free account
hub / github.com/pytorch/pytorch / conv_model_generators

Function conv_model_generators

caffe2/python/test/executor_test_util.py:52–60  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

50
51
52def conv_model_generators():
53 return {
54 'AlexNet': cb.AlexNet,
55 'OverFeat': cb.OverFeat,
56 'VGGA': cb.VGGA,
57 'Inception': cb.Inception,
58 'MLP': cb.MLP,
59 'Resnet50': gen_test_resnet50,
60 }
61
62
63def executor_test_model_names():

Callers 2

build_conv_modelFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…