Searched refs:BinaryCrossEntropyWithLogitsFuncOptions (Results 1 – 3 of 3) sorted by relevance
3148 F::BinaryCrossEntropyWithLogitsFuncOptions().weight(weight)), in TEST_F()3165 F::BinaryCrossEntropyWithLogitsFuncOptions().reduction( in TEST_F()3177 F::BinaryCrossEntropyWithLogitsFuncOptions().weight(weight)), in TEST_F()3189 F::BinaryCrossEntropyWithLogitsFuncOptions().reduction(torch::kSum)) in TEST_F()3202 F::BinaryCrossEntropyWithLogitsFuncOptions().weight(weight)); in TEST_F()3208 F::BinaryCrossEntropyWithLogitsFuncOptions().weight(weight)); in TEST_F()3216 F::BinaryCrossEntropyWithLogitsFuncOptions().weight(weight)); in TEST_F()3222 F::BinaryCrossEntropyWithLogitsFuncOptions().weight(weight)); in TEST_F()3236 F::BinaryCrossEntropyWithLogitsFuncOptions().pos_weight( in TEST_F()3246 F::BinaryCrossEntropyWithLogitsFuncOptions().pos_weight(pos_weight)); in TEST_F()[all …]
996 BinaryCrossEntropyWithLogitsFuncOptions::reduction_t reduction, in binary_cross_entropy_with_logits()1033 const BinaryCrossEntropyWithLogitsFuncOptions& options = {}) {
798 using BinaryCrossEntropyWithLogitsFuncOptions = BCEWithLogitsLossOptions; variable