Searched defs:class_dim (Results 1 – 4 of 4) sorted by relevance
496 const auto class_dim = self.dim() == 1 ? 0 : 1; in cross_entropy_loss_prob_target() local565 auto class_dim = self.dim() == 1 ? 0 : 1; in cross_entropy_loss_label_smoothing() local650 auto class_dim = self.dim() == 1 ? 0 : 1; in cross_entropy_loss_symint() local
551 auto class_dim = input.dim() - 1; in multilabel_soft_margin_loss() local
109 char class_dim; /* 1 to 8 */ member
219 DimensionHandle class_dim = c->Dim(scores, 2); in CombinedNMSShapeFn() local