Home
last modified time | relevance | path

Searched defs:BCEWithLogitsLoss (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dloss.py702 class BCEWithLogitsLoss(_Loss): class
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dfunctional.cpp3116 TEST_F(FunctionalTest, BCEWithLogitsLoss) { in TEST_F() argument
H A Dmodules.cpp3261 TEST_F(ModulesTest, BCEWithLogitsLoss) { in TEST_F() argument