• J
    Added shape (U)INT8/BF16/FP32 oneDNN kernel (#36033) · 52bbaae9
    jakpiase 提交于
    * added shape oneDNN kernel
    
    * removed unnecessary import from test
    
    * added skipping tests for GPU
    
    * refactoring
    
    * refactored shape kernel
    
    * added tests in new framework
    
    * removed one line
    
    * minor change
    
    * added newline at EOF
    
    * added formatting
    
    * added attributes as extra
    52bbaae9
shape_mkldnn_op.cc 1.5 KB