get_workload ( batch_size = batch_size, dtype = dtype ) elif name = "mxnet" : # an example for mxnet model from _zoo.vision import get_model block = get_model ( "resnet18_v1", pretrained = True ) mod, params = relay. get_workload ( batch_size = batch_size, version = "1.1", dtype = dtype ) elif name = "inception_v3" : input_shape = ( batch_size, 3, 299, 299 ) mod, params = relay. get_workload ( batch_size = batch_size ) elif name = "squeezenet_v1.1" : mod, params = relay. get_workload ( num_layers = n_layer, batch_size = batch_size, dtype = dtype ) elif name = "mobilenet" : mod, params = relay. get_workload ( num_layers = n_layer, batch_size = batch_size, dtype = dtype ) elif "vgg" in name : n_layer = int ( name. Use AutoScheduler for Template-Free SchedulingÄef get_network ( name, batch_size ): """Get the symbol definition and random weight of a network""" input_shape = ( batch_size, 3, 224, 224 ) output_shape = ( batch_size, 1000 ) if "resnet" in name : n_layer = int ( name. ![]() ![]() Auto-tuning a Convolutional Network for Mobile GPU.Auto-tuning a Convolutional Network for ARM CPU.Auto-tuning a Convolutional Network for x86 CPU.Auto-tuning a Convolutional Network for NVIDIA GPU.Tuning High Performance Convolution on NVIDIA GPUs.Work With Tensor Expression and Schedules.
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |