Operators with Channels Last support
Ma Mingfei edited this page May 22, 2022
·
2 revisions
absabs_acosacos_adaptive_avg_pool2dadaptive_max_pool2daddadd_addcdivaddcdiv_addcmuladdcmul_asinasin_atan2atan2_avg_pool2dbatch_normbfloat16boolbyteceilceil_charchannel_shuffleclonecontiguousconv2dcopy_digammadigamma_divdiv_doubleempty_likeerfinverfinv_expm1expm1_fill_floatfloorfloor_full_likegroup_normgthalfintisnanloglog_log10log10_log1plog1p_log2log2_longltmax_pool2dmax_unpool2dmulmul_nenegneg_ones_likepixel_shufflepixel_unshufflepowpow_randint_likerand_likerandn_likerelurelu_resize_resize_as_roundround_rsqrtrsqrt_shortsigmoidsigmoid_signsign_sinsin_sinhsinh_sqrtsqrt_totrunctrunc_typeupsample_bilinear2dupsample_nearest2dzero_zeros_like
absabs_acosacos_adaptive_avg_pool2daddadd_addcdivaddcdiv_addcmuladdcmul_asinasin_atan2atan2_batch_normbfloat16boolbytecatceilceil_charclonecontiguousconv2dconv_transpose2dcopy_cpucudacudnn_convolutioncudnn_convolution_transposedetachdigammadigamma_divdiv_doubleempty_likeerfinverfinv_expm1expm1_fill_floatfloorfloor_full_likegthalfintisnanloglog_log10log10_log1plog1p_log2log2_longltmax_pool2dmulmul_nenegneg_ones_likepowpow_randint_likerand_likerandn_likerelurelu_requires_grad_resize_resize_as_roundround_rsqrtrsqrt_shortsigmoidsigmoid_signsign_sinsin_sinhsinh_sqrtsqrt_totrunctrunc_typezero_zeros_like
- Install Prerequisites
- Fork, clone, and checkout the PyTorch source
- Install Dependencies
- Build PyTorch from source
- Tips for developing PyTorch
- PyTorch Workflow Git cheatsheet
- Overview of the Pull Request Lifecycle
- Finding Or Reporting Issues
- Pre Commit Checks
- Create a Pull Request
- Typical Pull Request Workflow
- Pull Request FAQs
- Getting Help
- Codebase structure
- Tensors, Operators, and Testing
- Autograd
- Dispatcher, Structured Kernels, and Codegen
- torch.nn
- CUDA basics
- Data (Optional)
- function transforms (Optional)