| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/tests/python/frontend/mxnet/model_zoo/ |
| H A D | __init__.py | 27 nnvm_mlp = nnvm.testing.mlp.get_workload(1, _num_class)[0]
34 nnvm_resnet[num_layer] = nnvm.testing.resnet.get_workload(
42 nnvm_vgg[num_layer] = nnvm.testing.vgg.get_workload(
50 nnvm_squeezenet[version] = nnvm.testing.squeezenet.get_workload(1, version=version)[0]
54 nnvm_inception_v3 = nnvm.testing.inception_v3.get_workload(1)[0]
58 nnvm_dqn = nnvm.testing.dqn.get_workload(1)[0]
62 nnvm_dcgan = nnvm.testing.dcgan.get_workload(1)[0]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/tests/python/frontend/mxnet/model_zoo/ |
| H A D | __init__.py | 27 nnvm_mlp = nnvm.testing.mlp.get_workload(1, _num_class)[0]
34 nnvm_resnet[num_layer] = nnvm.testing.resnet.get_workload(
42 nnvm_vgg[num_layer] = nnvm.testing.vgg.get_workload(
50 nnvm_squeezenet[version] = nnvm.testing.squeezenet.get_workload(1, version=version)[0]
54 nnvm_inception_v3 = nnvm.testing.inception_v3.get_workload(1)[0]
58 nnvm_dqn = nnvm.testing.dqn.get_workload(1)[0]
62 nnvm_dcgan = nnvm.testing.dcgan.get_workload(1)[0]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tests/python/frontend/mxnet/model_zoo/ |
| H A D | __init__.py | 30 return tvm.relay.testing.mlp.get_workload(1, num_class)[0]
39 return tvm.relay.testing.vgg.get_workload(
49 return tvm.relay.testing.resnet.get_workload(
57 return tvm.relay.testing.dqn.get_workload(1)[0]
64 return tvm.relay.testing.squeezenet.get_workload(1, version=version)[0]
70 return tvm.relay.testing.inception_v3.get_workload(1)[0]
76 return tvm.relay.testing.dcgan.get_workload(batch_size=batch_size)[0]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/frontend/mxnet/model_zoo/ |
| H A D | __init__.py | 31 return tvm.relay.testing.mlp.get_workload(1, num_class)[0]
42 return tvm.relay.testing.vgg.get_workload(1, num_class, num_layers=num_layers)[0]
53 return tvm.relay.testing.resnet.get_workload(1, num_class, num_layers=num_layers)[0]
61 return tvm.relay.testing.dqn.get_workload(1)[0]
70 return tvm.relay.testing.squeezenet.get_workload(1, version=version)[0]
78 return tvm.relay.testing.inception_v3.get_workload(1)[0]
86 return tvm.relay.testing.dcgan.get_workload(batch_size=batch_size)[0]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/frontend/mxnet/model_zoo/ |
| H A D | __init__.py | 30 return tvm.relay.testing.mlp.get_workload(1, num_class)[0]
39 return tvm.relay.testing.vgg.get_workload(
49 return tvm.relay.testing.resnet.get_workload(
57 return tvm.relay.testing.dqn.get_workload(1)[0]
64 return tvm.relay.testing.squeezenet.get_workload(1, version=version)[0]
70 return tvm.relay.testing.inception_v3.get_workload(1)[0]
76 return tvm.relay.testing.dcgan.get_workload(batch_size=batch_size)[0]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/relay/benchmarking/ |
| H A D | benchmark_vm.py | 94 mod, params = testing.mlp.get_workload(1)
100 mod, params = testing.vgg.get_workload(1, num_layers=n)
107 mod, params = testing.resnet.get_workload(batch_size=1, num_layers=n)
114 mod, params = testing.squeezenet.get_workload(version=version)
121 mod, params = testing.inception_v3.get_workload(image_shape=image_shape)
127 mod, params = testing.dqn.get_workload(batch_size=1, image_shape=image_shape)
133 mod, params = testing.dcgan.get_workload(batch_size=1)
138 mod, params = testing.mobilenet.get_workload(batch_size=1)
145 mod, params = testing.mobilenet.get_workload(
152 mod, params = testing.densenet.get_workload(batch_size=1)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tests/python/relay/benchmarking/ |
| H A D | benchmark_vm.py | 92 mod, params = testing.mlp.get_workload(1)
99 mod, params = testing.vgg.get_workload(1, num_layers=n)
106 mod, params = testing.resnet.get_workload(batch_size=1, num_layers=n)
113 mod, params = testing.squeezenet.get_workload(version=version)
120 mod, params = testing.inception_v3.get_workload(image_shape=image_shape)
127 mod, params = testing.dqn.get_workload(
134 mod, params = testing.dcgan.get_workload(batch_size=1)
139 mod, params = testing.mobilenet.get_workload(batch_size=1)
145 mod, params = testing.mobilenet.get_workload(batch_size=1,
151 mod, params = testing.densenet.get_workload(batch_size=1)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/relay/benchmarking/ |
| H A D | benchmark_vm.py | 92 mod, params = testing.mlp.get_workload(1)
99 mod, params = testing.vgg.get_workload(1, num_layers=n)
106 mod, params = testing.resnet.get_workload(batch_size=1, num_layers=n)
113 mod, params = testing.squeezenet.get_workload(version=version)
120 mod, params = testing.inception_v3.get_workload(image_shape=image_shape)
127 mod, params = testing.dqn.get_workload(
134 mod, params = testing.dcgan.get_workload(batch_size=1)
139 mod, params = testing.mobilenet.get_workload(batch_size=1)
145 mod, params = testing.mobilenet.get_workload(batch_size=1,
151 mod, params = testing.densenet.get_workload(batch_size=1)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/apps/benchmark/ |
| H A D | util.py | 50 net, params = testing.mobilenet.get_workload(batch_size=batch_size, dtype=dtype)
53 net, params = testing.inception_v3.get_workload(batch_size=batch_size, dtype=dtype)
56 … net, params = testing.resnet.get_workload(num_layers=n_layer, batch_size=batch_size, dtype=dtype)
59 … net, params = testing.vgg.get_workload(num_layers=n_layer, batch_size=batch_size, dtype=dtype)
62 …net, params = testing.densenet.get_workload(densenet_size=n_layer, batch_size=batch_size, dtype=dt…
65 … net, params = testing.squeezenet.get_workload(batch_size=batch_size, version=version, dtype=dtype)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/apps/benchmark/ |
| H A D | util.py | 51 net, params = testing.mobilenet.get_workload(batch_size=batch_size, dtype=dtype)
54 net, params = testing.inception_v3.get_workload(batch_size=batch_size, dtype=dtype)
57 net, params = testing.resnet.get_workload(
62 net, params = testing.vgg.get_workload(
67 net, params = testing.densenet.get_workload(
72 net, params = testing.squeezenet.get_workload(
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/apps/benchmark/ |
| H A D | util.py | 50 net, params = testing.mobilenet.get_workload(batch_size=batch_size, dtype=dtype)
53 net, params = testing.inception_v3.get_workload(batch_size=batch_size, dtype=dtype)
56 … net, params = testing.resnet.get_workload(num_layers=n_layer, batch_size=batch_size, dtype=dtype)
59 … net, params = testing.vgg.get_workload(num_layers=n_layer, batch_size=batch_size, dtype=dtype)
62 …net, params = testing.densenet.get_workload(densenet_size=n_layer, batch_size=batch_size, dtype=dt…
65 … net, params = testing.squeezenet.get_workload(batch_size=batch_size, version=version, dtype=dtype)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tests/python/relay/ |
| H A D | test_ir_text_printer.py | 137 net, params = tvm.relay.testing.mlp.get_workload(batch_size=1)
142 net, params = tvm.relay.testing.resnet.get_workload(batch_size=1)
147 net, params = tvm.relay.testing.mobilenet.get_workload(batch_size=1)
152 net, params = tvm.relay.testing.dqn.get_workload(batch_size=1)
157 net, params = tvm.relay.testing.dcgan.get_workload(batch_size=1)
162 net, params = tvm.relay.testing.lstm.get_workload(1, 1)
165 net, params = tvm.relay.testing.lstm.get_workload(4, 4)
169 net, params = tvm.relay.testing.inception_v3.get_workload(batch_size=1)
175 net, params = tvm.relay.testing.squeezenet.get_workload(batch_size=1, version=version)
180 net, params = tvm.relay.testing.vgg.get_workload(batch_size=1)
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/relay/ |
| H A D | test_ir_text_printer.py | 139 net, _ = tvm.relay.testing.mlp.get_workload(batch_size=1)
144 net, _ = tvm.relay.testing.resnet.get_workload(batch_size=1)
149 net, _ = tvm.relay.testing.mobilenet.get_workload(batch_size=1)
154 net, _ = tvm.relay.testing.dqn.get_workload(batch_size=1)
159 net, _ = tvm.relay.testing.dcgan.get_workload(batch_size=1)
164 net, _ = tvm.relay.testing.lstm.get_workload(1, 1)
167 net, _ = tvm.relay.testing.lstm.get_workload(4, 4)
172 net, _ = tvm.relay.testing.inception_v3.get_workload(batch_size=1)
178 net, _ = tvm.relay.testing.squeezenet.get_workload(batch_size=1, version=version)
183 net, _ = tvm.relay.testing.vgg.get_workload(batch_size=1)
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/relay/ |
| H A D | test_ir_text_printer.py | 137 net, params = tvm.relay.testing.mlp.get_workload(batch_size=1)
142 net, params = tvm.relay.testing.resnet.get_workload(batch_size=1)
147 net, params = tvm.relay.testing.mobilenet.get_workload(batch_size=1)
152 net, params = tvm.relay.testing.dqn.get_workload(batch_size=1)
157 net, params = tvm.relay.testing.dcgan.get_workload(batch_size=1)
162 net, params = tvm.relay.testing.lstm.get_workload(1, 1)
165 net, params = tvm.relay.testing.lstm.get_workload(4, 4)
169 net, params = tvm.relay.testing.inception_v3.get_workload(batch_size=1)
175 net, params = tvm.relay.testing.squeezenet.get_workload(batch_size=1, version=version)
180 net, params = tvm.relay.testing.vgg.get_workload(batch_size=1)
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tests/python/frontend/nnvm_to_relay/ |
| H A D | test_forward.py | 57 model, params = nnvm.testing.mlp.get_workload(1)
62 model, params = nnvm.testing.vgg.get_workload(1)
67 model, params = nnvm.testing.resnet.get_workload(1)
72 model, params = nnvm.testing.squeezenet.get_workload(1)
77 model, params = nnvm.testing.inception_v3.get_workload(1)
82 model, params = nnvm.testing.squeezenet.get_workload(1)
87 model, params = nnvm.testing.dqn.get_workload(1)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/frontend/nnvm_to_relay/ |
| H A D | test_forward.py | 57 model, params = nnvm.testing.mlp.get_workload(1)
62 model, params = nnvm.testing.vgg.get_workload(1)
67 model, params = nnvm.testing.resnet.get_workload(1)
72 model, params = nnvm.testing.squeezenet.get_workload(1)
77 model, params = nnvm.testing.inception_v3.get_workload(1)
82 model, params = nnvm.testing.squeezenet.get_workload(1)
87 model, params = nnvm.testing.dqn.get_workload(1)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/tests/python/compiler/ |
| H A D | test_autotvm_task_extraction.py | 29 net, params = nnvm.testing.resnet.get_workload(num_layers=18, batch_size=batch_size)
31 net, params = nnvm.testing.mobilenet.get_workload(batch_size=batch_size)
33 net, params = nnvm.testing.squeezenet.get_workload(batch_size=batch_size, version='1.1')
35 net, params = nnvm.testing.vgg.get_workload(num_layers=16, batch_size=batch_size)
37 net, params = nnvm.testing.dcgan.get_workload(batch_size=batch_size)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/tests/python/compiler/ |
| H A D | test_autotvm_task_extraction.py | 29 net, params = nnvm.testing.resnet.get_workload(num_layers=18, batch_size=batch_size)
31 net, params = nnvm.testing.mobilenet.get_workload(batch_size=batch_size)
33 net, params = nnvm.testing.squeezenet.get_workload(batch_size=batch_size, version='1.1')
35 net, params = nnvm.testing.vgg.get_workload(num_layers=16, batch_size=batch_size)
37 net, params = nnvm.testing.dcgan.get_workload(batch_size=batch_size)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/tutorials/ |
| H A D | tune_nnvm_x86.py | 52 net, params = nnvm.testing.resnet.get_workload(num_layers=n_layer, batch_size=batch_size)
55 net, params = nnvm.testing.vgg.get_workload(num_layers=n_layer, batch_size=batch_size)
57 net, params = nnvm.testing.mobilenet.get_workload(batch_size=batch_size)
59 net, params = nnvm.testing.squeezenet.get_workload(batch_size=batch_size, version='1.1')
62 net, params = nnvm.testing.inception_v3.get_workload(batch_size=batch_size)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tutorials/autotvm/ |
| H A D | tune_relay_cuda.py | 86 mod, params = relay.testing.resnet.get_workload(
91 mod, params = relay.testing.vgg.get_workload(
95 mod, params = relay.testing.mobilenet.get_workload(batch_size=batch_size, dtype=dtype)
97 mod, params = relay.testing.squeezenet.get_workload(
102 mod, params = relay.testing.inception_v3.get_workload(batch_size=batch_size, dtype=dtype)
|
| H A D | tune_relay_x86.py | 57 mod, params = relay.testing.resnet.get_workload(
62 mod, params = relay.testing.vgg.get_workload(
66 mod, params = relay.testing.mobilenet.get_workload(batch_size=batch_size, dtype=dtype)
68 mod, params = relay.testing.squeezenet.get_workload(
73 mod, params = relay.testing.inception_v3.get_workload(batch_size=batch_size, dtype=dtype)
|
| H A D | tune_relay_arm.py | 88 mod, params = relay.testing.resnet.get_workload(
93 mod, params = relay.testing.vgg.get_workload(
97 mod, params = relay.testing.mobilenet.get_workload(batch_size=batch_size)
99 mod, params = relay.testing.squeezenet.get_workload(
104 mod, params = relay.testing.inception_v3.get_workload(batch_size=batch_size, dtype=dtype)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/tutorials/ |
| H A D | tune_nnvm_x86.py | 52 net, params = nnvm.testing.resnet.get_workload(num_layers=n_layer, batch_size=batch_size)
55 net, params = nnvm.testing.vgg.get_workload(num_layers=n_layer, batch_size=batch_size)
57 net, params = nnvm.testing.mobilenet.get_workload(batch_size=batch_size)
59 net, params = nnvm.testing.squeezenet.get_workload(batch_size=batch_size, version='1.1')
62 net, params = nnvm.testing.inception_v3.get_workload(batch_size=batch_size)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tutorials/autotvm/ |
| H A D | tune_relay_cuda.py | 84 …mod, params = relay.testing.resnet.get_workload(num_layers=n_layer, batch_size=batch_size, dtype=d…
87 …mod, params = relay.testing.vgg.get_workload(num_layers=n_layer, batch_size=batch_size, dtype=dtyp…
89 mod, params = relay.testing.mobilenet.get_workload(batch_size=batch_size, dtype=dtype)
91 …mod, params = relay.testing.squeezenet.get_workload(batch_size=batch_size, version='1.1', dtype=dt…
94 mod, params = relay.testing.inception_v3.get_workload(batch_size=batch_size, dtype=dtype)
|
| H A D | tune_relay_x86.py | 54 …mod, params = relay.testing.resnet.get_workload(num_layers=n_layer, batch_size=batch_size, dtype=d…
57 …mod, params = relay.testing.vgg.get_workload(num_layers=n_layer, batch_size=batch_size, dtype=dtyp…
59 mod, params = relay.testing.mobilenet.get_workload(batch_size=batch_size, dtype=dtype)
61 …mod, params = relay.testing.squeezenet.get_workload(batch_size=batch_size, version='1.1', dtype=dt…
64 mod, params = relay.testing.inception_v3.get_workload(batch_size=batch_size, dtype=dtype)
|