xref: /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/model.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker# resnet50 layer shape
2*da0073e9SAndroid Build Coastguard Workerresnet50_layers = (
3*da0073e9SAndroid Build Coastguard Worker    # IN_H, IN_W, IN_C, KERNEL_H, KERNEL_W, KERNEL_N, stride, padding
4*da0073e9SAndroid Build Coastguard Worker    (224, 224, 3, 7, 7, 64, (2, 2), (0, 0)),
5*da0073e9SAndroid Build Coastguard Worker    # conv2_x
6*da0073e9SAndroid Build Coastguard Worker    (56, 56, 64, 1, 1, 64, (1, 1), (0, 0)),
7*da0073e9SAndroid Build Coastguard Worker    (56, 56, 64, 3, 3, 64, (1, 1), (0, 0)),
8*da0073e9SAndroid Build Coastguard Worker    (56, 56, 64, 1, 1, 256, (1, 1), (0, 0)),
9*da0073e9SAndroid Build Coastguard Worker    # conv3_x
10*da0073e9SAndroid Build Coastguard Worker    (56, 56, 256, 1, 1, 128, (2, 2), (0, 0)),
11*da0073e9SAndroid Build Coastguard Worker    (28, 28, 128, 3, 3, 128, (1, 1), (0, 0)),
12*da0073e9SAndroid Build Coastguard Worker    (28, 28, 128, 1, 1, 512, (1, 1), (0, 0)),
13*da0073e9SAndroid Build Coastguard Worker    # conv4_x
14*da0073e9SAndroid Build Coastguard Worker    (28, 28, 512, 1, 1, 256, (2, 2), (0, 0)),
15*da0073e9SAndroid Build Coastguard Worker    (14, 14, 256, 3, 3, 256, (1, 1), (0, 0)),
16*da0073e9SAndroid Build Coastguard Worker    (14, 14, 256, 1, 1, 1024, (1, 1), (0, 0)),
17*da0073e9SAndroid Build Coastguard Worker    # conv5_x
18*da0073e9SAndroid Build Coastguard Worker    (14, 14, 1024, 1, 1, 512, (2, 2), (0, 0)),
19*da0073e9SAndroid Build Coastguard Worker    (7, 7, 512, 3, 3, 512, (1, 1), (0, 0)),
20*da0073e9SAndroid Build Coastguard Worker    (7, 7, 512, 1, 1, 2048, (1, 1), (0, 0)),
21*da0073e9SAndroid Build Coastguard Worker)
22*da0073e9SAndroid Build Coastguard Worker
23*da0073e9SAndroid Build Coastguard Workeralexnet_layers = (
24*da0073e9SAndroid Build Coastguard Worker    # IN_H, IN_W, IN_C, KERNEL_H, KERNEL_W, KERNEL_N, stride, padding
25*da0073e9SAndroid Build Coastguard Worker    (224, 224, 3, 11, 11, 64, (4, 4), (2, 2)),
26*da0073e9SAndroid Build Coastguard Worker)
27