| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/python/nnvm/testing/ |
| H A D | vgg.py | 41 def get_classifier(input_data, num_classes): argument
53 def get_symbol(num_classes, num_layers=11, batch_norm=False): argument
77 def get_workload(batch_size, num_classes=1000, image_shape=(3, 224, 224), argument
|
| H A D | mlp.py | 23 def get_symbol(num_classes=1000): argument
34 def get_workload(batch_size, num_classes=1000, image_shape=(3, 224, 224), dtype="float32"): argument
|
| H A D | resnet.py | 90 def resnet(units, num_stages, filter_list, num_classes, image_shape, argument
139 def get_symbol(num_classes, num_layers=50, image_shape=(3, 224, 224), **kwargs): argument
190 def get_workload(batch_size=1, num_classes=1000, num_layers=18, argument
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/python/nnvm/testing/ |
| H A D | vgg.py | 41 def get_classifier(input_data, num_classes): argument
53 def get_symbol(num_classes, num_layers=11, batch_norm=False): argument
77 def get_workload(batch_size, num_classes=1000, image_shape=(3, 224, 224), argument
|
| H A D | mlp.py | 23 def get_symbol(num_classes=1000): argument
34 def get_workload(batch_size, num_classes=1000, image_shape=(3, 224, 224), dtype="float32"): argument
|
| H A D | resnet.py | 90 def resnet(units, num_stages, filter_list, num_classes, image_shape, argument
139 def get_symbol(num_classes, num_layers=50, image_shape=(3, 224, 224), **kwargs): argument
190 def get_workload(batch_size=1, num_classes=1000, num_layers=18, argument
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/testing/ |
| H A D | vgg.py | 51 def get_classifier(input_data, num_classes): argument
64 def get_net(batch_size, image_shape, num_classes, dtype, num_layers=11, batch_norm=False): argument
106 num_classes=1000, argument
|
| H A D | mlp.py | 25 def get_net(batch_size, num_classes=10, image_shape=(1, 28, 28), dtype="float32"): argument
61 def get_workload(batch_size, num_classes=10, image_shape=(1, 28, 28), dtype="float32"): argument
|
| H A D | resnet.py | 160 num_classes, argument
270 num_classes, argument
337 num_classes=1000, argument
|
| H A D | resnet_3d.py | 157 num_classes, argument
264 num_classes, argument
332 num_classes=1000, argument
|
| /dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/relay/testing/ |
| H A D | vgg.py | 46 def get_classifier(input_data, num_classes): argument
59 def get_net(batch_size, image_shape, num_classes, dtype, num_layers=11, batch_norm=False): argument
98 num_classes=1000, argument
|
| H A D | resnet.py | 111 num_classes, argument
176 num_classes, argument
235 num_classes=1000, argument
|
| H A D | mlp.py | 25 num_classes=10, argument
66 num_classes=10, argument
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/relay/testing/ |
| H A D | vgg.py | 46 def get_classifier(input_data, num_classes): argument
59 def get_net(batch_size, image_shape, num_classes, dtype, num_layers=11, batch_norm=False): argument
98 num_classes=1000, argument
|
| H A D | resnet.py | 111 num_classes, argument
176 num_classes, argument
235 num_classes=1000, argument
|
| H A D | mlp.py | 25 num_classes=10, argument
66 num_classes=10, argument
|
| /dports/misc/mmdnn/MMdnn-0.3.1/mmdnn/conversion/examples/tensorflow/models/ |
| H A D | nasnet.py | 219 def _build_aux_head(net, end_points, num_classes, hparams, scope): argument
283 images, num_classes, is_training=True): argument
326 def build_nasnet_mobile(images, num_classes, argument
374 def build_nasnet_large(images, num_classes, argument
425 num_classes, argument
|
| /dports/misc/mmdnn/MMdnn-0.3.1/mmdnn/conversion/examples/keras/ |
| H A D | utils.py | 17 def yolo_head(feats, anchors, num_classes, input_shape): argument
78 def yolo_boxes_and_scores(feats, anchors, num_classes, input_shape, image_shape): argument
93 num_classes, argument
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/example/ssd/symbol/ |
| H A D | legacy_vgg16_ssd_512.py | 22 def get_symbol_train(num_classes=20, nms_thresh=0.5, force_suppress=False, nms_topk=400): argument
179 def get_symbol(num_classes=20, nms_thresh=0.5, force_suppress=False, nms_topk=400): argument
|
| H A D | legacy_vgg16_ssd_300.py | 22 def get_symbol_train(num_classes=20, nms_thresh=0.5, force_suppress=False, argument
175 def get_symbol(num_classes=20, nms_thresh=0.5, force_suppress=False, argument
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/ssd/symbol/ |
| H A D | legacy_vgg16_ssd_512.py | 22 def get_symbol_train(num_classes=20, nms_thresh=0.5, force_suppress=False, nms_topk=400): argument
179 def get_symbol(num_classes=20, nms_thresh=0.5, force_suppress=False, nms_topk=400): argument
|
| H A D | legacy_vgg16_ssd_300.py | 22 def get_symbol_train(num_classes=20, nms_thresh=0.5, force_suppress=False, argument
175 def get_symbol(num_classes=20, nms_thresh=0.5, force_suppress=False, argument
|
| /dports/misc/py-gluonnlp/gluon-nlp-0.10.0/src/gluonnlp/model/ |
| H A D | sampled_block.py | 41 def __init__(self, num_classes, num_sampled, in_unit, remove_accidental_hits, argument
149 def __init__(self, num_classes, num_sampled, in_unit, remove_accidental_hits, argument
276 def __init__(self, num_classes, num_sampled, in_unit, remove_accidental_hits=False, argument
365 def __init__(self, num_classes, num_sampled, in_unit, remove_accidental_hits=True, argument
453 def __init__(self, num_classes, num_sampled, in_unit, remove_accidental_hits, argument
585 def __init__(self, num_classes, num_sampled, in_unit, remove_accidental_hits=True, argument
683 def __init__(self, num_classes, num_sampled, in_unit, remove_accidental_hits=True, argument
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tests/python/frontend/mxnet/model_zoo/ |
| H A D | vgg.py | 37 def get_classifier(input_data, num_classes, **kwargs): argument
57 def get_symbol(num_classes, num_layers=11, batch_norm=False, dtype='float32', **kwargs): argument
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/frontend/mxnet/model_zoo/ |
| H A D | vgg.py | 55 def get_classifier(input_data, num_classes, **kwargs): argument
76 def get_symbol(num_classes, num_layers=11, batch_norm=False, dtype="float32", **kwargs): argument
|