• J
    [MKL-DNN] MKL-DNN specific Tensor modification (#15429) · dec9cf53
    Jacek Czaja 提交于
    * - Implemented draft of primitive desc keeping in Tensor
    
    test=develop
    
    - TransposeMKLDNNHandler::AcquireSrcMemory was reimplemented
    
    - Added nchw and nc formats setting for sake of compatiblity
    
    Fixed unit tests
    
    - Worakaround to problem with 5D data in conv
    
    - Added 3D and 1D MKL-DNN formats for name handles for tensor
    
    test=develop
    
    - Fix to UTs
    
    test=develop
    
    - Conv fp32 op was updated
    
    Cosmetic fixes
    
    test=develop
    
    - tensor mkldnn cosmetics
    
    test=develop
    
    - Moved most of mkl-dnn specific code from Tensor to mkl-dnn utils
    
    * - Lint fixes
    
    test=develop
    
    * - setting prim dec in Tensor , sets also layout to kMKLDNN
    
    test=develop
    
    * - Moved creation of prim desc totally out of Tensor
    
    test=develop
    
    * - Cosmetic fixes adter review
    
    test=develop
    dec9cf53
transpose_mkldnn_op.cc 6.0 KB