#pragma once // @generated by torchgen/gen.py from Function.h #include #include #include #include #include #include #include #include #include #include #include #include #include namespace at { // aten::cross_entropy_loss(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, SymInt ignore_index=-100, float label_smoothing=0.0) -> Tensor inline at::Tensor cross_entropy_loss(const at::Tensor & self, const at::Tensor & target, const c10::optional & weight={}, int64_t reduction=at::Reduction::Mean, int64_t ignore_index=-100, double label_smoothing=0.0) { return at::_ops::cross_entropy_loss::call(self, target, weight, reduction, ignore_index, label_smoothing); } namespace symint { template ::value>> at::Tensor cross_entropy_loss(const at::Tensor & self, const at::Tensor & target, const c10::optional & weight={}, int64_t reduction=at::Reduction::Mean, int64_t ignore_index=-100, double label_smoothing=0.0) { return at::_ops::cross_entropy_loss::call(self, target, weight, reduction, ignore_index, label_smoothing); } } // aten::cross_entropy_loss(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, SymInt ignore_index=-100, float label_smoothing=0.0) -> Tensor inline at::Tensor cross_entropy_loss_symint(const at::Tensor & self, const at::Tensor & target, const c10::optional & weight={}, int64_t reduction=at::Reduction::Mean, c10::SymInt ignore_index=-100, double label_smoothing=0.0) { return at::_ops::cross_entropy_loss::call(self, target, weight, reduction, ignore_index, label_smoothing); } namespace symint { template ::value>> at::Tensor cross_entropy_loss(const at::Tensor & self, const at::Tensor & target, const c10::optional & weight={}, int64_t reduction=at::Reduction::Mean, c10::SymInt ignore_index=-100, double label_smoothing=0.0) { return at::_ops::cross_entropy_loss::call(self, target, weight, reduction, ignore_index, label_smoothing); } } }