frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity
Page: Operators with Channels Last support
Pages
A quick guide on how to add and cache dependencies on PyTorch CI Adding and maintaining type annotations in PyTorch Autograd Basics Autograd Onboarding Lab Best Practices to Edit and Compile PyTorch Source Code On Windows Bot commands Boxing and Unboxing in the PyTorch Operator Library Build PyTorch and LibTorch on Windows ARM64 CUDA basics Code review values Codegen and Structured Kernels Continuous Integration Core Frontend Onboarding Cpp API Quick Walkthrough Data Basics DataPipes Testing Requirements Debugging CI Failures without SSH Access Debugging Windows with Remote Desktop or CDB (CLI windbg) on CircleCI Debugging using with ssh for Github Actions Dev Infra Office Hours Developer FAQ Dispatcher Structured Kernels Lab Docker image build on CircleCI Docstring Guidelines Getting help as a contributor Guide for adding type annotations to PyTorch Home How to propose feature changes to PyTorch How to support `torch.set_deterministic()` in PyTorch operators How to use TensorIterator Life of a Tensor MPS Backend Memory format propagation rules Modular components for benchmarking PyTorch snippets. (Experimental) Module Onboarding Lab Multiprocessing Technical Notes OpInfos FAQ Operators with Channels Last support Public API definition and documentation Pull request review etiquette PyTorch's Python Frontend Backward and Forward Compatibility Policy PyTorch AutoLabel Bot PyTorch Basics PyTorch CI Metrics Dashboards: the HUD PyTorch Data Flow and Interface Diagram PyTorch IR PyTorch ONNX Exporter Code Reviews and Duty Rotation PyTorch ONNX Topics PyTorch ONNX exporter PyTorch OSS benchmark infra PyTorch Ops to oneDNN Functions Mapping PyTorch Versions PyTorch Workflow Cheatsheet PyTorch dispatcher walkthrough Pytorch Training Loops Running and writing tests Sharing design documents for discussion Software Architecture for c10 TH to ATen porting guide Tensor and Operator Basics The PyTorch Contribution Process The Ultimate Guide to PyTorch Contributions The torch.fft module in PyTorch 1.7 Troubleshooting Documentation Build Troubleshooting Using hud.pytorch.org What is considered a SEV? Where or how should I add documentation Writing Python in cpp (a manifesto) Writing memory format aware operators Writing tests in PyTorch 1.8 clang format function transforms (aka torch.func, functorch) lintrunner nn Basics torch.nn Module Documentation Style Guide torch.onnx Namespacing vmap Basics vmap Onboarding Lab
Clone
2
Operators with Channels Last support
Ma Mingfei edited this page 2022-05-22 09:12:39 +08:00
Table of Contents

Operators with Channels Last support

CPU operators

  • abs
  • abs_
  • acos
  • acos_
  • adaptive_avg_pool2d
  • adaptive_max_pool2d
  • add
  • add_
  • addcdiv
  • addcdiv_
  • addcmul
  • addcmul_
  • asin
  • asin_
  • atan2
  • atan2_
  • avg_pool2d
  • batch_norm
  • bfloat16
  • bool
  • byte
  • ceil
  • ceil_
  • char
  • channel_shuffle
  • clone
  • contiguous
  • conv2d
  • copy_
  • digamma
  • digamma_
  • div
  • div_
  • double
  • empty_like
  • erfinv
  • erfinv_
  • expm1
  • expm1_
  • fill_
  • float
  • floor
  • floor_
  • full_like
  • group_norm
  • gt
  • half
  • int
  • isnan
  • log
  • log_
  • log10
  • log10_
  • log1p
  • log1p_
  • log2
  • log2_
  • long
  • lt
  • max_pool2d
  • max_unpool2d
  • mul
  • mul_
  • ne
  • neg
  • neg_
  • ones_like
  • pixel_shuffle
  • pixel_unshuffle
  • pow
  • pow_
  • randint_like
  • rand_like
  • randn_like
  • relu
  • relu_
  • resize_
  • resize_as_
  • round
  • round_
  • rsqrt
  • rsqrt_
  • short
  • sigmoid
  • sigmoid_
  • sign
  • sign_
  • sin
  • sin_
  • sinh
  • sinh_
  • sqrt
  • sqrt_
  • to
  • trunc
  • trunc_
  • type
  • upsample_bilinear2d
  • upsample_nearest2d
  • zero_
  • zeros_like

GPU operators

  • abs
  • abs_
  • acos
  • acos_
  • adaptive_avg_pool2d
  • add
  • add_
  • addcdiv
  • addcdiv_
  • addcmul
  • addcmul_
  • asin
  • asin_
  • atan2
  • atan2_
  • batch_norm
  • bfloat16
  • bool
  • byte
  • cat
  • ceil
  • ceil_
  • char
  • clone
  • contiguous
  • conv2d
  • conv_transpose2d
  • copy_
  • cpu
  • cuda
  • cudnn_convolution
  • cudnn_convolution_transpose
  • detach
  • digamma
  • digamma_
  • div
  • div_
  • double
  • empty_like
  • erfinv
  • erfinv_
  • expm1
  • expm1_
  • fill_
  • float
  • floor
  • floor_
  • full_like
  • gt
  • half
  • int
  • isnan
  • log
  • log_
  • log10
  • log10_
  • log1p
  • log1p_
  • log2
  • log2_
  • long
  • lt
  • max_pool2d
  • mul
  • mul_
  • ne
  • neg
  • neg_
  • ones_like
  • pow
  • pow_
  • randint_like
  • rand_like
  • randn_like
  • relu
  • relu_
  • requires_grad_
  • resize_
  • resize_as_
  • round
  • round_
  • rsqrt
  • rsqrt_
  • short
  • sigmoid
  • sigmoid_
  • sign
  • sign_
  • sin
  • sin_
  • sinh
  • sinh_
  • sqrt
  • sqrt_
  • to
  • trunc
  • trunc_
  • type
  • zero_
  • zeros_like

I would love to contribute to PyTorch!