Searched defs:stats_bdim (Results 1 – 1 of 1) sorted by relevance
107 const auto stats_bdim = compute_stat_bdim(bdim_size, mean); in batch_norm_batch_rule() local499 const auto stats_bdim = compute_stat_bdim(input_bdim, mean); in native_layer_norm_batch_rule() local515 const auto stats_bdim = compute_stat_bdim(input_bdim, mean); in native_layer_norm_batch_rule() local