Searched refs:toNonOptTensor (Results 1 – 2 of 2) sorted by relevance
64 Tensor toNonOptTensor(const std::optional<Tensor>& t) { in toNonOptTensor() function4689 training ? toNonOptTensor(save_mean).to(input.scalar_type()) in batchnorm_double_backward()4690 : toNonOptTensor(running_mean), in batchnorm_double_backward()4694 training ? toNonOptTensor(save_invstd).to(input.scalar_type()) in batchnorm_double_backward()4695 : toNonOptTensor(running_var).add(Scalar(eps)).pow(-0.5), in batchnorm_double_backward()5011 .reshape_as(toNonOptTensor(gamma)); in infinitely_differentiable_native_group_norm_backward()5014 dbeta = db.sum(0).reshape_as(toNonOptTensor(gamma)); in infinitely_differentiable_native_group_norm_backward()
36 TORCH_API Tensor toNonOptTensor(const std::optional<Tensor>& t);