• J
    Added pad3d and pad2d FP32 FWD oneDNN kernels (#43990) · 2792b8de
    jakpiase 提交于
    * Piotrek's changes for pad3d
    
    * my changes
    
    * first version of pad3d, single copy, unnecessary reads
    
    * optimized pad3d kernel
    
    * test upadte
    
    * removed magic numbers
    
    * added support for pad2d
    
    * reverted two files
    
    * reverted one old change
    
    * added support for Paddings tensor
    
    * CI fix
    
    * CI fix
    
    * fixed timeout of tests
    
    * fixed typo
    
    * changes to GetKernelTypeForVar
    
    * Revert "changes to GetKernelTypeForVar"
    
    This reverts commit 469106115c49682b25038a666fd71bd4a10fb66b.
    
    * added AsExtra() to pad2d
    Co-authored-by: NPiotr Paturej <piotr.paturej@intel.com>
    2792b8de
pad3d_mkldnn_op.cc 8.2 KB