Searched refs:batch_normalization_flag (Results 1 – 4 of 4) sorted by relevance
60 batch_normalization_flag::use_scale_shift | in test_forward_inference()61 batch_normalization_flag::use_global_stats, in test_forward_inference()78 batch_normalization_flag::use_scale_shift, in test_forward_training()98 gradx, gscale, gshift, batch_normalization_flag::use_scale_shift, in test_backward()
106 using batch_normalization_flag = mkldnn::batch_normalization_flag; typedef
4664 unsigned flag = batch_normalization_flag::use_global_stats | in init()4665 batch_normalization_flag::use_scale_shift) { in init()4722 batch_normalization_flag::use_scale_shift, epsilon); in compute()4819 unsigned flags = batch_normalization_flag::use_scale_shift) { in init()5050 unsigned flags = batch_normalization_flag::use_scale_shift, in init()
124 | C++ | mkldnn::batch_normalization_flag | [mkldnn::normalization_flags](@ref dnnl::normalization_…