Searched refs:compute_stat_bdim (Results 1 – 1 of 1) sorted by relevance
19 static std::optional<int64_t> compute_stat_bdim( in compute_stat_bdim() function107 const auto stats_bdim = compute_stat_bdim(bdim_size, mean); in batch_norm_batch_rule()499 const auto stats_bdim = compute_stat_bdim(input_bdim, mean); in native_layer_norm_batch_rule()515 const auto stats_bdim = compute_stat_bdim(input_bdim, mean); in native_layer_norm_batch_rule()