• 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
..
util Loading commit data...
abs.cpp Loading commit data...
abs.hpp Loading commit data...
acos.cpp Loading commit data...
acos.hpp Loading commit data...
add.cpp Loading commit data...
add.hpp Loading commit data...
allreduce.cpp Loading commit data...
allreduce.hpp Loading commit data...
and.cpp Loading commit data...
and.hpp Loading commit data...
argmax.cpp Loading commit data...
argmax.hpp Loading commit data...
argmin.cpp Loading commit data...
argmin.hpp Loading commit data...
asin.cpp Loading commit data...
asin.hpp Loading commit data...
atan.cpp Loading commit data...
atan.hpp Loading commit data...
avg_pool.cpp Loading commit data...
avg_pool.hpp Loading commit data...
batch_norm.cpp Loading commit data...
batch_norm.hpp Loading commit data...
broadcast.cpp Loading commit data...
broadcast.hpp Loading commit data...
ceiling.cpp Loading commit data...
ceiling.hpp Loading commit data...
concat.cpp Loading commit data...
concat.hpp Loading commit data...
constant.cpp Loading commit data...
constant.hpp Loading commit data...
convert.cpp Loading commit data...
convert.hpp Loading commit data...
convolution.cpp Loading commit data...
convolution.hpp Loading commit data...
cos.cpp Loading commit data...
cos.hpp Loading commit data...
cosh.cpp Loading commit data...
cosh.hpp Loading commit data...
divide.cpp Loading commit data...
divide.hpp Loading commit data...
dot.cpp Loading commit data...
dot.hpp Loading commit data...
equal.cpp Loading commit data...
equal.hpp Loading commit data...
exp.cpp Loading commit data...
exp.hpp Loading commit data...
floor.cpp Loading commit data...
floor.hpp Loading commit data...
function_call.cpp Loading commit data...
function_call.hpp Loading commit data...
get_output_element.cpp Loading commit data...
get_output_element.hpp Loading commit data...
greater.cpp Loading commit data...
greater.hpp Loading commit data...
greater_eq.cpp Loading commit data...
greater_eq.hpp Loading commit data...
less.cpp Loading commit data...
less.hpp Loading commit data...
less_eq.cpp Loading commit data...
less_eq.hpp Loading commit data...
log.cpp Loading commit data...
log.hpp Loading commit data...
lrn.cpp Loading commit data...
lrn.hpp Loading commit data...
max.cpp Loading commit data...
max.hpp Loading commit data...
max_pool.cpp Loading commit data...
max_pool.hpp Loading commit data...
maximum.cpp Loading commit data...
maximum.hpp Loading commit data...
min.cpp Loading commit data...
min.hpp Loading commit data...
minimum.cpp Loading commit data...
minimum.hpp Loading commit data...
multiply.cpp Loading commit data...
multiply.hpp Loading commit data...
negative.cpp Loading commit data...
negative.hpp Loading commit data...
not.cpp Loading commit data...
not.hpp Loading commit data...
not_equal.cpp Loading commit data...
not_equal.hpp Loading commit data...
one_hot.cpp Loading commit data...
one_hot.hpp Loading commit data...
op.cpp Loading commit data...
op.hpp Loading commit data...
or.cpp Loading commit data...
or.hpp Loading commit data...
pad.cpp Loading commit data...
pad.hpp Loading commit data...
parameter.cpp Loading commit data...
parameter.hpp Loading commit data...
parameter_vector.cpp Loading commit data...
parameter_vector.hpp Loading commit data...
power.cpp Loading commit data...
power.hpp Loading commit data...
product.cpp Loading commit data...
product.hpp Loading commit data...
reduce.cpp Loading commit data...
reduce.hpp Loading commit data...
regmodule_pyngraph_op.cpp Loading commit data...
regmodule_pyngraph_op.hpp Loading commit data...
relu.cpp Loading commit data...
relu.hpp Loading commit data...
replace_slice.cpp Loading commit data...
replace_slice.hpp Loading commit data...
reshape.cpp Loading commit data...
reshape.hpp Loading commit data...
reverse.cpp Loading commit data...
reverse.hpp Loading commit data...
select.cpp Loading commit data...
select.hpp Loading commit data...
sign.cpp Loading commit data...
sign.hpp Loading commit data...
sin.cpp Loading commit data...
sin.hpp Loading commit data...
sinh.cpp Loading commit data...
sinh.hpp Loading commit data...
slice.cpp Loading commit data...
slice.hpp Loading commit data...
softmax.cpp Loading commit data...
softmax.hpp Loading commit data...
sqrt.cpp Loading commit data...
sqrt.hpp Loading commit data...
subtract.cpp Loading commit data...
subtract.hpp Loading commit data...
sum.cpp Loading commit data...
sum.hpp Loading commit data...
tan.cpp Loading commit data...
tan.hpp Loading commit data...
tanh.cpp Loading commit data...
tanh.hpp Loading commit data...
topk.cpp Loading commit data...
topk.hpp Loading commit data...