Searched defs:UnsortedSegmentMin (Results 1 – 2 of 2) sorted by relevance
31 __global__ void UnsortedSegmentMin(const T *input, const int *segment_ids, const int64_t num_segmen… in UnsortedSegmentMin() function
2104 class UnsortedSegmentMin(PrimitiveWithCheck): class