1. 25 2月, 2019 1 次提交
    • 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
  2. 29 1月, 2019 1 次提交
  3. 20 12月, 2018 1 次提交
  4. 19 12月, 2018 2 次提交
    • J
      - Added reusing MKL-DNN primitives for Transpose MKL-DNN op · 709d9e3c
      Jacek Czaja 提交于
      test=develop
      709d9e3c
    • J
      [MKL-DNN ]Added transpose/transpose2 Op (#14872) · aa6e9c30
      Jacek Czaja 提交于
      * - Added transpose MKLDNN Op
      - Few basic UT works
      - Added 1D transpose
      - implementing generic mem desc for MKLDNN transpose
      - Modified trnaspose op to support  more dimensional data eg. 5,6..10
      - Added is_test attribute to transpose op
      
      test=develop
      
      * - Added support for MKLDNN::memory::format::any for Transpose MKLDNN op
      
      test=develop
      
      * - Additional transpose mkldnn op correction to mkldnn layout
      
      test=develop
      
      * Cosmetic fixes
      
      test=develop
      
      * - Removed const_cast to obey coding standard
      
      test=develop
      aa6e9c30