Searched defs:SegmentSum (Results 1 – 3 of 3) sorted by relevance
994 inline void SegmentSum(const RuntimeShape& input_shape, const T* input_data, in SegmentSum() function
975 TEST_F(NaryGradTest, SegmentSum) { in TEST_F() argument
42328 func SegmentSum(scope *Scope, data tf.Output, segment_ids tf.Output) (output tf.Output) { func