Searched refs:affine_param_shape (Results 1 – 2 of 2) sorted by relevance
241 std::vector<int64_t> affine_param_shape(input.dim(), 1); in math_group_norm() local242 affine_param_shape[1] = C; in math_group_norm()244 out = bias.view(affine_param_shape) in math_group_norm()245 .addcmul(out, weight.view(affine_param_shape), 1); in math_group_norm()247 out = out.mul(weight.view(affine_param_shape)); in math_group_norm()249 out = out.add(bias.view(affine_param_shape)); in math_group_norm()
6323 std::vector<int64_t> affine_param_shape(input_p.dim(), 1); in group_norm_jvp() local6324 affine_param_shape[1] = C; in group_norm_jvp()6329 weight_p.defined() ? weight_p.view(affine_param_shape) : weight_p, in group_norm_jvp()6330 weight_t.defined() ? weight_t.view(affine_param_shape) : weight_t, in group_norm_jvp()6331 bias_t.defined() ? bias_t.view(affine_param_shape) : bias_t); in group_norm_jvp()