• Pruthvi's avatar
    bn fprop mkldnn optimized implementation (#581) · 9db548c6
    Pruthvi authored
    * - Added support optimized bn mkldnn implementation in cpu emitter
    - modified bn unit_test to support new implementation
    - added layout assignment for bn op
    - Style Fix
    
    (cherry picked from commit 7747a40806d62c126059d5c873adcd2e61a0adb0)
    
    * modified value initilization in cpu_fusion to be float explicit
    
    (cherry picked from commit 03499d380073d0197ab8cbd154eb03f63b042a48)
    
    * fix compilation issue
    
    * Addressed PR comments
    - added exception if gamma and beta layout isnot equal to memory::format::x
    - throw exception if bn Op is not mkldnn op
    
    * fix compilation issue
    
    * added support to handle multiple o/ps in fprop bn fusion
    
    * - Removed laytout pass for bn
    - fixed autodiff bug in bn
    - added "Add" for the dispatcher in cpu-layout pass
    
    * style fix
    
    * Fix bprop batchnorm test with get_output_elements
    
    * Style fix
    9db548c6
Name
Last commit
Last update
..
models Loading commit data...
ref_generators Loading commit data...
util Loading commit data...
CMakeLists.txt Loading commit data...
argon_fusion.cpp Loading commit data...
autodiff.in.cpp Loading commit data...
backend_debug_api.cpp Loading commit data...
backend_performance.cpp Loading commit data...
backend_test.in.cpp Loading commit data...
build_graph.cpp Loading commit data...
builder.cpp Loading commit data...
builder_autobroadcast.cpp Loading commit data...
builder_xla.cpp Loading commit data...
codegen.cpp Loading commit data...
convolution_test.in.cpp Loading commit data...
copy.cpp Loading commit data...
core_fusion.cpp Loading commit data...
cpu_fusion.cpp Loading commit data...
cudnn.cpp Loading commit data...
distributed.cpp Loading commit data...
eigen.cpp Loading commit data...
element_type.cpp Loading commit data...
file_util.cpp Loading commit data...
graph_partition.cpp Loading commit data...
includes.cpp Loading commit data...
inliner.cpp Loading commit data...
input_output_assign.cpp Loading commit data...
main.cpp Loading commit data...
mkldnn.cpp Loading commit data...
ngraph.cpp Loading commit data...
op.cpp Loading commit data...
pass_liveness.cpp Loading commit data...
pass_manager.cpp Loading commit data...
pass_memory_layout.cpp Loading commit data...
pattern.cpp Loading commit data...
reshape_elimination.cpp Loading commit data...
runtime_manager.cpp Loading commit data...
serialize.cpp Loading commit data...
shape.cpp Loading commit data...
tensor.cpp Loading commit data...
type_prop.cpp Loading commit data...
update_reference.sh Loading commit data...
util.cpp Loading commit data...
uuid.cpp Loading commit data...