Searched refs:kPrimArgMax (Results 1 – 3 of 3) sorted by relevance
28 prim::kPrimArgMax, prim::kPrimArgMin}; in IsReduce()
64 {kGPUDevice, OpLevel_1, prim::kPrimArgMax}, in GetClusterableOpList()
190 inline const PrimitivePtr kPrimArgMax = std::make_shared<Primitive>("Argmax"); variable