Searched refs:segment_flat (Results 1 – 1 of 1) sorted by relevance
286 auto segment_flat = segment_ids.flat<Index>(); in ComputeAsync() local288 segment_ids.shape(), segment_flat, input.NumElements(), data_ptr, in ComputeAsync()480 const auto segment_flat = segment_ids.flat<Index>(); in Compute() local495 reduction_functor_(context, output_rows, segment_ids.shape(), segment_flat, in Compute()