Searched refs:segment_id_dim (Results 1 – 1 of 1) sorted by relevance
2728 int64_t segment_id_dim = std::get<0>(shape_pair); in VerifyUnsortedSegmentReduction() local2730 if (!ShapedType::isDynamic(segment_id_dim) && in VerifyUnsortedSegmentReduction()2731 !ShapedType::isDynamic(data_dim) && segment_id_dim != data_dim) in VerifyUnsortedSegmentReduction()2735 << index << " differs: " << segment_id_dim << " vs. " in VerifyUnsortedSegmentReduction()