TVM@Alibaba AI Labs
Registr \L Compiler Toolchain 于 TVM TOPI Schedule Primitives & Optimizations Symbols NNVM & Param Frontends Operators Algorithm &Schedule CUDA TOPI Backends Machine Learning Automated Automated Optimizer Schedule explorer Cost model Mali TOPI ROCM TOPI PVRTOPI Alibaba Al.Labs 阿里巴巴人工智能实验室 PVR TOPI > TOPI for PVR,including what you want to compute and how to compute. w What you want to with tensor expression language'; #Return the out operation w How to compute. @autotvm.register_ topi_schedule(schedule_conv2d_nchw,pvr, [direct]) convolution def schedule_conv2d_nchw_pvr(cfg0 码力 | 12 页 | 1.94 MB | 5 月前3Dynamic Model in TVM
strategy.register_specialized_implement(wrap_compute_conv2d(topi.x86.conv2d_winograd), topi.x86.conv2d_winograd, nv2d(topi.x86.conv2d_nchw), topi.x86.schedule_conv2d_nchw) elif layout == "NHWC": strategy.register_default_implement(wrap_compute_conv2d(topi.nn.conv2d_nhwc) topi.x86.schedule_conv2d_nhwc) elif layout == "NCHWc": strategy.register_default_implement(wrap_compute_conv2d(topi.nn.conv2d_nchwc), topi.x86.0 码力 | 24 页 | 417.46 KB | 5 月前3
共 2 条
- 1