Searched defs:weight_broadcast_shape (Results 1 – 1 of 1) sorted by relevance
520 auto weight_broadcast_shape = SmallBuffer<int64_t, 5>(input.dim()); in cross_entropy_loss_prob_target() local574 auto weight_broadcast_shape = SmallBuffer<int64_t, 5>(input.dim()); in cross_entropy_loss_label_smoothing() local