Searched defs:segment_sum (Results 1 – 2 of 2) sorted by relevance
27 namespace segment_sum { namespace
267 def segment_sum(data: ragged_tensor.RaggedOrDense, function