Searched refs:SegmentId (Results 1 – 1 of 1) sorted by relevance
1028 typedef int32 SegmentId; in Compute() typedef1029 const SegmentId M = in Compute()1030 internal::SubtleMustCopy(output_dim0.scalar<SegmentId>()()); in Compute()1035 const auto segment_vec = segment_ids.vec<SegmentId>(); in Compute()1045 const SegmentId num_segments = input.dim_size(0); in Compute()1046 const SegmentId last_segment_id_plus_one = in Compute()1054 const SegmentId idx = internal::SubtleMustCopy(segment_vec(i)); in Compute()1079 const SegmentId idx = internal::SubtleMustCopy(segment_vec(i)); in Compute()