Searched defs:target_bdim (Results 1 – 1 of 1) sorted by relevance
32 std::optional<int64_t> target_bdim, int64_t reduction, in loss_batch_rule_helper()54 std::optional<int64_t> target_bdim, int64_t reduction) { in mse_loss_batch_rule()63 std::optional<int64_t> target_bdim, int64_t reduction, double delta) { in huber_loss_batch_rule()72 std::optional<int64_t> target_bdim, int64_t reduction, double beta) { in smooth_l1_loss_batch_rule()