Searched defs:rstd_bdim (Results 1 – 1 of 1) sorted by relevance
133 const at::Tensor & rstd, std::optional<int64_t> rstd_bdim, in batch_norm_backward_no_weight_bias_batch_rule()344 const at::Tensor & rstd, std::optional<int64_t> rstd_bdim, in group_norm_backward_no_weight_bias_batch_rule()538 const at::Tensor & rstd, std::optional<int64_t> rstd_bdim) { in native_layer_norm_backward_no_weight_bias_batch_rule()