Operators with Channels Last support (original) (raw)
Operators with Channels Last support
CPU operators
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
GPU operators
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