• Nick Korovaiko's avatar
    BatchNorm splitting into ops (2nd try) (#1828) · 1beec46b
    Nick Korovaiko authored
    * split bn into bn_inference bn_training
    
    * fix warnings
    
    * Add GPU support for the new BN ops (#1569)
    
    * Add GPU support and change batchnorm_globalstats test to use BNInference.
    
    * Changed test back to using BNTraining for global stats and updated cudnn backend to account for it.
    
    * Fix issues in merge with master.
    
    * Formatting.
    
    * CPU fixes
    
    * remove 5-arg training BN for now
    
    * more fixes
    
    * python batchnorm changes
    
    * fix onnx_import
    
    * fix a call BatchNormInference c-tor
    
    * yet another fix to BatchNormInference c-tor
    
    * AND yet another fix to batchnorm_inference c-tor
    
    * ops.py
    
    * address adam's feedback
    
    * Remove unnecessary parameter/argument.
    
    * remove batch_norm_training_relu_with_global_stats
    
    * remove bn_relu (training)
    1beec46b
Name
Last commit
Last update
..
files Loading commit data...
models Loading commit data...
ref_generators Loading commit data...
util Loading commit data...
CMakeLists.txt Loading commit data...
algebraic_simplification.cpp Loading commit data...
all_close_f.cpp Loading commit data...
assertion.cpp Loading commit data...
autodiff.in.cpp Loading commit data...
backend_api.cpp Loading commit data...
backend_binary_elementwise.in.cpp Loading commit data...
backend_broadcast.in.cpp Loading commit data...
backend_comparison.in.cpp Loading commit data...
backend_debug_api.cpp Loading commit data...
backend_dot.in.cpp Loading commit data...
backend_one_hot.in.cpp Loading commit data...
backend_performance.cpp Loading commit data...
backend_pool.in.cpp Loading commit data...
backend_reduce.in.cpp Loading commit data...
backend_reshape.in.cpp Loading commit data...
backend_sum.in.cpp Loading commit data...
backend_test.in.cpp Loading commit data...
backend_unary_elementwise.in.cpp Loading commit data...
build_graph.cpp Loading commit data...
builder.cpp Loading commit data...
builder_autobroadcast.cpp Loading commit data...
constant_folding.cpp Loading commit data...
control_dependencies.cpp Loading commit data...
convolution_test.in.cpp Loading commit data...
coordinate.cpp Loading commit data...
copy.cpp Loading commit data...
core_fusion.cpp Loading commit data...
cpio.cpp Loading commit data...
cpu_fusion.cpp Loading commit data...
cpu_reshape_sinking.cpp Loading commit data...
cpu_test.cpp Loading commit data...
cse.cpp Loading commit data...
cudnn.cpp Loading commit data...
distributed.in.cpp Loading commit data...
element_type.cpp Loading commit data...
file_util.cpp Loading commit data...
gpu_fusion.cpp Loading commit data...
gpu_test.cpp Loading commit data...
graph_partition.cpp Loading commit data...
halide.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...
nop_elimination.cpp Loading commit data...
onnx_import.cpp Loading commit data...
onnxifi.cpp Loading commit data...
op.cpp Loading commit data...
partial_shape.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...
quantize_cpu.cpp Loading commit data...
reshape_elimination.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...
zero_dim_tensor_elimination.cpp Loading commit data...