Searched refs:UNSORTED_SEGMENT_MAX (Results 1 – 4 of 4) sorted by relevance
62 {UNSORTED_SEGMENT_MAX, OperatorType::kRecUnsortedSegmentOp},
316 …auto segment_max = gen_g.PushBack({gen_g.NewOpInst(UNSORTED_SEGMENT_MAX), gen_g.virtual_input_node… in ComputeReplaceGraph()
392 constexpr char UNSORTED_SEGMENT_MAX[] = "UnsortedSegmentMax"; variable
174 …UNSORTED_SEGMENT_MAX, GATHER_ND, TOPK, SCATTER_UPDATE, VIRTUAL_OUTPUT, CONV2D_BACK_PROP_INPUT, CON… in IsSplittableOperator()