Searched refs:NetArgmaxWithValueBig (Results 1 – 2 of 2) sorted by relevance
38 class NetArgmaxWithValueBig(nn.Cell): class40 super(NetArgmaxWithValueBig, self).__init__()82 argmax = NetArgmaxWithValueBig(0)89 argmax = NetArgmaxWithValueBig(1)96 argmax = NetArgmaxWithValueBig(2)