• Home
  • Raw
  • Download

Lines Matching defs:reduction

64   static inline at::Tensor apply_loss_reduction(const at::Tensor& unreduced, int64_t reduction) {  in apply_loss_reduction()
144 …nst Tensor& input1, const Tensor& input2, const Tensor& target, double margin, int64_t reduction) { in cosine_embedding_loss()
182 … hinge_embedding_loss(const Tensor& self, const Tensor& target, double margin, int64_t reduction) { in hinge_embedding_loss()
200 double p, double eps, bool swap, int64_t reduction) { in triplet_margin_loss()
225 …nst Tensor& input1, const Tensor& input2, const Tensor& target, double margin, int64_t reduction) { in margin_ranking_loss()
238 Tensor kl_div(const Tensor& input, const Tensor& target, int64_t reduction, bool log_target) { in kl_div()
253 … Tensor& input, const Tensor& target, const std::optional<Tensor>& weight_opt, int64_t reduction) { in binary_cross_entropy_cpu()
263 …put, const Tensor& target, const std::optional<Tensor>& weight_opt, int64_t reduction, Tensor& los… in binary_cross_entropy_out_cpu()
310 … Tensor& input, const Tensor& target, const std::optional<Tensor>& weight_opt, int64_t reduction) { in binary_cross_entropy_backward_cpu()
320 …put, const Tensor& target, const std::optional<Tensor>& weight_opt, int64_t reduction, Tensor& gra… in binary_cross_entropy_backward_out_cpu()
362 …td::optional<Tensor>& weight_opt, const std::optional<Tensor>& pos_weight_opt, int64_t reduction) { in binary_cross_entropy_with_logits()
385 …t Tensor& target, const bool log_input, const bool full, const double eps, const int64_t reduction) in poisson_nll_loss()
402 …nst Tensor& grad_output, const Tensor& input, const Tensor& target, int64_t reduction, Tensor& gra… in soft_margin_loss_backward_out()
412 …backward(const Tensor& grad_output, const Tensor& input, const Tensor& target, int64_t reduction) { in soft_margin_loss_backward()
420 int64_t reduction, in soft_margin_loss_out()
435 int64_t reduction) { in soft_margin_loss()
441 …nst Tensor& grad_output, const Tensor& input, const Tensor& target, int64_t reduction, double beta… in smooth_l1_loss_backward_out()
456 …nst Tensor& grad_output, const Tensor& input, const Tensor& target, int64_t reduction, double beta… in smooth_l1_loss_backward()
461 Tensor huber_loss(const Tensor& input, const Tensor& target, int64_t reduction, double delta) { in huber_loss()
469 Tensor& huber_loss_out(const Tensor& input, const Tensor& target, int64_t reduction, double delta, … in huber_loss_out()
481 …nst Tensor& grad_output, const Tensor& input, const Tensor& target, int64_t reduction, double delt… in huber_loss_backward()
486 …nst Tensor& grad_output, const Tensor& input, const Tensor& target, int64_t reduction, double delt… in huber_loss_backward_out()
498 …backward(const Tensor& grad_output, const Tensor& input, const Tensor& target, int64_t reduction) { in mse_loss_backward()
504 const Tensor& input, const Tensor& target, int64_t reduction, Tensor& grad_input) { in mse_loss_backward_out()
516 Tensor l1_loss(const Tensor& input, const Tensor& target, int64_t reduction) { in l1_loss()