Searched refs:UnsortedSegmentReduce (Results 1 – 1 of 1) sorted by relevance
27 class UnsortedSegmentReduce : public XlaOpKernel { class29 explicit UnsortedSegmentReduce(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in UnsortedSegmentReduce() function in tensorflow::__anon7ff0dacf0111::UnsortedSegmentReduce96 class UnsortedSegmentSum : public UnsortedSegmentReduce {99 : UnsortedSegmentReduce(ctx) {} in UnsortedSegmentSum()111 class UnsortedSegmentProd : public UnsortedSegmentReduce {114 : UnsortedSegmentReduce(ctx) {} in UnsortedSegmentProd()126 class UnsortedSegmentMin : public UnsortedSegmentReduce {129 : UnsortedSegmentReduce(ctx) {} in UnsortedSegmentMin()143 class UnsortedSegmentMax : public UnsortedSegmentReduce {146 : UnsortedSegmentReduce(ctx) {} in UnsortedSegmentMax()