Searched defs:unsorted_segmented_sum (Results 1 – 1 of 1) sorted by relevance
37 auto unsorted_segmented_sum = llvm::cast<mlir::TF::UnsortedSegmentSumOp>(op); in ComputeLayoutForward() local60 auto unsorted_segmented_sum = llvm::cast<mlir::TF::UnsortedSegmentSumOp>(op); in ComputeLayoutBackward() local