Searched refs:reduction_functor_ (Results 1 – 1 of 1) sorted by relevance
470 : OpKernel(context), reduction_functor_(DeviceReductionFunctor()) {} in UnsortedSegmentReductionOp()495 reduction_functor_(context, output_rows, segment_ids.shape(), segment_flat, in Compute()500 DeviceReductionFunctor reduction_functor_; member in tensorflow::UnsortedSegmentReductionOp