Searched defs:Softmax2d (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/nn/modules/ | ||
| D | activation.py | 1673 class Softmax2d(Module): class |
| /external/pytorch/test/cpp/api/ | ||
| D | modules.cpp | 2702 TEST_F(ModulesTest, Softmax2d) { in TEST_F() argument |