Searched refs:kPrimUnsortedSegmentMax (Results 1 – 3 of 3) sorted by relevance
49 const auto kUnsortedSegmentMax = prim::kPrimUnsortedSegmentMax->name(); in GetDependsFormMap()116 {prim::kPrimUnsortedSegmentMax, {InferImplUnsortedSegmentMax, nullptr, true}}, in GetPrimitiveToEvalImplMap()
56 Register(prim::kPrimUnsortedSegmentMax->name(), {2}); in ConstInputToAttrInfoRegistry()
196 inline const PrimitivePtr kPrimUnsortedSegmentMax = std::make_shared<Primitive>("UnsortedSegmentMax… variable