Searched defs:ArgMax (Results 1 – 2 of 2) sorted by relevance
36 ArgMax() : PrimitiveC(kNameArgMax) { InitIOName({"x"}, {"output"}); } in ArgMax() function37 explicit ArgMax(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x"}, {"output"}); } in ArgMax() function
24 class ArgMax(nn.Cell): class